X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/c4b4a64757ba65018dc24334cb44a12ad2a4316a..fd9768648fe3e37115655245f7fe7aba2adcc5e5:/apps/catalogue/tests/book_import.py diff --git a/apps/catalogue/tests/book_import.py b/apps/catalogue/tests/book_import.py index e5fa031d4..f4588d169 100644 --- a/apps/catalogue/tests/book_import.py +++ b/apps/catalogue/tests/book_import.py @@ -1,14 +1,20 @@ # -*- coding: utf-8 -*- -from django.core.files.base import ContentFile +from __future__ import with_statement + +from django.core.files.base import ContentFile, File from catalogue.test_utils import * from catalogue import models +from nose.tools import raises +import tempfile +from os import unlink, path, makedirs + class BookImportLogicTests(WLTestCase): def setUp(self): WLTestCase.setUp(self) self.book_info = BookInfoStub( - url=u"http://wolnelektury.pl/example/default_book", + 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"), @@ -30,7 +36,7 @@ class BookImportLogicTests(WLTestCase): 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.assertEqual(book.slug, "default-book") self.assert_(book.parent is None) self.assertFalse(book.has_html_file()) @@ -40,7 +46,6 @@ class BookImportLogicTests(WLTestCase): # 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 ] @@ -76,7 +81,40 @@ class BookImportLogicTests(WLTestCase): 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 ]) + self.assert_(('theme', 'love') in [ (tag.category, tag.slug) for tag in book.fragments.all()[0].tags ]) + + def test_book_with_empty_theme(self): + """ empty themes should be ignored """ + + BOOK_TEXT = """ + + , Love , , Ala ma kota + + """ + + book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info) + self.assert_([('theme', 'love')], + [ (tag.category, tag.slug) for tag in book.fragments.all()[0].tags.filter(category='theme') ]) + + def test_book_with_no_theme(self): + """ fragments with no themes shouldn't be created at all """ + + BOOK_TEXT = """ + + Ala ma kota + + """ + + book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info) + self.assertEqual(book.fragments.count(), 0) + self.assertEqual(book.tags.filter(category='theme').count(), 0) + + @raises(ValueError) + def test_book_with_invalid_slug(self): + """ Book with invalid characters in slug shouldn't be imported """ + self.book_info.url = "http://wolnelektury.pl/example/default_book" + BOOK_TEXT = "" + book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info) def test_book_replace_title(self): BOOK_TEXT = """""" @@ -107,6 +145,29 @@ class BookImportLogicTests(WLTestCase): # the old tag shouldn't disappear models.Tag.objects.get(slug="jim-lazy", category="author") + def test_book_remove_fragment(self): + BOOK_TEXT = """ + + + LoveAla ma kota + HatredTo kot Ali + + + """ + BOOK_TEXT_AFTER = """ + + + LoveAla ma kota + To kot Ali + + + """ + + book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info) + self.assertEqual(book.fragments.count(), 2) + book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT_AFTER), self.book_info, overwrite=True) + self.assertEqual(book.fragments.count(), 1) + def test_multiple_tags(self): BOOK_TEXT = """""" self.book_info.authors = self.book_info.author, PersonStub(("Joe",), "Dilligent"), @@ -127,3 +188,83 @@ class BookImportLogicTests(WLTestCase): tags.sort() self.assertEqual(tags, self.expected_tags) + + +class ChildImportTests(WLTestCase): + + def setUp(self): + WLTestCase.setUp(self) + self.child_info = BookInfoStub( + genre='X-Genre', + epoch='X-Epoch', + kind='X-Kind', + author=PersonStub(("Joe",), "Doe"), + **info_args("Child") + ) + + self.parent_info = BookInfoStub( + genre='X-Genre', + epoch='X-Epoch', + kind='X-Kind', + author=PersonStub(("Jim",), "Lazy"), + parts=[self.child_info.url], + **info_args("Parent") + ) + + def test_child(self): + TEXT = """""" + child = models.Book.from_text_and_meta(ContentFile(TEXT), self.child_info) + parent = models.Book.from_text_and_meta(ContentFile(TEXT), self.parent_info) + author = parent.tags.get(category='author') + books = self.client.get(author.get_absolute_url()).context['object_list'] + self.assertEqual(len(books), 1, + "Only parent book should be visible on author's page") + + def test_child_replace(self): + PARENT_TEXT = """""" + CHILD_TEXT = """ + + PiesAla ma kota + + """ + child = models.Book.from_text_and_meta(ContentFile(CHILD_TEXT), self.child_info) + parent = models.Book.from_text_and_meta(ContentFile(PARENT_TEXT), self.parent_info) + CHILD_TEXT = """ + + KotAla ma kota + + """ + child = models.Book.from_text_and_meta(ContentFile(CHILD_TEXT), self.child_info, overwrite=True) + + themes = self.client.get(parent.get_absolute_url()).context['book_themes'] + + self.assertEqual(['Kot'], [tag.name for tag in themes], + 'wrong related theme list') + + +class BookImportGenerateTest(WLTestCase): + def setUp(self): + WLTestCase.setUp(self) + input = path.join(path.dirname(__file__), 'files/fraszka-do-anusie.xml') + self.book = models.Book.from_xml_file(input) + + def test_gen_pdf(self): + self.book.build_pdf() + self.assertTrue(path.exists(self.book.pdf_file.path)) + + def test_gen_pdf_parent(self): + """This book contains a child.""" + input = path.join(path.dirname(__file__), "files/fraszki.xml") + parent = models.Book.from_xml_file(input) + parent.build_pdf() + self.assertTrue(path.exists(parent.pdf_file.path)) + + def test_custom_pdf(self): + out = models.get_dynamic_path(None, 'test-custom', ext='pdf') + absoulute_path = path.join(settings.MEDIA_ROOT, out) + + if not path.exists(path.dirname(absoulute_path)): + makedirs(path.dirname(absoulute_path)) + + self.book.build_pdf(customizations=['nofootnotes', '13pt', 'a4paper'], file_name=out) + self.assertTrue(path.exists(absoulute_path))