def test_next_page_token(patch_base_class):
setup_responses()
stream = LemlistStream()
inputs = {
"response":
requests.get("https://api.lemlist.com/api/example_endpoint")
}
expected_token = None
assert stream.next_page_token(**inputs) == expected_token
def test_request_headers(patch_base_class):
stream = LemlistStream()
inputs = {
"stream_slice": None,
"stream_state": None,
"next_page_token": {
"offset": 100
}
}
assert stream.request_headers(**inputs) == {}
def test_request_params(patch_base_class):
stream = LemlistStream()
inputs = {
"stream_slice": None,
"stream_state": None,
"next_page_token": {
"offset": 100
},
}
expected_params = {"limit": stream.page_size, "offset": 100}
assert stream.request_params(**inputs) == expected_params
def test_parse_response(patch_base_class):
setup_responses()
stream = LemlistStream()
inputs = {
"response":
requests.get("https://api.lemlist.com/api/example_endpoint")
}
expected_parsed_object = {
"_id": "cam_aaWL92T22Sei3Bz6v",
"name": "Campaign1",
"labels": ["label 1", "label 2"]
}
assert next(stream.parse_response(**inputs)) == expected_parsed_object
def test_http_method(patch_base_class):
stream = LemlistStream()
expected_method = "GET"
assert stream.http_method == expected_method
def test_backoff_time(patch_base_class):
response_mock = MagicMock()
stream = LemlistStream()
expected_backoff_time = 2
assert stream.backoff_time(response_mock) == expected_backoff_time
def test_should_retry(patch_base_class, http_status, should_retry):
response_mock = MagicMock()
response_mock.status_code = http_status
stream = LemlistStream()
assert stream.should_retry(response_mock) == should_retry