Android promo

[wolnelektury.git] / apps / catalogue / tests / bookmedia.py

diff --git a/apps/catalogue/tests/bookmedia.py b/apps/catalogue/tests/bookmedia.py
index 72fb030..9ddd6aa 100644 (file)
--- a/apps/catalogue/tests/bookmedia.py
+++ b/apps/catalogue/tests/bookmedia.py
@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
 # -*- coding: utf-8 -*-

-
+from os.path import basename

 from django.core.files.base import ContentFile
 
 from catalogue.test_utils import *
 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')
     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):
 
     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):
 
     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")
                     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.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)
 
 
         # 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')