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)