Fix compact tests

apache · Jan 15, 2025 · bed2fa1 · bed2fa1
1 parent 4f5ad41
commit bed2fa1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests_common/pytest_plugin.py b/tests_common/pytest_plugin.py
@@ -832,7 +832,10 @@ def __exit__(self, type, value, traceback):
                 return
 
             dag.clear(session=self.session)
-            dag.bulk_write_to_db(self.bundle_name, None, [dag], session=self.session)
+            if AIRFLOW_V_3_0_PLUS:
+                dag.bulk_write_to_db(self.bundle_name, None, [dag], session=self.session)
+            else:
+                dag.sync_to_db(session=self.session)
 
             if dag.access_control:
                 from airflow.www.security_appless import ApplessAirflowSecurityManager