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',
}])
