X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/fe4a9a390a93aed8b2380f4864c25f05d7f317ff..0098b333f963f74628c4fb0768b6dd56d9c6a07b:/apps/catalogue/tests/bookmedia.py?ds=inline diff --git a/apps/catalogue/tests/bookmedia.py b/apps/catalogue/tests/bookmedia.py index 72fb03067..9ddd6aaa0 100644 --- a/apps/catalogue/tests/bookmedia.py +++ b/apps/catalogue/tests/bookmedia.py @@ -1,5 +1,5 @@ # -*- coding: utf-8 -*- - +from os.path import basename from django.core.files.base import ContentFile from catalogue.test_utils import * @@ -10,24 +10,77 @@ class BookMediaTests(WLTestCase): def setUp(self): WLTestCase.setUp(self) self.file = ContentFile('X') + self.file2 = ContentFile('Y') + self.book = models.Book.objects.create(slug='test-book') def test_diacritics(self): - bm = models.BookMedia.objects.create(type="ogg", - name="Zażółć gęślą jaźń") - bm.file.save(bm.name, self.file) - self.assertEqual(bm.file.name.rsplit('/', 1)[-1], 'zazolc-gesla-jazn.ogg') - + bm = models.BookMedia(book=self.book, type="ogg", + name=u"Zażółć gęślą jaźń") + bm.file.save(None, self.file) + self.assertEqual(basename(bm.file.name), 'zazolc-gesla-jazn.ogg') def test_long_name(self): - bm = models.BookMedia.objects.create(type="ogg", + bm = models.BookMedia(book=self.book, type="ogg", name="Some very very very very very very very very very very very very very very very very long file name") - - # save twice so Django adds some stuff bm.file.save(bm.name, self.file) - bm.file.save(bm.name, self.file) - bm.save() # reload to see what was really saved bm = models.BookMedia.objects.get(pk=bm.pk) self.assertEqual(bm.file.size, 1) + def test_overwrite(self): + """ + File gets overwritten with same filename on update. + """ + + bm = models.BookMedia(book=self.book, type='ogg', + name="Some media") + bm.file.save(None, self.file) + bm.file.save(None, self.file2) + + self.assertEqual(bm.file.read(), 'Y') + self.assertEqual(basename(bm.file.name), 'some-media.ogg') + + def test_no_clobber(self): + """ + File save doesn't clobber some other media with similar name. + """ + + bm = models.BookMedia(book=self.book, type='ogg', + name=u"Tytul") + bm.file.save(None, self.file) + bm2 = models.BookMedia(book=self.book, type='ogg', + 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') + self.assertEqual(bm.file.read(), 'X') + self.assertEqual(bm2.file.read(), 'Y') + + def test_change_name(self): + """ + File name reflects name change. + """ + + bm = models.BookMedia(book=self.book, type='ogg', name="Title") + bm.file.save(None, self.file) + bm.name = "Other Title" + bm.save() + self.assertEqual(basename(bm.file.name), 'other-title.ogg') + self.assertEqual(bm.file.read(), 'X') + + def test_change_name_no_clobber(self): + """ + File name after change won't clobber some other file + with similar name. + """ + + bm = models.BookMedia(book=self.book, type='ogg', name="Title") + bm.file.save(None, self.file) + bm2 = models.BookMedia(book=self.book, type='ogg', name="Other title") + bm2.file.save(None, self.file2) + bm2.name = "Title" + bm2.save() + self.assertNotEqual(basename(bm2.file.name), 'title.ogg') + self.assertEqual(bm.file.read(), 'X') + self.assertEqual(bm2.file.read(), 'Y')