fixup! fix for mysql and postgres

apache · Nov 22, 2024 · 031bfd8 · 031bfd8
1 parent 6bddfe9
commit 031bfd8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests_common/test_utils/db.py b/tests_common/test_utils/db.py
@@ -62,9 +62,12 @@ def initial_db_init():
     from airflow.www.extensions.init_appbuilder import init_appbuilder
     from airflow.www.extensions.init_auth_manager import get_auth_manager
 
+    from tests_common.test_utils.compat import AIRFLOW_V_3_0_PLUS
+
     db.resetdb()
-    db.downgrade(to_revision="5f2621c13b39")
-    db.upgradedb(to_revision="head")
+    if AIRFLOW_V_3_0_PLUS:
+        db.downgrade(to_revision="5f2621c13b39")
+        db.upgradedb(to_revision="head")
     db.bootstrap_dagbag()
     # minimal app to add roles
     flask_app = Flask(__name__)