apps/catalogue/test_utils.py

   1 from django.conf import settings
   2 from django.test import TestCase
   3 import shutil
   4 import tempfile
   5 from slughifi import slughifi
   6 from librarian import WLURI
   7
   8 class WLTestCase(TestCase):
   9     """
  10         Generic base class for tests. Adds settings freeze and clears MEDIA_ROOT.
  11     """
  12     def setUp(self):
  13         self._MEDIA_ROOT, settings.MEDIA_ROOT = settings.MEDIA_ROOT, tempfile.mkdtemp(prefix='djangotest_')
  14         settings.NO_BUILD_PDF = settings.NO_BUILD_MOBI = settings.NO_BUILD_EPUB = settings.NO_BUILD_TXT = True
  15         settings.CELERY_ALWAYS_EAGER = True
  16
  17     def tearDown(self):
  18         shutil.rmtree(settings.MEDIA_ROOT, True)
  19         settings.MEDIA_ROOT = self._MEDIA_ROOT
  20
  21 class PersonStub(object):
  22
  23     def __init__(self, first_names, last_name):
  24         self.first_names = first_names
  25         self.last_name = last_name
  26
  27
  28 class BookInfoStub(object):
  29
  30     def __init__(self, **kwargs):
  31         self.__dict = kwargs
  32
  33     def __setattr__(self, key, value):
  34         if not key.startswith('_'):
  35             self.__dict[key] = value
  36         return object.__setattr__(self, key, value)
  37
  38     def __getattr__(self, key):
  39         return self.__dict[key]
  40
  41     def to_dict(self):
  42         return dict((key, unicode(value)) for key, value in self.__dict.items())
  43
  44
  45 def info_args(title, language=None):
  46     """ generate some keywords for comfortable BookInfoCreation  """
  47     slug = unicode(slughifi(title))
  48     if language is None:
  49         language = u'pol'
  50     return {
  51         'title': unicode(title),
  52         'url': WLURI.from_slug_and_lang(slug, language),
  53         'about': u"http://wolnelektury.pl/example/URI/%s" % slug,
  54         'language': language,
  55     }