X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/46a90d7cfe54bb6df2f99175abd0ad7d26b3da72..e7c8c7e1fd0fe2eb867b26ae205b75822204a45f:/apps/catalogue/tests/book_import.py diff --git a/apps/catalogue/tests/book_import.py b/apps/catalogue/tests/book_import.py index 76061d05a..775fc2929 100644 --- a/apps/catalogue/tests/book_import.py +++ b/apps/catalogue/tests/book_import.py @@ -1,7 +1,10 @@ # -*- 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.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 @@ -115,7 +118,7 @@ class BookImportLogicTests(WLTestCase): """ Book with invalid characters in slug shouldn't be imported """ 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 = """""" @@ -292,13 +295,15 @@ class TreeImportTest(WLTestCase): 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}, + themes = self.parent.related_themes() + self.assertEqual(len(themes), 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." - ) + # 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 = """ @@ -317,13 +322,14 @@ class TreeImportTest(WLTestCase): ) 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}, + self.assertEqual(len(self.parent.related_themes()), 2, 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." - ) + # 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( @@ -354,13 +360,14 @@ class TreeImportTest(WLTestCase): 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}, + self.assertEqual(len(self.parent.related_themes()), 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." - ) + # book_count deprecated, update test. + #~ 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']), @@ -432,10 +439,9 @@ class BookImportGenerateTest(WLTestCase): def test_custom_pdf(self): from catalogue.tasks import build_custom_pdf - from catalogue.utils import get_dynamic_path - out = get_dynamic_path(None, 'test-custom', ext='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))