Skip to content

Commit b84e68a

Browse files
committed
Fix record filter unit tests
1 parent dfcf17f commit b84e68a

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

unit_tests/sources/declarative/extractors/test_record_filter.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -290,8 +290,7 @@ def test_client_side_record_filter_decorator_no_parent_stream(
290290
config={},
291291
condition=record_filter_expression,
292292
parameters={},
293-
date_time_based_cursor=date_time_based_cursor,
294-
substream_cursor=None,
293+
cursor=date_time_based_cursor,
295294
)
296295

297296
filtered_records = list(
@@ -429,8 +428,7 @@ def date_time_based_cursor_factory() -> DatetimeBasedCursor:
429428
record_filter_decorator = ClientSideIncrementalRecordFilterDecorator(
430429
config={},
431430
parameters={},
432-
date_time_based_cursor=date_time_based_cursor,
433-
substream_cursor=substream_cursor,
431+
cursor=substream_cursor or date_time_based_cursor,
434432
)
435433

436434
# The partition we're testing

0 commit comments

Comments
 (0)