1 # -*- coding: utf-8 -*-
2 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
5 from os.path import abspath, dirname, join
7 from traceback import extract_stack
8 from django.test import TestCase
9 from django.test.utils import override_settings
10 from slugify import slugify
11 from librarian import WLURI
12 from django.conf import settings
16 MEDIA_ROOT=tempfile.mkdtemp(prefix='djangotest_'),
17 CATALOGUE_DONT_BUILD={'pdf', 'mobi', 'epub', 'txt', 'fb2', 'cover'},
19 CELERY_ALWAYS_EAGER=True,
21 'default': {'BACKEND': 'django.core.cache.backends.dummy.DummyCache'},
23 SOLR=settings.SOLR_TEST,
26 class WLTestCase(TestCase):
28 Generic base class for tests. Adds settings freeze and clears MEDIA_ROOT.
33 class PersonStub(object):
35 def __init__(self, first_names, last_name):
36 self.first_names = first_names
37 self.last_name = last_name
40 return " ".join(self.first_names + (self.last_name,))
43 class BookInfoStub(object):
44 _empty_fields = ['cover_url', 'variant_of']
45 # allow single definition for multiple-value fields
50 def __init__(self, **kwargs):
53 def __setattr__(self, key, value):
54 if not key.startswith('_'):
55 self.__dict[key] = value
56 return object.__setattr__(self, key, value)
58 def __getattr__(self, key):
60 return self.__dict[key]
62 if key in self._empty_fields:
64 elif key in self._salias:
65 return [getattr(self, self._salias[key])]
70 return dict((key, unicode(value)) for key, value in self.__dict.items())
73 def info_args(title, language=None):
74 """ generate some keywords for comfortable BookInfoCreation """
75 slug = unicode(slugify(title))
79 'title': unicode(title),
80 'url': WLURI.from_slug(slug),
81 'about': u"http://wolnelektury.pl/example/URI/%s" % slug,
86 def get_fixture(path, app=None):
88 mod_path = app.__file__
89 f_path = join(dirname(abspath(mod_path)), 'tests/files', path)
91 mod_path = extract_stack(limit=2)[0][0]
92 f_path = join(dirname(abspath(mod_path)), 'files', path)
96 def get_mp3_length(path):