X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/6d42bc478e3d1bd90eb294464748c21e4de0fc63..a3b153ea6e30da275edf3e34b8d2f8ca1bb436cc:/src/catalogue/test_utils.py diff --git a/src/catalogue/test_utils.py b/src/catalogue/test_utils.py index a6e69db74..c15cba717 100644 --- a/src/catalogue/test_utils.py +++ b/src/catalogue/test_utils.py @@ -4,11 +4,11 @@ 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( @@ -17,9 +17,8 @@ from django.conf import settings NO_SEARCH_INDEX=True, CELERY_TASK_ALWAYS_EAGER=True, CACHES={ - 'default': {'BACKEND': 'django.core.cache.backends.dummy.DummyCache'}, - }, - SOLR=settings.SOLR_TEST, + 'default': {'BACKEND': 'django.core.cache.backends.dummy.DummyCache'}, + }, ) class WLTestCase(TestCase): """ @@ -28,7 +27,7 @@ class WLTestCase(TestCase): longMessage = True -class PersonStub(object): +class PersonStub: def __init__(self, first_names, last_name): self.first_names = first_names @@ -38,7 +37,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 = { @@ -59,10 +58,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()) @@ -75,7 +73,7 @@ def info_args(title, language=None): language = 'pol' return { 'title': str(title), - 'url': WLURI.from_slug(slug), + 'url': WLURI(slug), 'about': "http://wolnelektury.pl/example/URI/%s" % slug, 'language': language, }