# -*- 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]',
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 = "<utwor />"
+ 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 = """<utwor>
+ <opowiadanie>
+ <akap><begin id="m01" /><motyw id="m01">Love</motyw>Ala ma kota<end id="m01" /></akap>
+ </opowiadanie></utwor>
+ """
+
+ 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'<p class="paragraph">Ala ma kota</p>\n')
+
+ self.assert_(('theme', 'love') in [ (tag.category, tag.slug) for tag in book.tags ])
+
+ def test_book_replace_title(self):
+ BOOK_TEXT = """<utwor />"""
+ 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 = """<utwor />"""
+ 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])
+
+
fnp / wolnelektury.git / blobdiff