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 <[email protected]> (cherry picked from commit 1b26cc0)
apache · Oct 30, 2023 · 719de69 · 719de69
1 parent 7885869
commit 719de69
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/jobs/test_scheduler_job.py b/tests/jobs/test_scheduler_job.py
@@ -3039,7 +3039,6 @@ def run_with_error(ti, ignore_ti_state=False):
         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 @@ def test_retry_handling_job(self):
         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()