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.
5 from django.core.files.base import ContentFile
6 from django.test import Client
7 from catalogue import models
8 from catalogue.test_utils import *
11 class BooksByTagTests(WLTestCase):
12 """ tests the /katalog/category/tag page for found books """
15 WLTestCase.setUp(self)
16 author = PersonStub(("Common",), "Man")
19 self.gchild_info = BookInfoStub(genre='Genre', epoch='Epoch', kind='Kind', author=author,
20 **info_args("GChild"))
22 self.child_info = BookInfoStub(genre='Genre', epoch='Epoch', kind='Other Kind', author=author,
23 parts=[self.gchild_info.url],
26 self.parent_info = BookInfoStub(genre='Genre', epoch='Epoch', kind='Kind', author=author,
27 parts=[self.child_info.url],
28 **info_args("Parent"))
30 self.book_file = ContentFile('<utwor />')
32 def test_nonexistent_tag(self):
33 """ Looking for a non-existent tag should yield 404 """
34 self.assertEqual(404, self.client.get('/katalog/autor/czeslaw-milosz/').status_code)
36 def test_book_tag(self):
37 """ Looking for a book tag isn't permitted """
38 models.Book.from_text_and_meta(self.book_file, self.gchild_info)
39 self.assertEqual(404, self.client.get('/katalog/gchild/').status_code)
41 def test_tag_empty(self):
42 """ Tag with no books should return no books """
43 models.Book.from_text_and_meta(self.book_file, self.gchild_info)
44 models.Tag.objects.create(name='Empty tag', slug='empty', category='author')
46 context = self.client.get('/katalog/autor/empty/').context
47 self.assertEqual(0, len(context['object_list']))
49 def test_tag_eliminate(self):
50 """ Filtering by tag should only yield top-level qualifying books. """
51 for info in self.gchild_info, self.child_info, self.parent_info:
52 models.Book.from_text_and_meta(self.book_file, info)
55 context = self.client.get('/katalog/gatunek/genre/').context
56 self.assertEqual([book.title for book in context['object_list']],
59 # parent and gchild qualify, child doesn't
60 context = self.client.get('/katalog/rodzaj/kind/').context
61 self.assertEqual([book.title for book in context['object_list']],
64 # Filtering by child's tag should yield the child
65 context = self.client.get('/katalog/rodzaj/other-kind/').context
66 self.assertEqual([book.title for book in context['object_list']],
70 class TagRelatedTagsTests(WLTestCase):
71 """ tests the /katalog/category/tag/ page for related tags """
74 WLTestCase.setUp(self)
75 self.client = Client()
76 author = PersonStub(("Common",), "Man")
78 gchild_info = BookInfoStub(author=author, genre="GchildGenre", epoch='Epoch', kind="Kind",
79 **info_args(u"GChild"))
80 child1_info = BookInfoStub(author=author, genre="ChildGenre", epoch='Epoch', kind="ChildKind",
81 parts=[gchild_info.url],
82 **info_args(u"Child1"))
83 child2_info = BookInfoStub(author=author, genre="ChildGenre", epoch='Epoch', kind="ChildKind",
84 **info_args(u"Child2"))
85 parent_info = BookInfoStub(author=author, genre="Genre", epoch='Epoch', kind="Kind",
86 parts=[child1_info.url, child2_info.url],
87 **info_args(u"Parent"))
89 for info in gchild_info, child1_info, child2_info, parent_info:
90 book_text = """<utwor><opowiadanie><akap>
92 <motyw id="m01">Theme, %sTheme</motyw>
95 </akap></opowiadanie></utwor>
96 """ % info.title.encode('utf-8')
97 book = models.Book.from_text_and_meta(ContentFile(book_text), info)
100 tag_empty = models.Tag(name='Empty tag', slug='empty', category='author')
103 def test_empty(self):
104 """ empty tag should have no related tags """
106 cats = self.client.get('/katalog/autor/empty/').context['categories']
107 self.assertEqual({k: v for (k, v) in cats.items() if v}, {},
108 'tags related to empty tag')
110 def test_has_related(self):
111 """ related own and descendants' tags should be generated """
113 cats = self.client.get('/katalog/rodzaj/kind/').context['categories']
114 self.assertTrue('Common Man' in [tag.name for tag in cats['author']],
115 'missing `author` related tag')
116 self.assertTrue('Epoch' in [tag.name for tag in cats['epoch']],
117 'missing `epoch` related tag')
118 self.assertFalse(cats.get("kind", False),
119 "There should be no child-only related `kind` tags")
120 self.assertTrue("Genre" in [tag.name for tag in cats['genre']],
121 'missing `genre` related tag')
122 self.assertFalse("ChildGenre" in [tag.name for tag in cats['genre']],
123 "There should be no child-only related `genre` tags")
124 self.assertTrue("GchildGenre" in [tag.name for tag in cats['genre']],
125 "missing grandchild's related tag")
126 self.assertTrue('Theme' in [tag.name for tag in cats['theme']],
127 "missing related theme")
128 self.assertFalse('Child1Theme' in [tag.name for tag in cats['theme']],
129 "There should be no child-only related `theme` tags")
130 self.assertTrue('GChildTheme' in [tag.name for tag in cats['theme']],
131 "missing grandchild's related theme")
133 def test_related_differ(self):
134 """ related tags shouldn't include filtering tags """
136 response = self.client.get('/katalog/rodzaj/kind/')
137 cats = response.context['categories']
138 self.assertFalse(cats.get('kind', False),
139 'filtering tag wrongly included in related')
140 cats = self.client.get('/katalog/motyw/theme/').context['categories']
141 self.assertFalse('Theme' in [tag.name for tag in cats['theme']],
142 'filtering theme wrongly included in related')
144 def test_parent_tag_once(self):
145 """ if parent and descendants have a common tag, count it only once """
147 cats = self.client.get('/katalog/rodzaj/kind/').context['categories']
148 self.assertEqual([(tag.name, tag.count) for tag in cats['epoch']],
150 'wrong related tag epoch tag on tag page')
153 def test_siblings_tags_count(self):
154 """ if children have tags and parent hasn't, count the children """
156 cats = self.client.get('/katalog/epoka/epoch/').context['categories']
157 self.assertTrue(('ChildKind', 2) in [(tag.name, tag.count) for tag in cats['kind']],
158 'wrong related kind tags on tag page, got: ' +
159 unicode([(tag.name, tag.count) for tag in cats['kind']]))
161 # all occurencies of theme should be counted
162 self.assertTrue(('Theme', 4) in [(tag.name, tag.count) for tag in cats['theme']],
163 'wrong related theme count')
165 def test_query_child_tag(self):
167 If child and parent have a common tag, but parent isn't included
168 in the result, child should still count.
170 cats = self.client.get('/katalog/gatunek/childgenre/').context['categories']
171 self.assertTrue(('Epoch', 2) in [(tag.name, tag.count) for tag in cats['epoch']],
172 'wrong related kind tags on tag page, got: ' +
173 unicode([(tag.name, tag.count) for tag in cats['epoch']]))
176 class CleanTagRelationTests(WLTestCase):
177 """ tests for tag relations cleaning after deleting things """
180 WLTestCase.setUp(self)
181 author = PersonStub(("Common",), "Man")
183 book_info = BookInfoStub(author=author, genre="G", epoch='E', kind="K",
184 **info_args(u"Book"))
185 book_text = """<utwor><opowiadanie><akap>
186 <begin id="m01" /><motyw id="m01">Theme</motyw>Ala ma kota
188 </akap></opowiadanie></utwor>
190 self.book = models.Book.from_text_and_meta(ContentFile(book_text), book_info)
192 def test_delete_objects(self):
193 """ there should be no related tags left after deleting some objects """
195 models.Book.objects.all().delete()
196 cats = self.client.get('/katalog/rodzaj/k/').context['categories']
197 self.assertEqual({k: v for (k, v) in cats.items() if v}, {})
198 self.assertEqual(models.Fragment.objects.all().count(), 0,
199 "orphaned fragments left")
200 self.assertEqual(models.Tag.intermediary_table_model.objects.all().count(), 0,
201 "orphaned TagRelation objects left")
203 def test_deleted_tag(self):
204 """ there should be no tag relations left after deleting tags """
206 models.Tag.objects.all().delete()
207 self.assertEqual(len(self.book.related_themes()), 0)
208 self.assertEqual(models.Tag.intermediary_table_model.objects.all().count(), 0,
209 "orphaned TagRelation objects left")
212 class TestIdenticalTag(WLTestCase):
215 WLTestCase.setUp(self)
216 author = PersonStub((), "Tag")
218 self.book_info = BookInfoStub(author=author,
223 self.book_text = """<utwor>
226 <begin id="m01" /><motyw id="m01">tag</motyw>Ala ma kota<end id="m01" />
233 def test_book_tags(self):
234 """ there should be all related tags in relevant categories """
235 book = models.Book.from_text_and_meta(ContentFile(self.book_text), self.book_info)
237 related_themes = book.related_themes()
238 for category in 'author', 'kind', 'genre', 'epoch':
239 self.assertTrue('tag' in [tag.slug for tag in book.tags.filter(category=category)],
240 'missing related tag for %s' % category)
241 self.assertTrue('tag' in [tag.slug for tag in related_themes])
243 def test_qualified_url(self):
244 models.Book.from_text_and_meta(ContentFile(self.book_text), self.book_info)
245 categories = {'author': 'autor', 'theme': 'motyw', 'epoch': 'epoka', 'kind':'rodzaj', 'genre':'gatunek'}
246 for cat, localcat in categories.iteritems():
247 context = self.client.get('/katalog/%s/tag/' % localcat).context
248 self.assertEqual(1, len(context['object_list']))
249 self.assertNotEqual({}, context['categories'])
250 self.assertFalse(context['categories'].get(cat, False))
253 class BookTagsTests(WLTestCase):
254 """ tests the /katalog/lektura/book/ page for related tags """
257 WLTestCase.setUp(self)
258 author1 = PersonStub(("Common",), "Man")
259 author2 = PersonStub(("Jim",), "Lazy")
261 child_info = BookInfoStub(authors=(author1, author2), genre="ChildGenre", epoch='Epoch', kind="ChildKind",
262 **info_args(u"Child"))
263 parent_info = BookInfoStub(author=author1, genre="Genre", epoch='Epoch', kind="Kind",
264 parts=[child_info.url],
265 **info_args(u"Parent"))
267 for info in child_info, parent_info:
268 book_text = """<utwor><opowiadanie><akap>
270 <motyw id="m01">Theme, %sTheme</motyw>
273 </akap></opowiadanie></utwor>
274 """ % info.title.encode('utf-8')
275 models.Book.from_text_and_meta(ContentFile(book_text), info)
277 def test_book_tags(self):
278 """ book should have own tags and whole tree's themes """
280 book = models.Book.objects.get(slug='parent')
281 related_themes = book.related_themes()
283 self.assertEqual([t.slug for t in book.tags.filter(category='author')],
285 self.assertEqual([t.slug for t in book.tags.filter(category='kind')],
287 self.assertEqual([(tag.name, tag.count) for tag in related_themes],
288 [('ChildTheme', 1), ('ParentTheme', 1), ('Theme', 2)])
290 def test_catalogue_tags(self):
291 """ test main page tags and counts """
292 context = self.client.get('/katalog/').context
293 self.assertEqual([(tag.name, tag.count) for tag in context['categories']['author']],
294 [('Jim Lazy', 1), ('Common Man', 1)])
295 self.assertEqual([(tag.name, tag.count) for tag in context['categories']['theme']],
296 [('ChildTheme', 1), ('ParentTheme', 1), ('Theme', 2)])