def test_backoff_time(patch_base_class, requests_mock, header_name,
header_value, expected):
stream = LinnworksStream()
requests_mock.get("https://dummy",
headers={header_name: header_value},
status_code=429)
result = stream.backoff_time(requests.get("https://dummy"))
assert result == expected
def test_request_params(patch_base_class):
stream = LinnworksStream()
inputs = {
"stream_slice": None,
"stream_state": None,
"next_page_token": None
}
expected_params = {}
assert stream.request_params(**inputs) == expected_params
def test_parse_response(patch_base_class, requests_mock):
stream = LinnworksStream()
requests_mock.get(
"https://dummy",
json={
"Foo": "foo",
"Bar": {
"Baz": "baz",
},
},
)
resp = requests.get("https://dummy")
inputs = {"response": resp}
expected_parsed_object = {"Bar": {"Baz": "baz"}, "Foo": "foo"}
assert next(stream.parse_response(**inputs)) == expected_parsed_object
def test_http_method(patch_base_class):
stream = LinnworksStream()
expected_method = "POST"
assert stream.http_method == expected_method
def test_next_page_token(patch_base_class):
stream = LinnworksStream()
inputs = {"response": MagicMock()}
expected_token = None
assert stream.next_page_token(**inputs) == expected_token