Skip to content

Commit cdfaeb7

Browse files
committed
fix!: Fix problem with initdb on tests with use_migration_files param
1 parent 1507655 commit cdfaeb7

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/unit/dags/conftest.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,19 +4,19 @@
44
import pytest
55
from airflow.models import DagBag, Variable
66
from airflow.timetables.trigger import CronTriggerTimetable
7-
from airflow.utils.db import initdb
7+
from airflow.utils.db import resetdb, initdb
88
from dags.daily_materialized_views_update import DAILY_MATERIALISED_VIEWS_DAG_NAME
99
from dags.fetch_notices_by_date import FETCHER_DAG_NAME
1010
from tests import AIRFLOW_DAG_FOLDER
1111

12-
@pytest.fixture(scope="module", autouse=True)
12+
@pytest.fixture(scope="session", autouse=True)
1313
def setup_airflow():
1414
# Setup
1515
temp_db_file = tempfile.NamedTemporaryFile(mode="w+", suffix=".db")
1616
os.environ["AIRFLOW__CORE__SQL_ALCHEMY_CONN"] = f"sqlite:///{temp_db_file.name}"
1717
os.environ["AIRFLOW__CORE__LOAD_EXAMPLES"] = "False"
1818
os.environ["AIRFLOW__CORE__UNIT_TEST_MODE"] = "True"
19-
initdb()
19+
initdb(use_migration_files=False)
2020

2121
# Run tests
2222
yield

0 commit comments

Comments
 (0)