fix!: Fix problem with failing dags tests on test all

Author: duprijil

tests/unit/dags/conftest.py

@@ -1,36 +1,38 @@
 import os
 import tempfile
+from typing import Generator
 
 import pytest
 from airflow.models import DagBag, Variable
 from airflow.timetables.trigger import CronTriggerTimetable
 from airflow.utils.db import resetdb, initdb
+
 from dags.daily_materialized_views_update import DAILY_MATERIALISED_VIEWS_DAG_NAME
 from dags.fetch_notices_by_date import FETCHER_DAG_NAME
 from tests import AIRFLOW_DAG_FOLDER
 
+
 @pytest.fixture(scope="session", autouse=True)
-def setup_airflow():
-    # Setup
+def setup_airflow_for_all_tests() -> Generator[None, None, None]:
     temp_db_file = tempfile.NamedTemporaryFile(mode="w+", suffix=".db")
     os.environ["AIRFLOW__CORE__SQL_ALCHEMY_CONN"] = f"sqlite:///{temp_db_file.name}"
     os.environ["AIRFLOW__CORE__LOAD_EXAMPLES"] = "False"
     os.environ["AIRFLOW__CORE__UNIT_TEST_MODE"] = "True"
-    initdb(use_migration_files=False)
-
-    # Run tests
+    resetdb()
+    initdb()
     yield
+    if os.path.exists(temp_db_file.name):
+        temp_db_file.close()
 
-    # Cleanup
-    temp_db_file.close()
 
 @pytest.fixture
-def dag_bag(dag_materialised_view_update_schedule_variable_name, dag_fetch_schedule_variable_name) -> DagBag:
+def dag_bag(dag_materialised_view_update_schedule_variable_name: str,
+            dag_fetch_schedule_variable_name: str) -> DagBag:
     Variable.delete(key=dag_materialised_view_update_schedule_variable_name)
     Variable.delete(key=dag_fetch_schedule_variable_name)
     return DagBag(
         dag_folder=AIRFLOW_DAG_FOLDER,
-        include_examples=False, #Same as: os.environ["AIRFLOW__CORE__LOAD_EXAMPLES"] = "False"
+        include_examples=False,  # Same as: os.environ["AIRFLOW__CORE__LOAD_EXAMPLES"] = "False"
         read_dags_from_db=False,
         collect_dags=True)