fix test

Author: Oleksandr Bazarnov

airbyte_cdk/sources/declarative/async_job/job_orchestrator.py

@@ -382,15 +382,15 @@ def _stop_partition(self, partition: AsyncPartition) -> None:
 
     def _stop_timed_out_jobs(self, partition: AsyncPartition) -> None:
         for job in partition.jobs:
-            if job.status() == AsyncJobStatus.TIMED_OUT:
-                # we don't free allocation here because it is expected to retry the job
-                self._abort_job(job, free_job_allocation=False)
-            elif job.status() == AsyncJobStatus.FORCED_TIME_OUT:
+            if job.status() == AsyncJobStatus.FORCED_TIME_OUT:
                 self._abort_job(job, free_job_allocation=True)
                 raise AirbyteTracedException(
                     internal_message=f"Job {job.api_job_id()} has timed out. Try increasing the `polling job timeout`.",
                     failure_type=FailureType.config_error,
                 )
+            # we don't free allocation here because it is expected to retry the job
+            if job.status() == AsyncJobStatus.TIMED_OUT:
+                self._abort_job(job, free_job_allocation=False)
 
     def _abort_job(self, job: AsyncJob, free_job_allocation: bool = True) -> None:
         try:

unit_tests/sources/declarative/async_job/test_job.py

@@ -19,10 +19,10 @@ def test_given_timer_is_not_out_when_status_then_return_actual_status(self) -> N
         job = AsyncJob(_AN_API_JOB_ID, _ANY_STREAM_SLICE, _A_VERY_BIG_TIMEOUT)
         assert job.status() == AsyncJobStatus.RUNNING
 
-    def test_given_timer_is_out_when_status_then_return_timed_out(self) -> None:
+    def test_given_timer_is_out_when_status_then_return_forced_time_out(self) -> None:
         job = AsyncJob(_AN_API_JOB_ID, _ANY_STREAM_SLICE, _IMMEDIATELY_TIMED_OUT)
         time.sleep(0.001)
-        assert job.status() == AsyncJobStatus.TIMED_OUT
+        assert job.status() == AsyncJobStatus.FORCED_TIME_OUT
 
     def test_given_status_is_terminal_when_update_status_then_stop_timer(self) -> None:
         """