This is an automated email from the ASF dual-hosted git repository.
ephraimanierobi pushed a commit to branch v2-7-test
in repository https://gitbox.apache.org/repos/asf/airflow.git
commit e678fe160c97c976dbec8fe0e2d6ef296a1ac751
Author: Daniel Dyląg <bi...@users.noreply.github.com>
AuthorDate: Thu Sep 14 12:03:46 2023 +0200
Fix test_retry_handling_job (#34326)
Needed to resync dag to db so that scheduler schedules it with num_runs=1
Co-authored-by: daniel.dylag <danieldylag1...@gmail.com>
(cherry picked from commit 1b26cc05ff553685c50a15f56a954e465ed43478)
---
tests/jobs/test_scheduler_job.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/tests/jobs/test_scheduler_job.py b/tests/jobs/test_scheduler_job.py
index 57a02e782a..c1366449dd 100644
--- a/tests/jobs/test_scheduler_job.py
+++ b/tests/jobs/test_scheduler_job.py
@@ -3039,7 +3039,6 @@ class TestSchedulerJob:
ti.refresh_from_db()
assert ti.state == State.SUCCESS
- @pytest.mark.skip(reason="This test needs fixing. It's very wrong now and
always fails")
def test_retry_handling_job(self):
"""
Integration test of the scheduler not accidentally resetting
@@ -3048,9 +3047,11 @@ class TestSchedulerJob:
dag = self.dagbag.get_dag("test_retry_handling_job")
dag_task1 = dag.get_task("test_retry_handling_op")
dag.clear()
+ dag.sync_to_db()
- scheduler_job = Job(jobe_type=SchedulerJobRunner.job_type, heartrate=0)
- self.job_runner = SchedulerJobRunner(job=scheduler_job,
dag_id=dag.dag_id, num_runs=1)
+ scheduler_job = Job(job_type=SchedulerJobRunner.job_type, heartrate=0)
+ self.job_runner = SchedulerJobRunner(job=scheduler_job, num_runs=1)
+ self.job_runner.processor_agent = mock.MagicMock()
run_job(scheduler_job, execute_callable=self.job_runner._execute)
session = settings.Session()
