fix: use tearDownClass

PostHog · Dec 16, 2024 · 448daa3 · 448daa3
1 parent 1f993e8
commit 448daa3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/posthog/test/base.py b/posthog/test/base.py
@@ -792,8 +792,9 @@ def _execute_migration_with_snapshots(self, executor):
     def setUpBeforeMigration(self, apps):
         pass
 
-    def tearDown(self):
-        super().tearDown()  # type: ignore
+    @classmethod
+    def tearDownClass(cls):
+        super().tearDownClass()  # type: ignore
         executor = MigrationExecutor(connection)  # Reset Django's migration state
         targets = executor.loader.graph.leaf_nodes()
         executor.migrate(targets)  # Migrate to the latest migration