X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/46705d393ae76a1d8e0eb2dc8bf121269f6adb67..e97becbbb360b14527801bfc22c3e1ab3fd288a1:/apps/catalogue/tests/book_import.py diff --git a/apps/catalogue/tests/book_import.py b/apps/catalogue/tests/book_import.py index 29be73177..4b01a8f83 100644 --- a/apps/catalogue/tests/book_import.py +++ b/apps/catalogue/tests/book_import.py @@ -1,26 +1,30 @@ # -*- coding: utf-8 -*- -from __future__ import with_statement +# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later. +# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information. +# +from django.conf import settings -from django.core.files.base import ContentFile, File +from django.core.files.base import ContentFile from catalogue.test_utils import * from catalogue import models +from librarian import WLURI from nose.tools import raises -import tempfile -from os import unlink,path +from os import path, makedirs class BookImportLogicTests(WLTestCase): def setUp(self): WLTestCase.setUp(self) self.book_info = BookInfoStub( - url=u"http://wolnelektury.pl/example/default-book", + 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 = [ @@ -112,9 +116,9 @@ class BookImportLogicTests(WLTestCase): @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" + self.book_info.url = WLURI.from_slug(u"default_book") BOOK_TEXT = "" - book = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info) + models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.book_info) def test_book_replace_title(self): BOOK_TEXT = """""" @@ -211,6 +215,15 @@ class ChildImportTests(WLTestCase): **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 = """ @@ -226,42 +239,208 @@ class ChildImportTests(WLTestCase): """ 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'] - + themes = parent.related_themes() self.assertEqual(['Kot'], [tag.name for tag in themes], 'wrong related theme list') -class BookImportGenerateTest(WLTestCase): +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( - url=u"http://wolnelektury.pl/example/default-book", - about=u"http://wolnelektury.pl/example/URI/default_book", - title=u"Default Book", + 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"), - kind="X-Kind", - genre="X-Genre", - epoch="X-Epoch", + 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') + self.assertEqual(self.parent.theme_counter, {pies.pk: 1}, + u"There should be child theme in parent theme counter." + ) + 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(self.parent.theme_counter, {pies.pk: 1, kot.pk: 1}, + u"There should be child themes in parent theme counter." + ) + 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']), + set([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(self.parent.theme_counter, {kot.pk: 1}, + u"There should only be new child themes in parent theme counter." + ) + 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.expected_tags = [ - ('author', 'jim-lazy'), - ('genre', 'x-genre'), - ('epoch', 'x-epoch'), - ('kind', 'x-kind'), - ] - self.expected_tags.sort() + 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""" + + book1 = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.pol_info) + book2 = models.Book.from_text_and_meta(ContentFile(BOOK_TEXT), self.eng_info) + + self.assertEqual( + set([b.language for b in models.Book.objects.all()]), + set(['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): - input = open(path.dirname(__file__) + '/but-w-butonierce-but-w-butonierce.xml') - book = models.Book.from_text_and_meta(File(input), self.book_info, overwrite=True) - book.build_pdf() + 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_child(self): - input = open(path.dirname(__file__) + "/fraszka-do-anusie.xml") - book = models.Book.from_text_and_meta(File(input), self.book_info, overwrite=True) - book.build_pdf() - 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))