def test_download_error(self):
"""Test download_error."""
dl = self.get_or_make_it()
downloadservices.download_error(self.user.id, dl.id, u"Kaploey")
dl = downloadservices.get_download_by_id(self.user.id, dl.id)
self.assertEquals(dl.status, model.DOWNLOAD_STATUS_ERROR)
self.assertEquals(dl.error_message, u"Kaploey")
def test_get_failed_downloads(self):
"""Test the get_failed_downloads() function."""
# Create downloads in different shards.
udf1 = self.user.make_udf(u"~/path/name")
d1 = downloadservices.make_download(
self.user.id, udf1.id, u"path/file", u"http://example.com")
downloadservices.download_error(
self.user.id, d1.id, u"download failed")
user2 = self.obj_factory.make_user(
2, u"User 2", u"User 2", 10 * 23, shard_id=u"shard1")
udf2 = user2.make_udf(u"~/path/name")
d2 = downloadservices.make_download(
user2.id, udf2.id, u"path/file", u"http://example.com")
downloadservices.download_error(user2.id, d2.id, u"download failed")
# Finally, a failed download that will be outside the time window.
d3 = downloadservices.make_download(
user2.id, udf2.id, u"path/file2", u"http://example.com")
downloadservices.download_error(user2.id, d3.id, u"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)
