mygpo/test.py

   1 import os.path
   2
   3 from django.conf import settings
   4
   5 from couchdbkit import Database
   6 from couchdbkit.loaders import FileSystemDocsLoader
   7 from couchdbkit.ext.django.testrunner import CouchDbKitTestSuiteRunner
   8
   9 from mygpo.db.couchdb.utils import sync_design_docs
  10
  11
  12 # inspired by
  13 # http://djangosnippets.org/snippets/2211/
  14
  15 class MygpoTestSuiteRunner(CouchDbKitTestSuiteRunner):
  16     """
  17     Test runner that is able to skip some tests according to settings.py
  18     """
  19
  20     def __init__(self, *args, **kwargs):
  21         self.EXCLUDED_APPS = getattr(settings, 'TEST_EXCLUDE', [])
  22         settings.TESTING = True
  23         super(MygpoTestSuiteRunner, self).__init__(*args, **kwargs)
  24
  25
  26     def setup_databases(self, **kwargs):
  27         ret = super(MygpoTestSuiteRunner, self).setup_databases(**kwargs)
  28         sync_design_docs()
  29         return ret
  30
  31
  32     def build_suite(self, *args, **kwargs):
  33         suite = super(MygpoTestSuiteRunner, self).build_suite(*args, **kwargs)
  34         if not args[0] and not getattr(settings, 'RUN_ALL_TESTS', False):
  35             tests = []
  36             for case in suite:
  37                 pkg = case.__class__.__module__.split('.')[0]
  38                 if pkg not in self.EXCLUDED_APPS:
  39                     tests.append(case)
  40             suite._tests = tests
  41         return suite
  42
  43
  44 def create_auth_string(username, password):
  45     import base64
  46     credentials = base64.encodestring("%s:%s" % (username, password)).rstrip()
  47     auth_string = 'Basic %s' % credentials
  48     return auth_string