def test_get_image_list_retries_with_none_response():
with patch.object(
flickr.delayed_requester,
'get',
return_value=None
) as mock_get:
flickr._get_image_list('1234', '5678', 'test', 4, retries=2)
assert mock_get.call_count == 3
def test_get_image_list_retries_with_non_ok_response():
response_json = _get_resource_json('flickr_example_pretty.json')
r = requests.Response()
r.status_code = 504
r.json = MagicMock(return_value=response_json)
with patch.object(flickr.delayed_requester, 'get',
return_value=r) as mock_get:
flickr._get_image_list('1234', '5678', 'test', 4, max_tries=3)
assert mock_get.call_count == 3
def test_get_image_list_with_partial_response():
response_json = _get_resource_json('total_pages_but_no_image_list.json')
r = requests.Response()
r.status_code = 200
r.json = MagicMock(return_value=response_json)
with patch.object(flickr.delayed_requester, 'get',
return_value=r) as mock_get:
image_list, total_pages = flickr._get_image_list('1234',
'5678',
'test',
4,
max_tries=3)
assert mock_get.call_count == 3
def test_get_image_list_with_realistic_response():
response_json = _get_resource_json('flickr_example_pretty.json')
r = requests.Response()
r.status_code = 200
r.json = MagicMock(return_value=response_json)
with patch.object(flickr.delayed_requester, 'get',
return_value=r) as mock_get:
image_list, total_pages = flickr._get_image_list('1234',
'5678',
'test',
4,
max_tries=3)
expect_image_list = _get_resource_json('flickr_example_photo_list.json')
assert mock_get.call_count == 1
assert image_list == expect_image_list
assert total_pages == 1
