# -*- coding: utf-8 -*-
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
from django.core.files.base import ContentFile
from django.test import Client
from catalogue import models
related_info = book.related_info()
related_themes = book.related_themes()
for category in 'author', 'kind', 'genre', 'epoch':
- self.assertTrue('tag' in [tag[1] for tag in related_info['tags'][category]],
+ self.assertTrue('tag' in [tag['slug'] for tag in related_info['tags'][category]],
'missing related tag for %s' % category)
self.assertTrue('tag' in [tag.slug for tag in related_themes])
related_info = book.related_info()
related_themes = book.related_themes()
- self.assertEqual(related_info['tags']['author'],
- [('Common Man', 'common-man')])
- self.assertEqual(related_info['tags']['kind'],
- [('Kind', 'kind')])
+ self.assertEqual([t['slug'] for t in related_info['tags']['author']],
+ ['common-man'])
+ self.assertEqual([t['slug'] for t in related_info['tags']['kind']],
+ ['kind'])
self.assertEqual([(tag.name, tag.count) for tag in related_themes],
[('ChildTheme', 1), ('ParentTheme', 1), ('Theme', 2)])
- def test_main_page_tags(self):
+ def test_catalogue_tags(self):
""" test main page tags and counts """
- from catalogue.templatetags.catalogue_tags import catalogue_menu
- menu = catalogue_menu()
- self.assertEqual([(tag.name, tag.book_count) for tag in menu['author']],
+ context = self.client.get('/katalog/').context
+ self.assertEqual([(tag.name, tag.book_count) for tag in context['categories']['author']],
[('Jim Lazy', 1), ('Common Man', 1)])
- self.assertEqual([(tag.name, tag.book_count) for tag in menu['theme']],
+ self.assertEqual([(tag.name, tag.book_count) for tag in context['categories']['theme']],
[('ChildTheme', 1), ('ParentTheme', 1), ('Theme', 2)])