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')