@@ -28,28 +28,28 @@ def create_response(headers):
28
28
def test_next_page_token_has_next (applications_stream ):
29
29
headers = {"link" : '<https://harvest.greenhouse.io/v1/applications?per_page=100&since_id=123456789>; rel="next"' }
30
30
response = create_response (headers )
31
- next_page_token = applications_stream .retriever .next_page_token (response = response )
31
+ next_page_token = applications_stream .retriever ._next_page_token (response = response )
32
32
assert next_page_token == {"next_page_token" : "https://harvest.greenhouse.io/v1/applications?per_page=100&since_id=123456789" }
33
33
34
34
35
35
def test_next_page_token_has_not_next (applications_stream ):
36
36
response = create_response ({})
37
- next_page_token = applications_stream .retriever .next_page_token (response = response )
37
+ next_page_token = applications_stream .retriever ._next_page_token (response = response )
38
38
39
39
assert next_page_token is None
40
40
41
41
42
42
def test_request_params_next_page_token_is_not_none (applications_stream ):
43
43
response = create_response ({"link" : f'<https://harvest.greenhouse.io/v1/applications?per_page={ 100 } &since_id=123456789>; rel="next"' })
44
- next_page_token = applications_stream .retriever .next_page_token (response = response )
45
- request_params = applications_stream .retriever .request_params (next_page_token = next_page_token , stream_state = {})
46
- path = applications_stream .retriever .path ( next_page_token = next_page_token , stream_state = {} )
44
+ next_page_token = applications_stream .retriever ._next_page_token (response = response )
45
+ request_params = applications_stream .retriever ._request_params (next_page_token = next_page_token , stream_state = {})
46
+ path = applications_stream .retriever ._paginator_path ( )
47
47
assert "applications?per_page=100&since_id=123456789" == path
48
48
assert request_params == {"per_page" : 100 }
49
49
50
50
51
51
def test_request_params_next_page_token_is_none (applications_stream ):
52
- request_params = applications_stream .retriever .request_params (stream_state = {})
52
+ request_params = applications_stream .retriever ._request_params (stream_state = {})
53
53
54
54
assert request_params == {"per_page" : 100 }
55
55
@@ -138,7 +138,7 @@ def test_parse_response_expected_response(applications_stream):
138
138
]
139
139
"""
140
140
response ._content = response_content
141
- parsed_response = applications_stream .retriever .parse_response (response , stream_state = {})
141
+ parsed_response = applications_stream .retriever ._parse_response (response , stream_state = {})
142
142
records = [dict (record ) for record in parsed_response ]
143
143
144
144
assert records == json .loads (response_content )
@@ -148,7 +148,7 @@ def test_parse_response_empty_content(applications_stream):
148
148
response = requests .Response ()
149
149
response .status_code = 200
150
150
response ._content = b"[]"
151
- parsed_response = applications_stream .retriever .parse_response (response , stream_state = {})
151
+ parsed_response = applications_stream .retriever ._parse_response (response , stream_state = {})
152
152
records = [record for record in parsed_response ]
153
153
154
154
assert records == []
@@ -164,7 +164,7 @@ def test_ignore_403(applications_stream):
164
164
response = requests .Response ()
165
165
response .status_code = 403
166
166
response ._content = b""
167
- parsed_response = applications_stream .retriever .parse_response (response , stream_state = {})
167
+ parsed_response = applications_stream .retriever ._parse_response (response , stream_state = {})
168
168
records = [record for record in parsed_response ]
169
169
assert records == []
170
170
@@ -173,5 +173,5 @@ def test_retry_429(applications_stream):
173
173
response = requests .Response ()
174
174
response .status_code = 429
175
175
response ._content = b"{}"
176
- should_retry = applications_stream .retriever .should_retry (response )
176
+ should_retry = applications_stream .retriever .requester . _should_retry (response )
177
177
assert should_retry is True
0 commit comments