add again declare custom exchange test

Author: DanielePalaia

tests/test_connection.py

@@ -59,20 +59,20 @@ def test_environment_connections_management() -> None:
     connection3 = environment.connection("amqp://guest:guest@localhost:5672/")
     connection3.dial()
 
-    assert len(environment.connections()) == 3
+    assert environment.active_connections == 3
 
     # this shouldn't happen but we test it anyway
     connection.close()
 
-    assert len(environment.connections()) == 2
+    assert environment.active_connections == 2
 
     connection2.close()
 
-    assert len(environment.connections()) == 1
+    assert environment.active_connections == 1
 
     connection3.close()
 
-    assert len(environment.connections()) == 0
+    assert environment.active_connections == 0
 
     environment.close()
 

tests/test_management.py

@@ -2,6 +2,7 @@
 
 from rabbitmq_amqp_python_client import (
     ClassicQueueSpecification,
+    ExchangeCustomSpecification,
     ExchangeSpecification,
     ExchangeToExchangeBindingSpecification,
     ExchangeToQueueBindingSpecification,
@@ -42,6 +43,26 @@ def test_declare_delete_exchange_headers(management: Management) -> None:
     management.delete_exchange(exchange_name)
 
 
+def test_declare_delete_exchange_custom(management: Management) -> None:
+
+    exchange_name = "test-exchange"
+
+    exchange_arguments = {}
+    exchange_arguments["x-delayed-type"] = "direct"
+
+    exchange_info = management.declare_exchange(
+        ExchangeCustomSpecification(
+            name=exchange_name,
+            exchange_type="x-local-random",
+            arguments=exchange_arguments,
+        )
+    )
+
+    assert exchange_info.name == exchange_name
+
+    management.delete_exchange(exchange_name)
+
+
 def test_declare_delete_exchange_with_args(management: Management) -> None:
 
     exchange_name = "test-exchange-with-args"