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