adding a publisher test

Author: DanielePalaia

examples/getting_started/main.py

@@ -4,7 +4,7 @@
     ExchangeSpecification,
     Message,
     QueueType,
-    StreamSpecification,
+    QuorumQueueSpecification,
     exchange_address,
 )
 
@@ -24,7 +24,7 @@ def main() -> None:
     management.declare_exchange(ExchangeSpecification(name=exchange_name, arguments={}))
 
     management.declare_queue(
-        StreamSpecification(name=queue_name, queue_type=QueueType.stream)
+        QuorumQueueSpecification(name=queue_name, queue_type=QueueType.quorum)
     )
 
     print("binding queue to exchange")

tests/test_publisher.py

@@ -1,13 +1,16 @@
 import time
 
 from rabbitmq_amqp_python_client import (
+    BindingSpecification,
     Connection,
+    ExchangeSpecification,
     Message,
     QuorumQueueSpecification,
+    exchange_address,
 )
 
 
-def test_publish_exchange(connection: Connection) -> None:
+def test_publish_queue(connection: Connection) -> None:
 
     queue_name = "test-queue"
     management = connection.management()
@@ -29,6 +32,43 @@ def test_publish_exchange(connection: Connection) -> None:
     management.delete_queue(queue_name)
 
 
+def test_publish_exchange(connection: Connection) -> None:
+
+    exchange_name = "test-exchange"
+    queue_name = "test-queue"
+    management = connection.management()
+    routing_key = "routing-key"
+
+    management.declare_exchange(ExchangeSpecification(name=exchange_name, arguments={}))
+
+    management.declare_queue(QuorumQueueSpecification(name=queue_name))
+
+    management.bind(
+        BindingSpecification(
+            source_exchange=exchange_name,
+            destination_queue=queue_name,
+            binding_key=routing_key,
+        )
+    )
+
+    addr = exchange_address(exchange_name, routing_key)
+
+    raised = False
+
+    try:
+        publisher = connection.publisher(addr)
+        publisher.publish(Message(body="test"))
+    except Exception:
+        raised = True
+
+    assert raised is False
+
+    publisher.close()
+
+    management.delete_exchange(exchange_name)
+    management.delete_queue(queue_name)
+
+
 def test_publish_purge(connection: Connection) -> None:
     connection = Connection("amqp://guest:guest@localhost:5672/")
     connection.dial()