fix an assertion on a test

airbytehq · Dec 13, 2024 · 588fdb7 · 588fdb7
1 parent ec98aa3
commit 588fdb7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/unit_tests/sources/declarative/async_job/test_integration.py b/unit_tests/sources/declarative/async_job/test_integration.py
@@ -38,7 +38,7 @@
 
 class MockAsyncJobRepository(AsyncJobRepository):
     def start(self, stream_slice: StreamSlice) -> AsyncJob:
-        return AsyncJob("a_job_id", StreamSlice(partition={}, cursor_slice={}))
+        return AsyncJob("a_job_id", stream_slice)
 
     def update_jobs_status(self, jobs: Set[AsyncJob]) -> None:
         for job in jobs:

diff --git a/unit_tests/sources/declarative/partition_routers/test_async_job_partition_router.py b/unit_tests/sources/declarative/partition_routers/test_async_job_partition_router.py
@@ -78,5 +78,7 @@ def test_stream_slices_with_parent_slicer():
         attempts_per_job = list(partition.jobs)
         assert len(attempts_per_job) == 1
         assert attempts_per_job[0].api_job_id() == "a_job_id"
-        assert attempts_per_job[0].job_parameters() == StreamSlice(partition={}, cursor_slice={})
+        assert attempts_per_job[0].job_parameters() == StreamSlice(
+            partition={"parent_id": str(i)}, cursor_slice={}
+        )
         assert attempts_per_job[0].status() == AsyncJobStatus.COMPLETED