Skip to content

Commit ae9ff67

Browse files
committed
fix: Only sync dags in Airflow 3.1 or later
1 parent 236f463 commit ae9ff67

1 file changed

Lines changed: 17 additions & 6 deletions

File tree

tests/dags/test_dbt_dags.py

Lines changed: 17 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,8 @@ def basic_dag(
205205

206206
dbt_seed >> dbt_run >> dbt_test
207207

208-
sync_dag_to_db(dag)
208+
if AIRFLOW_V_3_1_PLUS:
209+
sync_dag_to_db(dag)
209210
return dag
210211

211212

@@ -315,7 +316,10 @@ def prepare_dbt_project_dir() -> str:
315316
dbt_seed >> dbt_run >> dbt_test
316317

317318
d = generate_dag()
318-
sync_dag_to_db(d)
319+
320+
if AIRFLOW_V_3_1_PLUS:
321+
sync_dag_to_db(d)
322+
319323
return d
320324

321325

@@ -457,7 +461,8 @@ def target_connection_dag(
457461

458462
dbt_seed >> dbt_run >> dbt_test
459463

460-
sync_dag_to_db(dag)
464+
if AIRFLOW_V_3_1_PLUS:
465+
sync_dag_to_db(dag)
461466
return dag
462467

463468

@@ -536,7 +541,9 @@ def test_example_basic_dag(
536541
dbt_run.target = "test"
537542
dbt_run.profile = "default"
538543

539-
sync_dag_to_db(dag)
544+
if AIRFLOW_V_3_1_PLUS:
545+
sync_dag_to_db(dag)
546+
540547
dagrun = _create_dagrun(
541548
dag,
542549
state=DagRunState.RUNNING,
@@ -587,7 +594,9 @@ def test_example_dbt_project_in_github_dag(
587594
assert dag is not None
588595
assert len(dag.tasks) == 3
589596

590-
sync_dag_to_db(dag)
597+
if AIRFLOW_V_3_1_PLUS:
598+
sync_dag_to_db(dag)
599+
591600
dagrun = _create_dagrun(
592601
dag,
593602
state=DagRunState.RUNNING,
@@ -639,7 +648,9 @@ def test_example_complete_dbt_workflow_dag(
639648
assert dag is not None
640649
assert len(dag.tasks) == 5
641650

642-
sync_dag_to_db(dag)
651+
if AIRFLOW_V_3_1_PLUS:
652+
sync_dag_to_db(dag)
653+
643654
dagrun = _create_dagrun(
644655
dag,
645656
state=DagRunState.RUNNING,

0 commit comments

Comments
 (0)