Force airflow db reset

dlt-hub · Jul 5, 2024 · abf3e0c · abf3e0c
1 parent b6d08aa
commit abf3e0c
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -111,3 +111,14 @@ def _create_pipeline_instance_id(self) -> str:
     # disable databricks logging
     for log in ["databricks.sql.client"]:
         logging.getLogger(log).setLevel("WARNING")
+
+    # reset airflow db
+    from airflow.utils import db
+
+    for log in ["airflow"]:
+        logging.getLogger(log).setLevel("ERROR")
+
+    try:
+        db.resetdb()
+    except Exception:
+        pass