class TestZips(TestModelsBase):
def setUp(self):
super(TestZips, self).setUp()
self.zip = join(abspath(dirname(__file__)), "media", "test.zip")
self.g = Gallery(name="test gallery", owner=self.user)
self.g.save()
def test_creating_from_zip(self):
self.g.from_zip(self.zip, initial=True)
all_gallery_media = GalleryMedia.objects.all()
self.assertEqual(5, len(all_gallery_media))
for gallery_media in all_gallery_media:
self.assertEqual(self.g, gallery_media.gallery)
self.cleanup_files.append(gallery_media.media.media.name)
def test_adding_from_zip(self):
self.g.from_zip(self.zip)
all_media = self.g.media.all()
self.assertEqual(5, len(all_media))
self.assertFalse(self.g.cover)
for m in all_media:
self.cleanup_files.append(m.media.name)
def setUp(self):
super(TestAddingMedia, self).setUp()
self.g = Gallery(name="test gallery", owner=self.user)
self.g.save()
class TestAddingMedia(TestModelsBase):
def setUp(self):
super(TestAddingMedia, self).setUp()
self.g = Gallery(name="test gallery", owner=self.user)
self.g.save()
def test_adding_media(self):
image = MediaUpload(caption="", media=self.test_file, creator=self.user)
image.save()
self.cleanup_files.append(image.media.name)
self.g.add_media(image)
self.assertTrue(image in self.g.media.all())
def test_adding_multiple_media(self):
im1 = MediaUpload(caption="", media=self.test_file, creator=self.user)
im1.save()
im2 = MediaUpload(caption="", media=self.test_file, creator=self.user)
im2.save()
im3 = MediaUpload(caption="", media=self.test_file, creator=self.user)
im3.save()
self.cleanup_files.extend([im.media.name for im in [im1, im2, im3]])
self.g.add_media(im1)
self.g.add_media(im2)
self.g.add_media(im3)
all_media = self.g.media.all()
self.assertTrue(im1 in all_media)
self.assertTrue(im2 in all_media)
self.assertTrue(im3 in all_media)
def test_remove_media(self):
im = MediaUpload(caption="", media=self.test_file, creator=self.user)
im.save()
self.cleanup_files.append(im.media.name)
self.g.add_media(im)
self.assertTrue(im in self.g.media.all())
self.g.remove_media(im)
self.assertTrue(im not in self.g.media.all())
def setUp(self):
super(TestZips, self).setUp()
self.zip = join(abspath(dirname(__file__)), "media", "test.zip")
self.g = Gallery(name="test gallery", owner=self.user)
self.g.save()
