Add check for connection closure in shutdown

NeonGeckoCom · Jan 14, 2025 · 3ebd805 · 3ebd805
1 parent f6045fc
commit 3ebd805
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/neon_mq_connector/utils/client_utils.py b/neon_mq_connector/utils/client_utils.py
@@ -60,6 +60,8 @@ def __init__(self, config: dict, service_name: str, vhost: str):
     def shutdown(self):
         MQConnector.stop(self)
         self.connection.close()
+        if not self.connection.is_closed:
+            raise RuntimeError(f"Connection is still open: {self.connection}")
 
 
 def send_mq_request(vhost: str, request_data: dict, target_queue: str,