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)
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())