def test_download_error(self):
"""Test download_error."""
dl = self.get_or_make_it()
downloadservices.download_error(self.user.id, dl.id, "Kaploey")
dl = downloadservices.get_download_by_id(self.user.id, dl.id)
self.assertEqual(dl.status, Download.STATUS_ERROR)
self.assertEqual(dl.error_message, "Kaploey")
def test_download_error(self):
"""Test download_error."""
dl = self.get_or_make_it()
downloadservices.download_error(self.user.id, dl.id, "Kaploey")
dl = downloadservices.get_download_by_id(self.user.id, dl.id)
self.assertEqual(dl.status, Download.STATUS_ERROR)
self.assertEqual(dl.error_message, "Kaploey")
def test_get_failed_downloads(self):
"""Test the get_failed_downloads() function."""
# Create downloads
udf1 = self.user.make_udf("~/path/name")
d1 = downloadservices.make_download(self.user.id, udf1.id, "path/file",
"http://example.com")
downloadservices.download_error(self.user.id, d1.id, "download failed")
user2 = services.make_storage_user("user2", 10 * 23)
udf2 = user2.make_udf("~/path/name")
d2 = downloadservices.make_download(user2.id, udf2.id, "path/file",
"http://example.com")
downloadservices.download_error(user2.id, d2.id, "download failed")
# Finally, a failed download that will be outside the time window.
d3 = downloadservices.make_download(user2.id, udf2.id, "path/file2",
"http://example.com")
downloadservices.download_error(user2.id, d3.id, "download failed")
result = downloadservices.get_failed_downloads(
start_date=d1.status_change_date, end_date=d3.status_change_date)
download_ids = [download.id for download in result]
self.assertTrue(d1.id in download_ids)
self.assertTrue(d2.id in download_ids)
self.assertFalse(d3.id in download_ids)
def test_get_failed_downloads(self):
"""Test the get_failed_downloads() function."""
# Create downloads
udf1 = self.user.make_udf("~/path/name")
d1 = downloadservices.make_download(
self.user.id, udf1.id, "path/file", "http://example.com")
downloadservices.download_error(
self.user.id, d1.id, "download failed")
user2 = services.make_storage_user("user2", 10 * 23)
udf2 = user2.make_udf("~/path/name")
d2 = downloadservices.make_download(
user2.id, udf2.id, "path/file", "http://example.com")
downloadservices.download_error(user2.id, d2.id, "download failed")
# Finally, a failed download that will be outside the time window.
d3 = downloadservices.make_download(
user2.id, udf2.id, "path/file2", "http://example.com")
downloadservices.download_error(user2.id, d3.id, "download failed")
result = downloadservices.get_failed_downloads(
start_date=d1.status_change_date,
end_date=d3.status_change_date)
download_ids = [download.id for download in result]
self.assertTrue(d1.id in download_ids)
self.assertTrue(d2.id in download_ids)
self.assertFalse(d3.id in download_ids)
