def test_get_batch_object_no_response():
query_param = {
"has_image": 1,
"image_license": "CC",
"page[size]": 1,
"page[number]": 1,
"date[from]": 0,
"date[to]": 1500
}
response = None
with patch.object(sm.delay_request, 'get',
return_value=response) as mock_call:
actual_response = sm._get_batch_objects(query_param=query_param)
assert mock_call.call_count == 3
assert actual_response is None
def test_get_batch_object_failure():
query_param = {
"has_image": 1,
"image_license": "CC",
"page[size]": 1,
"page[number]": 51,
"date[from]": 0,
"date[to]": 1500
}
response = _get_resource_json("response_failure.json")
r = requests.Response()
r.status_code = 400
r.json = MagicMock(return_value=response)
with patch.object(sm.delay_request, 'get', return_value=r) as mock_call:
actual_response = sm._get_batch_objects(query_param=query_param)
assert mock_call.call_count == 3
assert actual_response is None