def setUp(self): self.media = models.Media(title="Test media", duration=100) self.media.file.save('example.mp4', ContentFile("A amazing example music video")) self.extensionless_media = models.Media(title="Test media", duration=101) self.extensionless_media.file.save( 'example', ContentFile("A boring example music video"))
def setUp(self): add_media_permission = Permission.objects.get( content_type__app_label="wagtailmedia", codename="add_media") admin_permission = Permission.objects.get( content_type__app_label="wagtailadmin", codename="access_admin") self.root_collection = Collection.get_first_root_node() self.evil_plans_collection = self.root_collection.add_child( name="Evil plans") conspirators_group = Group.objects.create(name="Evil conspirators") conspirators_group.permissions.add(admin_permission) GroupCollectionPermission.objects.create( group=conspirators_group, collection=self.evil_plans_collection, permission=add_media_permission, ) user = get_user_model().objects.create_user( username="******", email="*****@*****.**", password="******") user.groups.add(conspirators_group) fake_file = ContentFile(b("A boring song"), name="test-song.mp3") media = models.Media( title="Test", duration=100, file=fake_file, type="audio", collection=self.root_collection, ) media.save()
def setUpTestData(cls): fake_audio = ContentFile(b("A boring example song")) fake_audio.name = "song.mp3" audio = models.Media(title="Test audio", duration=100, file=fake_audio, type="audio") audio.save() fake_video = ContentFile(b("An exciting video")) fake_video.name = "video.mp4" video = models.Media(title="Test video", duration=100, file=fake_video, type="video") video.save()
def make_media(): fake_file = ContentFile(b("A boring example song"), name="song.mp3") for i in range(50): media = models.Media(title="Test " + str(i), duration=100 + i, file=fake_file, type="audio") media.save()
def test_sources_unknown_type(self): fake_file = ContentFile(b("A boring example movie")) fake_file.name = 'movie' media = models.Media() media.file = File(fake_file) self.assertEqual(media.sources, [{ 'src': '/media/movie', 'type': 'application/octet-stream', }])
def test_sources_mp4_type(self): fake_file = ContentFile(b("A boring example movie")) fake_file.name = 'movie.mp4' media = models.Media() media.file = File(fake_file) self.assertEqual(media.sources, [{ 'src': '/media/movie.mp4', 'type': 'video/mp4', }])