def make_user_data(user): """Make some user data.""" u = services.get_storage_user(username=user_names[random.randint(0, 4)]) s = user.root.share(u.id, u"%s to %s" % (user.username, u.username)) u.get_share(s.id).accept() for i in range(10): d = user.root.make_subdirectory(u"%s dir%s" % (user.username, i)) for f in range(10): f = d.make_file(u"%s file%s.txt" % (user.username, f)) uj = f.make_uploadjob(f.content_hash, get_fake_hash(str(random.random())), random.randint(1, 100), random.randint(100, 1000000), random.randint(100, 1000000)) uj.commit_content(f.content_hash, uuid.uuid4(), None) services.add_music_metadata(uj.file.content_hash, TEST_MUSIC_DATA, user.shard_id) for i in range(10): user.make_musicstore_download(random.randint(100, 1000000), random.randint(1, 100), u"https://fake.com/%s/%s" % (user.id, i), u"%s" % user.username, u"%s Greatest Hits" % user.username, u"Track %s" % i, "US") for i in range(10): downloadservices.get_or_make_download( user.id, user.root_volume_id, u"https://fake.com/%s/%s" % (user.id, i), u"/a/b", u"%s%s" % (user.id, i))
def make_user_data(user): """Make some user data.""" u = services.get_storage_user(username=user_names[random.randint(0, 4)]) s = user.root.share(u.id, u"%s to %s" % (user.username, u.username)) u.get_share(s.id).accept() for i in range(10): d = user.root.make_subdirectory(u"%s dir%s" % (user.username, i)) for f in range(10): f = d.make_file(u"%s file%s.txt" % (user.username, f)) uj = f.make_uploadjob( f.content_hash, get_fake_hash(str(random.random())), random.randint(1, 100), random.randint(100, 1000000), random.randint(100, 1000000)) uj.commit_content(f.content_hash, uuid.uuid4(), None) services.add_music_metadata( uj.file.content_hash, TEST_MUSIC_DATA, user.shard_id) for i in range(10): user.make_musicstore_download( random.randint(100, 1000000), random.randint(1, 100), u"https://fake.com/%s/%s" % (user.id, i), u"%s" % user.username, u"%s Greatest Hits" % user.username, u"Track %s" % i, "US") for i in range(10): downloadservices.get_or_make_download( user.id, user.root_volume_id, u"https://fake.com/%s/%s" % (user.id, i), u"/a/b", u"%s%s" % (user.id, i))
def test_get_or_make_download(self): """Test get_or_make_download.""" dl = downloadservices.get_or_make_download( self.user.id, self.volume_id, self.fpath, self.dl_url, self.dl_key) self.assertEquals(dl.owner_id, self.user.id) self.assertEquals(dl.volume_id, self.volume_id) self.assertEquals(dl.file_path, self.fpath) self.assertEquals(dl.download_url, self.dl_url) self.assertEquals(dl.download_key, unicode(repr(self.dl_key))) # do it again, make sure we get the same one dl2 = downloadservices.get_or_make_download( self.user.id, self.volume_id, self.fpath, self.dl_url, self.dl_key) self.assertEquals(dl.id, dl2.id)
def get_or_make_it(self): """Make the download.""" return downloadservices.get_or_make_download( self.user.id, self.volume_id, self.fpath, self.dl_url, self.dl_key)