fix(tests): restore the latest migration after migration tests (#26928)

PostHog · Dec 16, 2024 · 16730d3 · 16730d3
1 parent d149515
commit 16730d3
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/posthog/test/base.py b/posthog/test/base.py
@@ -792,6 +792,14 @@ def _execute_migration_with_snapshots(self, executor):
     def setUpBeforeMigration(self, apps):
         pass
 
+    @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
+        executor.loader.build_graph()  # Reload.
+
 
 class TestMigrations(BaseTestMigrations, BaseTest):
     """