Skip to content

Commit fbf907c

Browse files
author
maxime.c
committed
fix test
1 parent 2ae3f57 commit fbf907c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

unit_tests/sources/declarative/retrievers/test_simple_retriever.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1488,7 +1488,7 @@ def test_given_requester_raise_pagination_reset_exception_when_read_records_than
14881488
x = list(retriever.read_records(A_RECORD_SCHEMA, A_STREAM_SLICE))
14891489

14901490
assert len(x) == 2
1491-
assert pagination_tracker.reset.call_count == 1
1491+
assert pagination_tracker.reduce_slice_range_if_possible.call_count == 1
14921492
assert requester.send_request.call_count == 3
14931493
assert requester.send_request.call_args_list[1].kwargs["stream_slice"] == A_STREAM_SLICE
14941494
assert requester.send_request.call_args_list[1].kwargs["next_page_token"] == {
@@ -1542,7 +1542,7 @@ def test_given_reach_pagination_limit_after_two_pages_when_read_records_than_red
15421542
x = list(retriever.read_records(A_RECORD_SCHEMA, A_STREAM_SLICE))
15431543

15441544
assert len(x) == 3
1545-
assert pagination_tracker.reset.call_count == 1
1545+
assert pagination_tracker.reduce_slice_range_if_possible.call_count == 1
15461546
assert requester.send_request.call_count == 3
15471547
assert requester.send_request.call_args_list[1].kwargs["stream_slice"] == A_STREAM_SLICE
15481548
assert requester.send_request.call_args_list[1].kwargs["next_page_token"] == {

0 commit comments

Comments
 (0)