X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/75957f735219259d3b4bc361f80ccd3d7b92a0e9..8d1920873ca67f9e506ff222f8c6d90894925371:/src/catalogue/test_utils.py diff --git a/src/catalogue/test_utils.py b/src/catalogue/test_utils.py index dd5647a7d..6bc5569c6 100644 --- a/src/catalogue/test_utils.py +++ b/src/catalogue/test_utils.py @@ -1,25 +1,24 @@ -# -*- 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 os.path import abspath, dirname, join import tempfile from traceback import extract_stack +from django.conf import settings from django.test import TestCase from django.test.utils import override_settings from slugify import slugify from librarian import WLURI -from django.conf import settings @override_settings( MEDIA_ROOT=tempfile.mkdtemp(prefix='djangotest_'), CATALOGUE_DONT_BUILD={'pdf', 'mobi', 'epub', 'txt', 'fb2', 'cover'}, NO_SEARCH_INDEX=True, - CELERY_ALWAYS_EAGER=True, + CELERY_TASK_ALWAYS_EAGER=True, CACHES={ - 'default': {'BACKEND': 'django.core.cache.backends.dummy.DummyCache'}, - }, + 'default': {'BACKEND': 'django.core.cache.backends.dummy.DummyCache'}, + }, SOLR=settings.SOLR_TEST, ) class WLTestCase(TestCase): @@ -29,7 +28,7 @@ class WLTestCase(TestCase): longMessage = True -class PersonStub(object): +class PersonStub: def __init__(self, first_names, last_name): self.first_names = first_names @@ -39,7 +38,7 @@ class PersonStub(object): return " ".join(self.first_names + (self.last_name,)) -class BookInfoStub(object): +class BookInfoStub: _empty_fields = ['cover_url', 'variant_of'] # allow single definition for multiple-value fields _salias = { @@ -60,10 +59,9 @@ class BookInfoStub(object): except KeyError as e: if key in self._empty_fields: return None - elif key in self._salias: + if key in self._salias: return [getattr(self, self._salias[key])] - else: - raise AttributeError(e) + raise AttributeError(e) def to_dict(self): return dict((key, str(value)) for key, value in self.__dict.items()) @@ -73,11 +71,11 @@ def info_args(title, language=None): """ generate some keywords for comfortable BookInfoCreation """ slug = str(slugify(title)) if language is None: - language = u'pol' + language = 'pol' return { 'title': str(title), 'url': WLURI.from_slug(slug), - 'about': u"http://wolnelektury.pl/example/URI/%s" % slug, + 'about': "http://wolnelektury.pl/example/URI/%s" % slug, 'language': language, }