Skip to content

Commit e3df407

Browse files
grubberrjhammarstedt
authored andcommitted
requester = MagicMock(use_cache=False) (airbytehq#18289)
Signed-off-by: Sergey Chvalyuk <[email protected]>
1 parent 3b6a08f commit e3df407

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

airbyte-cdk/python/unit_tests/sources/declarative/retrievers/test_simple_retriever.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ def test_simple_retriever_full(mock_http_stream):
115115
],
116116
)
117117
def test_should_retry(test_name, requester_response, expected_should_retry, expected_backoff_time):
118-
requester = MagicMock()
118+
requester = MagicMock(use_cache=False)
119119
retriever = SimpleRetriever(name="stream_name", primary_key=primary_key, requester=requester, record_selector=MagicMock(), options={})
120120
requester.should_retry.return_value = requester_response
121121
assert retriever.should_retry(requests.Response()) == expected_should_retry
@@ -132,7 +132,7 @@ def test_should_retry(test_name, requester_response, expected_should_retry, expe
132132
],
133133
)
134134
def test_parse_response(test_name, status_code, response_status, len_expected_records):
135-
requester = MagicMock()
135+
requester = MagicMock(use_cache=False)
136136
record_selector = MagicMock()
137137
record_selector.select_records.return_value = [{"id": 100}]
138138
retriever = SimpleRetriever(
@@ -162,7 +162,7 @@ def test_parse_response(test_name, status_code, response_status, len_expected_re
162162
],
163163
)
164164
def test_backoff_time(test_name, response_action, retry_in, expected_backoff_time):
165-
requester = MagicMock()
165+
requester = MagicMock(use_cache=False)
166166
record_selector = MagicMock()
167167
record_selector.select_records.return_value = [{"id": 100}]
168168
response = requests.Response()
@@ -205,7 +205,7 @@ def test_get_request_options_from_pagination(test_name, paginator_mapping, strea
205205
stream_slicer.get_request_body_json.return_value = stream_slicer_mapping
206206

207207
base_mapping = {"key": "value"}
208-
requester = MagicMock()
208+
requester = MagicMock(use_cache=False)
209209
requester.get_request_params.return_value = base_mapping
210210
requester.get_request_body_data.return_value = base_mapping
211211
requester.get_request_body_json.return_value = base_mapping
@@ -251,7 +251,7 @@ def test_get_request_headers(test_name, paginator_mapping, expected_mapping):
251251
# This test is separate from the other request options because request headers must be strings
252252
paginator = MagicMock()
253253
paginator.get_request_headers.return_value = paginator_mapping
254-
requester = MagicMock()
254+
requester = MagicMock(use_cache=False)
255255

256256
base_mapping = {"key": "value"}
257257
requester.get_request_headers.return_value = base_mapping
@@ -290,7 +290,7 @@ def test_get_request_headers(test_name, paginator_mapping, expected_mapping):
290290
def test_request_body_data(test_name, requester_body_data, paginator_body_data, expected_body_data):
291291
paginator = MagicMock()
292292
paginator.get_request_body_data.return_value = paginator_body_data
293-
requester = MagicMock()
293+
requester = MagicMock(use_cache=False)
294294

295295
requester.get_request_body_data.return_value = requester_body_data
296296

@@ -325,7 +325,7 @@ def test_request_body_data(test_name, requester_body_data, paginator_body_data,
325325
def test_path(test_name, requester_path, paginator_path, expected_path):
326326
paginator = MagicMock()
327327
paginator.path.return_value = paginator_path
328-
requester = MagicMock()
328+
requester = MagicMock(use_cache=False)
329329

330330
requester.get_path.return_value = requester_path
331331

0 commit comments

Comments
 (0)