src/catalogue/tests/test_bookmedia.py

   1 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   3 #
   4 from os.path import basename, exists
   5 from unittest import skip
   6
   7 from django.core.files.base import ContentFile
   8
   9 from catalogue.test_utils import *
  10 from catalogue import models, utils
  11
  12
  13 class BookMediaTests(WLTestCase):
  14
  15     def setUp(self):
  16         WLTestCase.setUp(self)
  17         self.file = ContentFile(b'X')
  18         self.file2 = ContentFile(b'Y')
  19         self.book = models.Book.objects.create(slug='test-book', title='Test')
  20
  21     def set_title(self, title):
  22         self.book.title = title
  23         self.book.save()
  24
  25     def test_diacritics(self):
  26         bm = models.BookMedia(book=self.book, type="ogg", name="Zażółć gęślą jaźń")
  27         self.set_title(bm.name)
  28         bm.file.save(None, self.file)
  29         self.assertEqual(basename(bm.file.name), 'zazolc-gesla-jazn.ogg')
  30
  31     def test_long_name(self):
  32         bm = models.BookMedia(
  33             book=self.book, type="ogg",
  34             name="Some very very very very very very very very very very very very very very very very long file name")
  35         self.set_title(bm.name)
  36         bm.file.save(bm.name, self.file)
  37
  38         # reload to see what was really saved
  39         bm = models.BookMedia.objects.get(pk=bm.pk)
  40         self.assertEqual(bm.file.size, 1)
  41
  42     def test_overwrite(self):
  43         """
  44             File gets overwritten with same filename on update.
  45         """
  46
  47         bm = models.BookMedia(book=self.book, type='ogg', name="Some media")
  48         self.set_title(bm.name)
  49         bm.file.save(None, self.file)
  50         bm.file.save(None, self.file2)
  51
  52         self.assertEqual(bm.file.read(), b'Y')
  53         self.assertEqual(basename(bm.file.name), 'some-media.ogg')
  54
  55     @skip('broken, but is it needed?')
  56     def test_no_clobber(self):
  57         """
  58             File save doesn't clobber some other media with similar name.
  59         """
  60
  61         bm = models.BookMedia(book=self.book, type='ogg', name="Tytul")
  62         self.set_title(bm.name)
  63         bm.file.save(None, self.file)
  64         bm2 = models.BookMedia(book=self.book, type='ogg', name="Tytuł")
  65         self.set_title(bm2.name)
  66         bm2.file.save(None, self.file2)
  67         self.assertEqual(basename(bm.file.name), 'tytul.ogg')
  68         self.assertNotEqual(basename(bm2.file.name), 'tytul.ogg')
  69         self.assertEqual(bm.file.read(), b'X')
  70         self.assertEqual(bm2.file.read(), b'Y')
  71
  72     def test_change_name(self):
  73         """
  74             File name reflects name change.
  75         """
  76
  77         bm = models.BookMedia(book=self.book, type='ogg', name="Title")
  78         self.set_title(bm.name)
  79         bm.file.save(None, self.file)
  80         self.set_title("Other Title")
  81         bm.save()
  82         self.assertEqual(basename(bm.file.name), 'other-title.ogg')
  83         self.assertEqual(bm.file.read(), b'X')
  84
  85     @skip('broken, but is it needed?')
  86     def test_change_name_no_clobber(self):
  87         """
  88             File name after change won't clobber some other file
  89             with similar name.
  90         """
  91
  92         bm = models.BookMedia(book=self.book, type='ogg', name="Title")
  93         self.set_title(bm.name)
  94         bm.file.save(None, self.file)
  95         bm2 = models.BookMedia(book=self.book, type='ogg', name="Other title")
  96         self.set_title(bm2.name)
  97         bm2.file.save(None, self.file2)
  98         self.set_title("Title")
  99         bm2.save()
 100         self.assertNotEqual(basename(bm2.file.name), 'title.ogg')
 101         self.assertEqual(bm.file.read(), b'X')
 102         self.assertEqual(bm2.file.read(), b'Y')
 103
 104     def test_zip_audiobooks(self):
 105         paths = [
 106             (None, join(dirname(__file__), "files/fraszka-do-anusie.xml")),
 107             (None, join(dirname(__file__), "files/fraszki.xml")),
 108             ]
 109
 110         url = utils.create_zip(paths, 'test-zip-slug')
 111         self.assertEqual("zip/test-zip-slug.zip", url)
 112         self.assertTrue(exists(join(settings.MEDIA_ROOT, url)))
 113
 114         utils.remove_zip('test-zip-slug')
 115         self.assertFalse(exists(join(settings.MEDIA_ROOT, url)))
 116
 117     def test_remove_zip_on_media_change(self):
 118         bm = models.BookMedia(book=self.book, type='ogg', name="Title")
 119         self.set_title(bm.name)
 120         bm.file.save(None, self.file)
 121         bm.save()
 122
 123         zip_url = self.book.zip_audiobooks('ogg')
 124         self.assertEqual('zip/'+self.book.slug+'_ogg.zip', zip_url)
 125         self.assertTrue(exists(join(settings.MEDIA_ROOT, zip_url)))
 126
 127         bm2 = models.BookMedia(book=self.book, type='ogg', name="Other title")
 128         self.set_title(bm2.name)
 129         bm2.file.save(None, self.file2)
 130         self.set_title("Title")
 131         bm2.save()
 132         # was the audiobook zip deleted?
 133         self.assertFalse(exists(join(settings.MEDIA_ROOT, zip_url)))