def test_download_http_raises_error_on_bad_status_code(self, mock_request):
def _request_callback(request, context):
context.status_code = 404
return 'Not found'
fake_url = 'https://fake_url'
mock_request.get(fake_url, json=_request_callback)
with self.assertRaises(requests.HTTPError):
utils.download_http(fake_url)
def test_download_http_no_error_on_301(self, mock_request):
def _request_callback(request, context):
context.status_code = 301
return 'foo'
fake_url = 'https://fake_url'
mock_request.get(fake_url, json=_request_callback)
try:
utils.download_http(fake_url)
except requests.HTTPError:
self.fail('Raised HTTPError for status 301')
