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