test

Author: DanielePalaia

tests/conftest.py

@@ -46,11 +46,11 @@ def environment_auth(pytestconfig):
         uri="amqp://localhost:5672",
         oauth2_options=OAuth2Options(token=token_string),
     )
-    # try:
-    return environment
+    try:
+        yield environment
 
-    # finally:
-    #     environment.close()
+    finally:
+        environment.close()
 
 
 @pytest.fixture()

tests/test_connection.py

@@ -50,6 +50,14 @@ def test_connection_auth(environment_auth: Environment) -> None:
     connection.close()
 
 
+def test_connection_auth_with_timeout(environment_auth: Environment) -> None:
+
+    connection = environment_auth.connection()
+    connection.dial()
+    time.sleep(3)
+    connection.close()
+
+
 def test_environment_connections_management() -> None:
 
     environment = Environment(uri="amqp://guest:guest@localhost:5672/")