X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/afddfc678f3ec8bbb39ae6a7d4ccbc4d2007416f..0c1644b480a81a3b1d55efcd4fc328f7f7ed77a6:/apps/catalogue/tests/bookmedia.py?ds=inline diff --git a/apps/catalogue/tests/bookmedia.py b/apps/catalogue/tests/bookmedia.py index 0578ecd61..da427e8b5 100644 --- a/apps/catalogue/tests/bookmedia.py +++ b/apps/catalogue/tests/bookmedia.py @@ -1,9 +1,10 @@ # -*- coding: utf-8 -*- -from os.path import basename -from django.core.files.base import ContentFile +from django.conf import settings +from os.path import basename, exists, join, dirname +from django.core.files.base import ContentFile, File from catalogue.test_utils import * -from catalogue import models +from catalogue import models, utils class BookMediaTests(WLTestCase): @@ -47,10 +48,10 @@ class BookMediaTests(WLTestCase): """ bm = models.BookMedia(book=self.book, type='ogg', - name="Tytul") + name=u"Tytul") bm.file.save(None, self.file) bm2 = models.BookMedia(book=self.book, type='ogg', - name="Tytuł") + name=u"Tytuł") bm2.file.save(None, self.file2) self.assertEqual(basename(bm.file.name), 'tytul.ogg') self.assertNotEqual(basename(bm2.file.name), 'tytul.ogg') @@ -84,3 +85,32 @@ class BookMediaTests(WLTestCase): self.assertNotEqual(basename(bm2.file.name), 'title.ogg') self.assertEqual(bm.file.read(), 'X') self.assertEqual(bm2.file.read(), 'Y') + + def test_zip_audiobooks(self): + paths = [ + (None, join(dirname(__file__), "files/fraszka-do-anusie.xml")), + (None, join(dirname(__file__), "files/fraszki.xml")), + ] + + url = utils.create_zip(paths, 'test-zip-slug') + self.assertEqual("zip/test-zip-slug.zip", url) + self.assertTrue(exists(join(settings.MEDIA_ROOT, url))) + + utils.remove_zip('test-zip-slug') + self.assertFalse(exists(join(settings.MEDIA_ROOT, url))) + + def test_remove_zip_on_media_change(self): + bm = models.BookMedia(book=self.book, type='ogg', name="Title") + bm.file.save(None, self.file) + bm.save() + + zip_url = self.book.zip_audiobooks('ogg') + self.assertEqual('zip/'+self.book.slug+'_ogg.zip', zip_url) + self.assertTrue(exists(join(settings.MEDIA_ROOT, zip_url))) + + bm2 = models.BookMedia(book=self.book, type='ogg', name="Other title") + bm2.file.save(None, self.file2) + bm2.name = "Title" + bm2.save() + # was the audiobook zip deleted? + self.assertFalse(exists(join(settings.MEDIA_ROOT, zip_url)))