def test_download_url_http_error(self, mocked_urlretrieve, mocked_error,
_):
"""Tests that download_url doesn't retry when there is an HTTP error."""
self.assertFalse(fuzz_target.download_url(self.URL, self.FILE_PATH))
mocked_error.assert_called_with('Unable to download from: %s.',
self.URL)
self.assertEqual(1, mocked_urlretrieve.call_count)
def test_download_url_connection_error(self, mocked_urlretrieve, mocked_error,
mocked_sleep):
"""Tests that download_url doesn't retry when there is an HTTP error."""
self.assertFalse(fuzz_target.download_url(self.URL, self.FILE_PATH))
self.assertEqual(3, mocked_urlretrieve.call_count)
self.assertEqual(3, mocked_sleep.call_count)
mocked_error.assert_called_with('Failed to download %s, %d times.',
self.URL, 3)
def test_download_url_no_error(self, mocked_urlretrieve, _): """Tests that download_url works when there is no error.""" self.assertTrue(fuzz_target.download_url(self.URL, self.FILE_PATH)) self.assertEqual(1, mocked_urlretrieve.call_count)
