X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/a70cbcf401eb1ebc0009cc1ce9d0d8e664cc9642..4298f7948e8963176debf5877fce30c49ae4e3ad:/src/catalogue/tests/test_book_import.py diff --git a/src/catalogue/tests/test_book_import.py b/src/catalogue/tests/test_book_import.py new file mode 100644 index 000000000..da5e58621 --- /dev/null +++ b/src/catalogue/tests/test_book_import.py @@ -0,0 +1,431 @@ +# -*- 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 catalogue.test_utils import * +from catalogue import models +from librarian import WLURI + +from os import path, makedirs + + +class BookImportLogicTests(WLTestCase): + + def setUp(self): + WLTestCase.setUp(self) + self.book_info = BookInfoStub( + url=WLURI.from_slug(u"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", + language=u"pol", + ) + + self.expected_tags = [ + ('author', 'jim-lazy'), + ('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.description, '') + + tags = [(tag.category, tag.slug) for tag in book.tags] + tags.sort() + + self.assertEqual(tags, self.expected_tags) + + def test_not_quite_empty_book(self): + """ Not empty, but without any real text. + + Should work like any other non-empty book. + """ + + book_text = """ + + Nic + + """ + + book = models.Book.from_text_and_meta(ContentFile(book_text), self.book_info) + self.assertTrue(book.has_html_file()) + + 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.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')], + book.fragments.all()[0].tags.filter(category='theme').values_list('category', 'slug')) + + 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) + + def test_book_with_invalid_slug(self): + """ Book with invalid characters in slug shouldn't be imported """ + self.book_info.url = WLURI.from_slug(u"default_book") + book_text = "" + with self.assertRaises(ValueError): + models.Book.from_text_and_meta(ContentFile(book_text), self.book_info) + + def test_book_replace_title(self): + book_text = """""" + models.Book.from_text_and_meta(ContentFile(book_text), self.book_info) + self.book_info.title = u"Extraordinary" + book = models.Book.from_text_and_meta(ContentFile(book_text), self.book_info, overwrite=True) + + 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 = """""" + models.Book.from_text_and_meta(ContentFile(book_text), self.book_info) + self.book_info.author = PersonStub(("Hans", "Christian"), "Andersen") + book = models.Book.from_text_and_meta(ContentFile(book_text), self.book_info, overwrite=True) + + 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 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"), + self.book_info.kinds = self.book_info.kind, 'Y-Kind', + self.book_info.genres = self.book_info.genre, 'Y-Genre', + self.book_info.epochs = self.book_info.epoch, 'Y-Epoch', + + self.expected_tags.extend([ + ('author', 'joe-dilligent'), + ('genre', 'y-genre'), + ('epoch', 'y-epoch'), + ('kind', 'y-kind'), + ]) + self.expected_tags.sort() + + 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) + + +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 = parent.related_themes() + self.assertEqual(['Kot'], [tag.name for tag in themes], 'wrong related theme list') + + +class TreeImportTest(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.CHILD_TEXT = """ + + Pies + Ala ma kota + + """ + self.child = models.Book.from_text_and_meta( + ContentFile(self.CHILD_TEXT), self.child_info) + + self.book_info = BookInfoStub( + genre='X-Genre', + epoch='X-Epoch', + kind='X-Kind', + author=PersonStub(("Joe",), "Doe"), + parts=[self.child_info.url], + **info_args("Book") + ) + self.BOOK_TEXT = """""" + self.book = models.Book.from_text_and_meta( + ContentFile(self.BOOK_TEXT), self.book_info) + + self.parent_info = BookInfoStub( + genre='X-Genre', + epoch='X-Epoch', + kind='X-Kind', + author=PersonStub(("Jim",), "Lazy"), + parts=[self.book_info.url], + **info_args("Parent") + ) + self.PARENT_TEXT = """""" + self.parent = models.Book.from_text_and_meta( + ContentFile(self.PARENT_TEXT), self.parent_info) + + def test_ok(self): + self.assertEqual( + list(self.client.get('/katalog/gatunek/x-genre/').context['object_list']), + [self.parent], + u"There should be only parent on common tag page." + ) + # pies = models.Tag.objects.get(slug='pies') + themes = self.parent.related_themes() + self.assertEqual(len(themes), 1, u"There should be child theme in parent theme counter.") + # TODO: book_count is deprecated, update here. + # epoch = models.Tag.objects.get(slug='x-epoch') + # self.assertEqual(epoch.book_count, 1, u"There should be only parent in common tag's counter.") + + def test_child_republish(self): + child_text = """ + + Pies, Kot + Ala ma kota + + """ + models.Book.from_text_and_meta( + ContentFile(child_text), self.child_info, overwrite=True) + self.assertEqual( + list(self.client.get('/katalog/gatunek/x-genre/').context['object_list']), + [self.parent], + u"There should only be parent on common tag page." + ) + # pies = models.Tag.objects.get(slug='pies') + # kot = models.Tag.objects.get(slug='kot') + self.assertEqual(len(self.parent.related_themes()), 2, + u"There should be child themes in parent theme counter.") + # TODO: book_count is deprecated, update here. + # epoch = models.Tag.objects.get(slug='x-epoch') + # self.assertEqual(epoch.book_count, 1, u"There should only be parent in common tag's counter.") + + def test_book_change_child(self): + second_child_info = BookInfoStub( + genre='X-Genre', + epoch='X-Epoch', + kind='Other-Kind', + author=PersonStub(("Joe",), "Doe"), + **info_args("Second Child") + ) + second_child_text = """ + + Kot + Ala ma kota + + """ + # Import a second child. + second_child = models.Book.from_text_and_meta( + ContentFile(second_child_text), second_child_info) + # The book has only this new child now. + self.book_info.parts = [second_child_info.url] + self.book = models.Book.from_text_and_meta( + ContentFile(self.BOOK_TEXT), self.book_info, overwrite=True) + + self.assertEqual( + set(self.client.get('/katalog/gatunek/x-genre/').context['object_list']), + {self.parent, self.child}, + u"There should be parent and old child on common tag page." + ) + # kot = models.Tag.objects.get(slug='kot') + self.assertEqual(len(self.parent.related_themes()), 1, + u"There should only be new child themes in parent theme counter.") + # # book_count deprecated, update test. + # epoch = models.Tag.objects.get(slug='x-epoch') + # self.assertEqual(epoch.book_count, 2, + # u"There should be parent and old child in common tag's counter.") + self.assertEqual( + list(self.client.get('/katalog/lektura/parent/motyw/kot/').context['fragments']), + [second_child.fragments.all()[0]], + u"There should be new child's fragments on parent's theme page." + ) + self.assertEqual( + list(self.client.get('/katalog/lektura/parent/motyw/pies/').context['fragments']), + [], + u"There should be no old child's fragments on parent's theme page." + ) + + +class MultilingualBookImportTest(WLTestCase): + def setUp(self): + WLTestCase.setUp(self) + common_uri = WLURI.from_slug('common-slug') + + self.pol_info = BookInfoStub( + genre='X-Genre', + epoch='X-Epoch', + kind='X-Kind', + author=PersonStub(("Joe",), "Doe"), + variant_of=common_uri, + **info_args(u"Książka") + ) + + self.eng_info = BookInfoStub( + genre='X-Genre', + epoch='X-Epoch', + kind='X-Kind', + author=PersonStub(("Joe",), "Doe"), + variant_of=common_uri, + **info_args("A book", "eng") + ) + + def test_multilingual_import(self): + book_text = """A""" + + models.Book.from_text_and_meta(ContentFile(book_text), self.pol_info) + models.Book.from_text_and_meta(ContentFile(book_text), self.eng_info) + + self.assertEqual( + set([b.language for b in models.Book.objects.all()]), + {'pol', 'eng'}, + 'Books imported in wrong languages.' + ) + + +class BookImportGenerateTest(WLTestCase): + def setUp(self): + WLTestCase.setUp(self) + xml = path.join(path.dirname(__file__), 'files/fraszka-do-anusie.xml') + self.book = models.Book.from_xml_file(xml) + + def test_gen_pdf(self): + self.book.pdf_file.build() + book = models.Book.objects.get(pk=self.book.pk) + self.assertTrue(path.exists(book.pdf_file.path)) + + def test_gen_pdf_parent(self): + """This book contains a child.""" + xml = path.join(path.dirname(__file__), "files/fraszki.xml") + parent = models.Book.from_xml_file(xml) + parent.pdf_file.build() + parent = models.Book.objects.get(pk=parent.pk) + self.assertTrue(path.exists(parent.pdf_file.path)) + + def test_custom_pdf(self): + from catalogue.tasks import build_custom_pdf + out = 'test-custom.pdf' + absoulute_path = path.join(settings.MEDIA_ROOT, out) + + if not path.exists(path.dirname(absoulute_path)): + makedirs(path.dirname(absoulute_path)) + + build_custom_pdf(self.book.id, customizations=['nofootnotes', '13pt', 'a4paper'], file_name=out) + self.assertTrue(path.exists(absoulute_path))