close rabbitmq connections on teardown

canonical · Sep 11, 2024 · 8fbcc59 · 8fbcc59
1 parent f266001
commit 8fbcc59
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/examples/flask/test_rock/app.py b/examples/flask/test_rock/app.py
@@ -143,6 +143,12 @@ def teardown_database(_):
     boto3_client = g.pop("boto3_client", None)
     if boto3_client is not None:
         boto3_client.close()
+    rabbitmq = g.pop("rabbitmq", None)
+    if rabbitmq is not None:
+        rabbitmq.close()
+    rabbitmq_from_uri = g.pop("rabbitmq_from_uri", None)
+    if rabbitmq_from_uri is not None:
+        rabbitmq_from_uri.close()
 
 
 @app.route("/")