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