X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/199ba2f0ed2a8abdceddadd17e27d0f541f600c9..8e7ed6210507c648358a936028fff70a28db0626:/apps/catalogue/tests.py diff --git a/apps/catalogue/tests.py b/apps/catalogue/tests.py index ddd6fb73e..068939806 100644 --- a/apps/catalogue/tests.py +++ b/apps/catalogue/tests.py @@ -1,11 +1,15 @@ # -*- coding: utf-8 -*- from django.test import TestCase from catalogue import models, views +from django.core.files.base import ContentFile from django.contrib.auth.models import User, AnonymousUser +from django.test.client import Client from nose.tools import raises +from StringIO import StringIO + +class BasicSearchLogicTests(TestCase): -class BasicSearchTests(TestCase): def setUp(self): self.author_tag = models.Tag.objects.create( name=u'Adam Mickiewicz [SubWord]', @@ -69,3 +73,185 @@ class BasicSearchTests(TestCase): self.assertEqual(views.find_best_matches(u'Żelenski'), (self.unicode_tag,)) self.assertEqual(views.find_best_matches(u'zelenski'), (self.unicode_tag,)) + +class PersonStub(object): + + def __init__(self, first_names, last_name): + self.first_names = first_names + self.last_name = last_name + +class BookInfoStub(object): + + def __init__(self, **kwargs): + self.__dict = kwargs + + def __setattr__(self, key, value): + if not key.startswith('_'): + self.__dict[key] = value + return object.__setattr__(self, key, value) + + def __getattr__(self, key): + return self.__dict[key] + + def to_dict(self): + return dict((key, unicode(value)) for key, value in self.__dict.items()) + +class BookImportLogicTests(TestCase): + + def setUp(self): + self.book_info = BookInfoStub( + url=u"http://wolnelektury.pl/example/default_book", + about=u"http://wolnelektury.pl/example/URI/default_book", + title=u"Default Book", + author=PersonStub(("Jim",), "Lazy"), + kind="X-Kind", + genre="X-Genre", + epoch="X-Epoch", + ) + + self.expected_tags = [ + ('author', 'jim-lazy'), + ('book', 'l-default_book'), + ('genre', 'x-genre'), + ('epoch', 'x-epoch'), + ('kind', 'x-kind'), + ] + self.expected_tags.sort() + + def test_empty_book(self): + BOOK_TEXT = "" + book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info) + + self.assertEqual(book.title, "Default Book") + self.assertEqual(book.slug, "default_book") + self.assert_(book.parent is None) + self.assertFalse(book.has_html_file()) + + # no fragments generated + self.assertEqual(book.fragments.count(), 0) + + # TODO: this should be filled out probably... + self.assertEqual(book.wiki_link, '') + self.assertEqual(book.gazeta_link, '') + self.assertEqual(book._short_html, '') + self.assertEqual(book.description, '') + + tags = [ (tag.category, tag.slug) for tag in book.tags ] + tags.sort() + + self.assertEqual(tags, self.expected_tags) + + def test_book_with_fragment(self): + BOOK_TEXT = """ + + LoveAla ma kota + + """ + + book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info) + self.assertTrue(book.has_html_file()) + + self.assertEqual(book.fragments.count(), 1) + self.assertEqual(book.fragments.all()[0].text, u'

Ala ma kota

\n') + + self.assert_(('theme', 'love') in [ (tag.category, tag.slug) for tag in book.tags ]) + + def test_book_replace_title(self): + BOOK_TEXT = """""" + self.book_info.title = u"Extraordinary" + book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info) + + tags = [ (tag.category, tag.slug) for tag in book.tags ] + tags.sort() + + self.assertEqual(tags, self.expected_tags) + + def test_book_replace_author(self): + BOOK_TEXT = """""" + self.book_info.author = PersonStub(("Hans", "Christian"), "Andersen") + book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info) + + tags = [ (tag.category, tag.slug) for tag in book.tags ] + tags.sort() + + self.expected_tags.remove(('author', 'jim-lazy')) + self.expected_tags.append(('author', 'hans-christian-andersen')) + self.expected_tags.sort() + + self.assertEqual(tags, self.expected_tags) + + # the old tag should disappear + self.assertRaises(models.Tag.DoesNotExist, models.Tag.objects.get, + slug="jim-lazy", category="author") + + + +class BooksByTagFlat(TestCase): + def setUp(self): + self.tag_empty = models.Tag(name='Empty tag', slug='empty', category='author') + self.tag_common = models.Tag(name='Common author', slug='common', category='author') + + self.tag_kind1 = models.Tag(name='Type 1', slug='type1', category='kind') + self.tag_kind2 = models.Tag(name='Type 2', slug='type2', category='kind') + self.tag_kind3 = models.Tag(name='Type 3', slug='type3', category='kind') + for tag in self.tag_empty, self.tag_common, self.tag_kind1, self.tag_kind2, self.tag_kind3: + tag.save() + + + self.parent = models.Book(title='Parent', slug='parent') + self.parent.save() + + self.similar_child = models.Book(title='Similar child', + slug='similar_child', + parent=self.parent) + self.similar_child.save() + self.similar_grandchild = models.Book(title='Similar grandchild', + slug='similar_grandchild', + parent=self.similar_child) + self.similar_grandchild.save() + for book in self.parent, self.similar_child, self.similar_grandchild: + book.tags = [self.tag_common, self.tag_kind1] + book.save() + + self.different_child = models.Book(title='Different child', + slug='different_child', + parent=self.parent) + self.different_child.save() + self.different_child.tags = [self.tag_common, self.tag_kind2] + self.different_child.save() + self.different_grandchild = models.Book(title='Different grandchild', + slug='different_grandchild', + parent=self.different_child) + self.different_grandchild.save() + self.different_grandchild.tags = [self.tag_common, self.tag_kind3] + self.different_grandchild.save() + + for book in models.Book.objects.all(): + l_tag = models.Tag(name=book.title, slug='l-'+book.slug, category='book') + l_tag.save() + book.tags = list(book.tags) + [l_tag] + + + self.client = Client() + + def test_nonexistent_tag(self): + """ Looking for a non-existent tag should yield 404 """ + self.assertEqual(404, self.client.get('/katalog/czeslaw_milosz/').status_code) + + def test_book_tag(self): + """ Looking for a book tag isn't permitted """ + self.assertEqual(404, self.client.get('/katalog/parent/').status_code) + + def test_tag_empty(self): + """ Tag with no books should return no books and no related tags """ + context = self.client.get('/katalog/empty/').context + self.assertEqual(0, len(context['object_list'])) + self.assertEqual(0, len(context['categories'])) + + def test_tag_common(self): + """ Filtering by tag should only yield top-level books """ + context = self.client.get('/katalog/%s/' % self.tag_common.slug).context + self.assertEqual(list(context['object_list']), + [self.parent]) + +