minor template fix
[wolnelektury.git] / apps / catalogue / tests / book_import.py
1 # -*- coding: utf-8 -*-
2 from django.core.files.base import ContentFile
3 from catalogue.test_utils import *
4 from catalogue import models
5
6 from nose.tools import raises
7
8
9 class BookImportLogicTests(WLTestCase):
10
11     def setUp(self):
12         WLTestCase.setUp(self)
13         self.book_info = BookInfoStub(
14             url=u"http://wolnelektury.pl/example/default-book",
15             about=u"http://wolnelektury.pl/example/URI/default_book",
16             title=u"Default Book",
17             author=PersonStub(("Jim",), "Lazy"),
18             kind="X-Kind",
19             genre="X-Genre",
20             epoch="X-Epoch",
21         )
22
23         self.expected_tags = [
24            ('author', 'jim-lazy'),
25            ('genre', 'x-genre'),
26            ('epoch', 'x-epoch'),
27            ('kind', 'x-kind'),
28         ]
29         self.expected_tags.sort()
30
31     def test_empty_book(self):
32         BOOK_TEXT = "<utwor />"
33         book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info)
34
35         self.assertEqual(book.title, "Default Book")
36         self.assertEqual(book.slug, "default-book")
37         self.assert_(book.parent is None)
38         self.assertFalse(book.has_html_file())
39
40         # no fragments generated
41         self.assertEqual(book.fragments.count(), 0)
42
43         # TODO: this should be filled out probably...
44         self.assertEqual(book.wiki_link, '')
45         self.assertEqual(book.gazeta_link, '')
46         self.assertEqual(book._short_html, '')
47         self.assertEqual(book.description, '')
48
49         tags = [ (tag.category, tag.slug) for tag in book.tags ]
50         tags.sort()
51
52         self.assertEqual(tags, self.expected_tags)
53
54     def test_not_quite_empty_book(self):
55         """ Not empty, but without any real text.
56
57         Should work like any other non-empty book.
58         """
59
60         BOOK_TEXT = """<utwor>
61         <liryka_l>
62             <nazwa_utworu>Nic</nazwa_utworu>
63         </liryka_l></utwor>
64         """
65
66         book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info)
67         self.assertTrue(book.has_html_file())
68
69     def test_book_with_fragment(self):
70         BOOK_TEXT = """<utwor>
71         <opowiadanie>
72             <akap><begin id="m01" /><motyw id="m01">Love</motyw>Ala ma kota<end id="m01" /></akap>
73         </opowiadanie></utwor>
74         """
75
76         book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info)
77         self.assertTrue(book.has_html_file())
78
79         self.assertEqual(book.fragments.count(), 1)
80         self.assertEqual(book.fragments.all()[0].text, u'<p class="paragraph">Ala ma kota</p>\n')
81
82         self.assert_(('theme', 'love') in [ (tag.category, tag.slug) for tag in book.fragments.all()[0].tags ])
83
84     def test_book_with_empty_theme(self):
85         """ empty themes should be ignored """
86
87         BOOK_TEXT = """<utwor>
88         <opowiadanie>
89             <akap><begin id="m01" /><motyw id="m01"> , Love , , </motyw>Ala ma kota<end id="m01" /></akap>
90         </opowiadanie></utwor>
91         """
92
93         book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info)
94         self.assert_([('theme', 'love')],
95                          [ (tag.category, tag.slug) for tag in book.fragments.all()[0].tags.filter(category='theme') ])
96
97     def test_book_with_no_theme(self):
98         """ fragments with no themes shouldn't be created at all """
99
100         BOOK_TEXT = """<utwor>
101         <opowiadanie>
102             <akap><begin id="m01" /><motyw id="m01"></motyw>Ala ma kota<end id="m01" /></akap>
103         </opowiadanie></utwor>
104         """
105
106         book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info)
107         self.assertEqual(book.fragments.count(), 0)
108         self.assertEqual(book.tags.filter(category='theme').count(), 0)
109
110     @raises(ValueError)
111     def test_book_with_invalid_slug(self):
112         """ Book with invalid characters in slug shouldn't be imported """
113         self.book_info.url = "http://wolnelektury.pl/example/default_book"
114         BOOK_TEXT = "<utwor />"
115         book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info)
116
117     def test_book_replace_title(self):
118         BOOK_TEXT = """<utwor />"""
119         book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info)
120         self.book_info.title = u"Extraordinary"
121         book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info, overwrite=True)
122
123         tags = [ (tag.category, tag.slug) for tag in book.tags ]
124         tags.sort()
125
126         self.assertEqual(tags, self.expected_tags)
127
128     def test_book_replace_author(self):
129         BOOK_TEXT = """<utwor />"""
130         book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info)
131         self.book_info.author = PersonStub(("Hans", "Christian"), "Andersen")
132         book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info, overwrite=True)
133
134         tags = [ (tag.category, tag.slug) for tag in book.tags ]
135         tags.sort()
136
137         self.expected_tags.remove(('author', 'jim-lazy'))
138         self.expected_tags.append(('author', 'hans-christian-andersen'))
139         self.expected_tags.sort()
140
141         self.assertEqual(tags, self.expected_tags)
142
143         # the old tag shouldn't disappear
144         models.Tag.objects.get(slug="jim-lazy", category="author")
145
146     def test_multiple_tags(self):
147         BOOK_TEXT = """<utwor />"""
148         self.book_info.authors = self.book_info.author, PersonStub(("Joe",), "Dilligent"),
149         self.book_info.kinds = self.book_info.kind, 'Y-Kind',
150         self.book_info.genres = self.book_info.genre, 'Y-Genre',
151         self.book_info.epochs = self.book_info.epoch, 'Y-Epoch',
152
153         self.expected_tags.extend([
154            ('author', 'joe-dilligent'),
155            ('genre', 'y-genre'),
156            ('epoch', 'y-epoch'),
157            ('kind', 'y-kind'),
158         ])
159         self.expected_tags.sort()
160
161         book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info)
162         tags = [ (tag.category, tag.slug) for tag in book.tags ]
163         tags.sort()
164
165         self.assertEqual(tags, self.expected_tags)
166
167
168 class ChildImportTests(WLTestCase):
169
170     def setUp(self):
171         WLTestCase.setUp(self)
172         self.child_info = BookInfoStub(
173             genre='X-Genre',
174             epoch='X-Epoch',
175             kind='X-Kind',
176             author=PersonStub(("Joe",), "Doe"),
177             **info_args("Child")
178         )
179
180         self.parent_info = BookInfoStub(
181             genre='X-Genre',
182             epoch='X-Epoch',
183             kind='X-Kind',
184             author=PersonStub(("Jim",), "Lazy"),
185             parts=[self.child_info.url],
186             **info_args("Parent")
187         )
188
189     def test_child_replace(self):
190         PARENT_TEXT = """<utwor />"""
191         CHILD_TEXT = """<utwor>
192         <opowiadanie>
193             <akap><begin id="m01" /><motyw id="m01">Pies</motyw>Ala ma kota<end id="m01" /></akap>
194         </opowiadanie></utwor>
195         """
196         child = models.Book.from_text_and_meta(ContentFile(CHILD_TEXT), self.child_info)
197         parent = models.Book.from_text_and_meta(ContentFile(PARENT_TEXT), self.parent_info)
198         CHILD_TEXT = """<utwor>
199         <opowiadanie>
200             <akap><begin id="m01" /><motyw id="m01">Kot</motyw>Ala ma kota<end id="m01" /></akap>
201         </opowiadanie></utwor>
202         """
203         child = models.Book.from_text_and_meta(ContentFile(CHILD_TEXT), self.child_info, overwrite=True)
204
205         themes = self.client.get(parent.get_absolute_url()).context['book_themes']
206
207         self.assertEqual(['Kot'], [tag.name for tag in themes],
208                         'wrong related theme list')