diff --git a/pyinfra/test/unit_tests/conftest.py b/pyinfra/test/unit_tests/conftest.py index ce920b6..a0296ea 100644 --- a/pyinfra/test/unit_tests/conftest.py +++ b/pyinfra/test/unit_tests/conftest.py @@ -126,9 +126,14 @@ def queue_manager(queue_manager_name): queue_manager = get_queue_manager(queue_manager_name) yield queue_manager if queue_manager_name == "pika": + # try: + queue_manager.connection.close() + # except (pika.exceptions.StreamLostError, ConnectionResetError): + # pass try: - queue_manager.connection.close() - except (pika.exceptions.StreamLostError, ConnectionResetError): + queue_manager.channel.close() + except pika.exceptions.ChannelWrongStateError: + logger.debug("Channel was already closed when attempting to close explicitly.") pass