mygpo/test.py

   1 from django.test.simple import DjangoTestSuiteRunner
   2 from django.conf import settings
   3
   4 # inspired by
   5 # http://djangosnippets.org/snippets/2211/
   6
   7 class MygpoTestSuiteRunner(DjangoTestSuiteRunner):
   8     """
   9     Test runner that doesn't create or destroy a test database (it requires
  10     a pre-setup database), and is able to skip some tests according to
  11     settings.py
  12     """
  13
  14     def __init__(self, *args, **kwargs):
  15         self.EXCLUDED_APPS = getattr(settings, 'TEST_EXCLUDE', [])
  16         settings.TESTING = True
  17         super(MygpoTestSuiteRunner, self).__init__(*args, **kwargs)
  18
  19
  20     def build_suite(self, *args, **kwargs):
  21         suite = super(MygpoTestSuiteRunner, self).build_suite(*args, **kwargs)
  22         if not args[0] and not getattr(settings, 'RUN_ALL_TESTS', False):
  23             tests = []
  24             for case in suite:
  25                 pkg = case.__class__.__module__.split('.')[0]
  26                 if pkg not in self.EXCLUDED_APPS:
  27                     tests.append(case)
  28             suite._tests = tests
  29         return suite
  30
  31
  32     def setup_databases(*args, **kwargs):
  33         from django.db import connections
  34         old_names = []
  35         mirrors = []
  36         for alias in connections:
  37             connection = connections[alias]
  38             # If the database is a test mirror, redirect it's connection
  39             # instead of creating a test database.
  40             if connection.settings_dict['TEST_MIRROR']:
  41                 mirrors.append((alias, connection))
  42                 mirror_alias = connection.settings_dict['TEST_MIRROR']
  43                 connections._connections[alias] = connections[mirror_alias]
  44             else:
  45                 old_names.append((connection, connection.settings_dict['NAME']))
  46
  47                 # don't create a test-database, because we require
  48                 # a pre-existing one
  49                 connection.creation.connection.settings_dict['NAME'] = \
  50                     'test_' + connection.creation.connection.settings_dict['NAME']
  51                 connection.creation.connection.features.confirm()
  52                 #connection.creation.create_test_db(self.verbosity, autoclobber=not self.interactive)
  53
  54         return old_names, mirrors
  55
  56     def teardown_databases(*args, **kwargs):
  57         pass
  58
  59 def create_auth_string(username, password):
  60     import base64
  61     credentials = base64.encodestring("%s:%s" % (username, password)).rstrip()
  62     auth_string = 'Basic %s' % credentials
  63     return auth_string