validate message

Author: Gsantomaggio

examples/streams/example_with_streams.py

@@ -85,7 +85,7 @@ def create_connection(environment: Environment) -> Connection:
 
 
 def main() -> None:
-    queue_name = "example-queue"
+    queue_name = "stream-example-queue"
 
     print("connection to amqp server")
     environment = Environment("amqp://guest:guest@localhost:5672/")

rabbitmq_amqp_python_client/qpid/proton/_message.py

@@ -113,7 +113,7 @@ class Message(object):
     """ Default AMQP message priority"""
 
     def __init__(
-        self, body: Union[str, bytes, dict, None] = None, inferred=True, **kwargs
+        self, body: Union[bytes, None] = None, inferred=True, **kwargs
     ) -> None:
         # validate the types
 

tests/test_publisher.py

@@ -19,6 +19,31 @@
 from .utils import create_binding, publish_per_message
 
 
+def test_validate_message_for_publishing(connection: Connection) -> None:
+    queue_name = "validate-publishing"
+    management = connection.management()
+    management.declare_queue(QuorumQueueSpecification(name=queue_name))
+    publisher = connection.publisher(
+        destination=AddressHelper.queue_address(queue_name)
+    )
+    try:
+        publisher.publish(
+            Message(body=Converter.string_to_bytes("test"), inferred=False)
+        )
+    except ArgumentOutOfRangeException as e:
+        assert e.msg == "Message inferred must be True"
+
+    try:
+        publisher.publish(Message(body="test"))
+    except ArgumentOutOfRangeException as e:
+        assert e.msg == "Message body must be of type bytes or None"
+
+    try:
+        publisher.publish(Message(body={"key": "value"}))
+    except ArgumentOutOfRangeException as e:
+        assert e.msg == "Message body must be of type bytes or None"
+
+
 def test_publish_queue(connection: Connection) -> None:
 
     queue_name = "test-queue"