From e3812725acff1df2a26e337dbc88df11cd6b8b22 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Thu, 24 Mar 2022 10:12:15 +0100 Subject: [PATCH] pika tearndown adjusted --- pyinfra/test/unit_tests/conftest.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) 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