Skip to content

Commit

Permalink
fix test_dagrun failures after merge
Browse files Browse the repository at this point in the history
  • Loading branch information
xBis7 committed Dec 31, 2024
1 parent 11d5a6c commit 09ed2e9
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/models/test_dagrun.py
Original file line number Diff line number Diff line change
Expand Up @@ -527,7 +527,7 @@ def test_start_dr_spans_if_needed_new_span(self, session):
schedule=datetime.timedelta(days=1),
start_date=datetime.datetime(2017, 1, 1),
)
DAG.bulk_write_to_db(dags=[dag], processor_subdir="/tmp/test", session=session)
DAG.bulk_write_to_db(dags=[dag], session=session)

dag_task1 = EmptyOperator(task_id="test_task1", dag=dag)
dag_task2 = EmptyOperator(task_id="test_task2", dag=dag)
Expand Down Expand Up @@ -563,7 +563,7 @@ def test_start_dr_spans_if_needed_span_with_continuance(self, session):
schedule=datetime.timedelta(days=1),
start_date=datetime.datetime(2017, 1, 1),
)
DAG.bulk_write_to_db(dags=[dag], processor_subdir="/tmp/test", session=session)
DAG.bulk_write_to_db(dags=[dag], session=session)

dag_task1 = EmptyOperator(task_id="test_task1", dag=dag)
dag_task2 = EmptyOperator(task_id="test_task2", dag=dag)
Expand Down Expand Up @@ -608,7 +608,7 @@ def test_end_dr_span_if_needed(self, session):
schedule=datetime.timedelta(days=1),
start_date=datetime.datetime(2017, 1, 1),
)
DAG.bulk_write_to_db(dags=[dag], processor_subdir="/tmp/test", session=session)
DAG.bulk_write_to_db(dags=[dag], session=session)

dag_task1 = EmptyOperator(task_id="test_task1", dag=dag)
dag_task2 = EmptyOperator(task_id="test_task2", dag=dag)
Expand Down Expand Up @@ -648,7 +648,7 @@ def test_end_dr_span_if_needed_with_span_from_another_scheduler(self, session):
schedule=datetime.timedelta(days=1),
start_date=datetime.datetime(2017, 1, 1),
)
DAG.bulk_write_to_db(dags=[dag], processor_subdir="/tmp/test", session=session)
DAG.bulk_write_to_db(dags=[dag], session=session)

dag_task1 = EmptyOperator(task_id="test_task1", dag=dag)
dag_task2 = EmptyOperator(task_id="test_task2", dag=dag)
Expand Down

0 comments on commit 09ed2e9

Please sign in to comment.