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)