fix "InvalidOperation: Cannot use MongoClient after close" in test_db

StackStorm · Sep 19, 2024 · cfec5e0 · cfec5e0
1 parent 72fb6fb
commit cfec5e0
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/st2common/tests/unit/test_db.py b/st2common/tests/unit/test_db.py
@@ -114,6 +114,13 @@ def tearDown(self):
         disconnect()
         cfg.CONF.reset()
 
+    @classmethod
+    def tearDownClass(cls):
+        # since tearDown discconnects, dropping the database in tearDownClass
+        # fails withotu establishing a new connection.
+        cls._establish_connection_and_re_create_db()
+        super().tearDownClass()
+
     def test_check_connect(self):
         """
         Tests connectivity to the db server. Requires the db server to be