Skip to content

Commit ee498b1

Browse files
committed
minor fix to test
1 parent c1435f5 commit ee498b1

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

unit_tests/sources/declarative/stream_slicers/test_stream_slicer_read_decorator.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,8 @@ def mocked_partition_router():
4545

4646
def test_show_as_wrapped_instance():
4747
first_partition = {"first_partition_key": "first_partition_value"}
48-
mocked_partition_router().stream_slices.return_value = [
48+
partition_router = mocked_partition_router()
49+
partition_router.stream_slices.return_value = [
4950
StreamSlice(
5051
partition=first_partition, cursor_slice={}, extra_fields={"extra_field": "extra_value"}
5152
),
@@ -56,7 +57,7 @@ def test_show_as_wrapped_instance():
5657
.build()
5758
)
5859

59-
global_cursor = GlobalSubstreamCursor(cursor, mocked_partition_router)
60+
global_cursor = GlobalSubstreamCursor(cursor, partition_router)
6061
wrapped_slicer = StreamSlicerTestReadDecorator(
6162
wrapped_slicer=global_cursor,
6263
maximum_number_of_slices=5,

0 commit comments

Comments
 (0)