Update conftest.py

Author: duprijil

tests/unit/dags/conftest.py

@@ -9,23 +9,19 @@
 from dags.fetch_notices_by_date import FETCHER_DAG_NAME
 from tests import AIRFLOW_DAG_FOLDER
 
-@pytest.fixture(scope="function", autouse=True)
+@pytest.fixture(scope="function")
 def setup_airflow():
-    # Setup
     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()
 
-    # Run tests
-    yield
-
-    # 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,
+            dag_fetch_schedule_variable_name,
+            setup_airflow) -> DagBag:
     Variable.delete(key=dag_materialised_view_update_schedule_variable_name)
     Variable.delete(key=dag_fetch_schedule_variable_name)
     return DagBag(