update unit tests for additional job tracker argument

Author: pnilan

unit_tests/sources/declarative/async_job/test_integration.py

@@ -87,7 +87,7 @@ def streams(self, config: Mapping[str, Any]) -> List[Stream]:
                         job_orchestrator_factory=lambda stream_slices: AsyncJobOrchestrator(
                             MockAsyncJobRepository(),
                             stream_slices,
-                            JobTracker(_NO_LIMIT),
+                            JobTracker(_NO_LIMIT, config={}),
                             self._message_repository,
                         ),
                         config={},

unit_tests/sources/declarative/async_job/test_job_orchestrator.py

@@ -137,7 +137,7 @@ def test_given_one_job_still_running_when_create_and_get_completed_partitions_th
     def test_given_timeout_when_create_and_get_completed_partitions_then_free_budget_and_raise_exception(
         self, mock_sleep: MagicMock
     ) -> None:
-        job_tracker = JobTracker(1)
+        job_tracker = JobTracker(1, config={})
         self._job_repository.start.return_value = self._job_for_a_slice
         self._job_repository.update_jobs_status.side_effect = _status_update_per_jobs(
             {self._job_for_a_slice: [AsyncJobStatus.TIMED_OUT]}
@@ -184,15 +184,15 @@ def test_when_fetch_records_then_yield_records_from_each_job(self) -> None:
     def _orchestrator(
         self, slices: List[StreamSlice], job_tracker: Optional[JobTracker] = None
     ) -> AsyncJobOrchestrator:
-        job_tracker = job_tracker if job_tracker else JobTracker(_NO_JOB_LIMIT)
+        job_tracker = job_tracker if job_tracker else JobTracker(_NO_JOB_LIMIT, config={})
         return AsyncJobOrchestrator(
             self._job_repository, slices, job_tracker, self._message_repository
         )
 
     def test_given_more_jobs_than_limit_when_create_and_get_completed_partitions_then_still_return_all_slices_and_free_job_budget(
         self,
     ) -> None:
-        job_tracker = JobTracker(1)
+        job_tracker = JobTracker(1, config={})
         self._job_repository.start.side_effect = [
             self._job_for_a_slice,
             self._job_for_another_slice,
@@ -220,7 +220,7 @@ def test_given_exception_to_break_when_start_job_and_raise_this_exception_and_ab
         orchestrator = AsyncJobOrchestrator(
             self._job_repository,
             [_A_STREAM_SLICE, _ANOTHER_STREAM_SLICE],
-            JobTracker(_NO_JOB_LIMIT),
+            JobTracker(_NO_JOB_LIMIT, config={}),
             self._message_repository,
             exceptions_to_break_on=[ValueError],
         )
@@ -241,7 +241,7 @@ def test_given_traced_config_error_when_start_job_and_raise_this_exception_and_a
         """
         Since this is a config error, we assume the other jobs will fail for the same reasons.
         """
-        job_tracker = JobTracker(1)
+        job_tracker = JobTracker(1, config={})
         self._job_repository.start.side_effect = MessageRepresentationAirbyteTracedErrors(
             "Can't create job", failure_type=FailureType.config_error
         )
@@ -318,7 +318,7 @@ def test_given_jobs_failed_more_than_max_attempts_when_create_and_get_completed_
     def given_budget_already_taken_before_start_when_create_and_get_completed_partitions_then_wait_for_budget_to_be_freed(
         self,
     ) -> None:
-        job_tracker = JobTracker(1)
+        job_tracker = JobTracker(1, config={})
         intent_to_free = job_tracker.try_to_get_intent()
 
         def wait_and_free_intent(_job_tracker: JobTracker, _intent_to_free: str) -> None:
@@ -341,7 +341,7 @@ def wait_and_free_intent(_job_tracker: JobTracker, _intent_to_free: str) -> None
     def test_given_start_job_raise_when_create_and_get_completed_partitions_then_free_budget(
         self,
     ) -> None:
-        job_tracker = JobTracker(1)
+        job_tracker = JobTracker(1, config={})
         self._job_repository.start.side_effect = ValueError("Can't create job")
 
         orchestrator = AsyncJobOrchestrator(

unit_tests/sources/declarative/async_job/test_job_tracker.py

@@ -15,7 +15,7 @@
 
 class JobTrackerTest(TestCase):
     def setUp(self) -> None:
-        self._tracker = JobTracker(_LIMIT)
+        self._tracker = JobTracker(_LIMIT, config={})
 
     def test_given_limit_reached_when_remove_job_then_can_get_intent_again(self) -> None:
         intents = self._reach_limit()

unit_tests/sources/declarative/interpolation/test_macros.py

@@ -75,13 +75,13 @@ def test_macros_export(test_name, fn_name, found_in_macros):
             "2022-01-01T01:01:01Z",
             "%s",
             "%Y-%m-%dT%H:%M:%SZ",
-            "1640998861",
+            "1641027661",
         ),
         (
             "2022-01-01T01:01:01Z",
             "%ms",
             "%Y-%m-%dT%H:%M:%SZ",
-            "1640998861000000",
+            "1641027661000000",
         ),
     ],
     ids=[

unit_tests/sources/declarative/partition_routers/test_async_job_partition_router.py

@@ -26,7 +26,7 @@ def test_stream_slices_with_single_partition_router():
         job_orchestrator_factory=lambda stream_slices: AsyncJobOrchestrator(
             MockAsyncJobRepository(),
             stream_slices,
-            JobTracker(_NO_LIMIT),
+            JobTracker(_NO_LIMIT, config={}),
             NoopMessageRepository(),
         ),
         config={},
@@ -58,7 +58,7 @@ def test_stream_slices_with_parent_slicer():
         job_orchestrator_factory=lambda stream_slices: AsyncJobOrchestrator(
             MockAsyncJobRepository(),
             stream_slices,
-            JobTracker(_NO_LIMIT),
+            JobTracker(_NO_LIMIT, config={}),
             NoopMessageRepository(),
         ),
         config={},

unit_tests/sources/declarative/test_concurrent_declarative_source.py

@@ -349,6 +349,7 @@
                     "path": "{{stream_slice['url']}}",
                     "http_method": "GET",
                 },
+                "max_concurrent_jobs": 1,
             },
             "incremental_sync": {"$ref": "#/definitions/incremental_cursor"},
             "schema_loader": {