def test_get_status(self): """Test get_status.""" status = downloadservices.get_status( self.user.id, self.volume_id, self.fpath, self.dl_url, self.dl_key) self.assertEquals(status, downloadservices.UNKNOWN) dl = self.get_or_make_it() status = downloadservices.get_status( self.user.id, self.volume_id, self.fpath, self.dl_url, self.dl_key) self.assertEquals(status, downloadservices.QUEUED) # go ahead and complete it and create a file mime = u'image/tif' hash = get_fake_hash() storage_key = uuid.uuid4() crc = 12345 size = deflated_size = 300 dl = self.get_or_make_it() downloadservices.download_complete( self.user.id, dl.id, hash, crc, size, deflated_size, mime, storage_key) status = downloadservices.get_status( self.user.id, self.volume_id, self.fpath, self.dl_url, self.dl_key) self.assertEquals(status, downloadservices.DOWNLOADED) #delete the file f = self.user.volume().get_node_by_path(self.fpath) f.delete() status = downloadservices.get_status( self.user.id, self.volume_id, self.fpath, self.dl_url, self.dl_key) self.assertEquals(status, downloadservices.DOWNLOADED_NOT_PRESENT)
def test_download_complete(self): """Test download_complete.""" mime = u'image/tif' hash = get_fake_hash() storage_key = uuid.uuid4() crc = 12345 size = deflated_size = 300 dl = self.get_or_make_it() downloadservices.download_complete( self.user.id, dl.id, hash, crc, size, deflated_size, mime, storage_key) dl = downloadservices.get_download_by_id(self.user.id, dl.id) self.assertEquals(dl.status, model.DOWNLOAD_STATUS_COMPLETE) f = self.user.volume().get_node_by_path( self.fpath, with_content=True) self.assertEquals(f.full_path, self.fpath) self.assertEquals(f.content_hash, hash) self.assertEquals(f.content.storage_key, storage_key) self.assertEquals(f.mimetype, mime)