/django/branches/soc2010/app-loading/tests/runtests.py
Python | 201 lines | 180 code | 17 blank | 4 comment | 20 complexity | 8aeb8168a28644884e16efeaa8c66a98 MD5 | raw file
Possible License(s): BSD-3-Clause
- #!/usr/bin/env python
- import os, sys, traceback
- import unittest
- import django.contrib as contrib
- CONTRIB_DIR_NAME = 'django.contrib'
- MODEL_TESTS_DIR_NAME = 'modeltests'
- REGRESSION_TESTS_DIR_NAME = 'regressiontests'
- TEST_TEMPLATE_DIR = 'templates'
- CONTRIB_DIR = os.path.dirname(contrib.__file__)
- MODEL_TEST_DIR = os.path.join(os.path.dirname(__file__), MODEL_TESTS_DIR_NAME)
- REGRESSION_TEST_DIR = os.path.join(os.path.dirname(__file__), REGRESSION_TESTS_DIR_NAME)
- REGRESSION_SUBDIRS_TO_SKIP = ['locale']
- ALWAYS_INSTALLED_APPS = [
- 'django.contrib.contenttypes',
- 'django.contrib.auth',
- 'django.contrib.sites',
- 'django.contrib.flatpages',
- 'django.contrib.redirects',
- 'django.contrib.sessions',
- 'django.contrib.messages',
- 'django.contrib.comments',
- 'django.contrib.admin',
- 'django.contrib.admindocs',
- ]
- def get_test_models():
- models = []
- for loc, dirpath in (MODEL_TESTS_DIR_NAME, MODEL_TEST_DIR), (REGRESSION_TESTS_DIR_NAME, REGRESSION_TEST_DIR), (CONTRIB_DIR_NAME, CONTRIB_DIR):
- for f in os.listdir(dirpath):
- if f.startswith('__init__') or f.startswith('.') or \
- f.startswith('sql') or f.startswith('invalid') or \
- os.path.basename(f) in REGRESSION_SUBDIRS_TO_SKIP:
- continue
- models.append((loc, f))
- return models
- def get_invalid_models():
- models = []
- for loc, dirpath in (MODEL_TESTS_DIR_NAME, MODEL_TEST_DIR), (REGRESSION_TESTS_DIR_NAME, REGRESSION_TEST_DIR), (CONTRIB_DIR_NAME, CONTRIB_DIR):
- for f in os.listdir(dirpath):
- if f.startswith('__init__') or f.startswith('.') or f.startswith('sql'):
- continue
- if f.startswith('invalid'):
- models.append((loc, f))
- return models
- class InvalidModelTestCase(unittest.TestCase):
- def __init__(self, model_label):
- unittest.TestCase.__init__(self)
- self.model_label = model_label
- def runTest(self):
- from django.core.management.validation import get_validation_errors
- from django.db.models.loading import load_app
- from cStringIO import StringIO
- try:
- module = load_app(self.model_label)
- except Exception, e:
- self.fail('Unable to load invalid model module')
- # Make sure sys.stdout is not a tty so that we get errors without
- # coloring attached (makes matching the results easier). We restore
- # sys.stderr afterwards.
- orig_stdout = sys.stdout
- s = StringIO()
- sys.stdout = s
- count = get_validation_errors(s, module)
- sys.stdout = orig_stdout
- s.seek(0)
- error_log = s.read()
- actual = error_log.split('\n')
- expected = module.model_errors.split('\n')
- unexpected = [err for err in actual if err not in expected]
- missing = [err for err in expected if err not in actual]
- self.assert_(not unexpected, "Unexpected Errors: " + '\n'.join(unexpected))
- self.assert_(not missing, "Missing Errors: " + '\n'.join(missing))
- def django_tests(verbosity, interactive, failfast, test_labels):
- from django.conf import settings
- old_installed_apps = settings.INSTALLED_APPS
- old_root_urlconf = getattr(settings, "ROOT_URLCONF", "")
- old_template_dirs = settings.TEMPLATE_DIRS
- old_use_i18n = settings.USE_I18N
- old_login_url = settings.LOGIN_URL
- old_language_code = settings.LANGUAGE_CODE
- old_middleware_classes = settings.MIDDLEWARE_CLASSES
- # Redirect some settings for the duration of these tests.
- settings.INSTALLED_APPS = ALWAYS_INSTALLED_APPS
- settings.ROOT_URLCONF = 'urls'
- settings.TEMPLATE_DIRS = (os.path.join(os.path.dirname(__file__), TEST_TEMPLATE_DIR),)
- settings.USE_I18N = True
- settings.LANGUAGE_CODE = 'en'
- settings.LOGIN_URL = '/accounts/login/'
- settings.MIDDLEWARE_CLASSES = (
- 'django.contrib.sessions.middleware.SessionMiddleware',
- 'django.contrib.auth.middleware.AuthenticationMiddleware',
- 'django.contrib.messages.middleware.MessageMiddleware',
- 'django.middleware.common.CommonMiddleware',
- )
- settings.SITE_ID = 1
- # For testing comment-utils, we require the MANAGERS attribute
- # to be set, so that a test email is sent out which we catch
- # in our tests.
- settings.MANAGERS = ("admin@djangoproject.com",)
- # This import statement is intentionally delayed until after we
- # access settings because of the USE_I18N dependency.
- from django.db.models.loading import load_app
- # Load all the test model apps.
- test_labels_set = set([label.split('.')[0] for label in test_labels])
- for model_dir, model_name in get_test_models():
- model_label = '.'.join([model_dir, model_name])
- # if the model was named on the command line, or
- # no models were named (i.e., run all), import
- # this model and add it to the list to test.
- if not test_labels or model_name in test_labels_set:
- if verbosity >= 2:
- print "Importing model %s" % model_name
- mod = load_app(model_label)
- if mod:
- if model_label not in settings.INSTALLED_APPS:
- settings.INSTALLED_APPS.append(model_label)
- # Add tests for invalid models.
- extra_tests = []
- for model_dir, model_name in get_invalid_models():
- model_label = '.'.join([model_dir, model_name])
- if not test_labels or model_name in test_labels:
- extra_tests.append(InvalidModelTestCase(model_label))
- try:
- # Invalid models are not working apps, so we cannot pass them into
- # the test runner with the other test_labels
- test_labels.remove(model_name)
- except ValueError:
- pass
- # Run the test suite, including the extra validation tests.
- from django.test.utils import get_runner
- if not hasattr(settings, 'TEST_RUNNER'):
- settings.TEST_RUNNER = 'django.test.simple.DjangoTestSuiteRunner'
- TestRunner = get_runner(settings)
- if hasattr(TestRunner, 'func_name'):
- # Pre 1.2 test runners were just functions,
- # and did not support the 'failfast' option.
- import warnings
- warnings.warn(
- 'Function-based test runners are deprecated. Test runners should be classes with a run_tests() method.',
- PendingDeprecationWarning
- )
- failures = TestRunner(test_labels, verbosity=verbosity, interactive=interactive,
- extra_tests=extra_tests)
- else:
- test_runner = TestRunner(verbosity=verbosity, interactive=interactive, failfast=failfast)
- failures = test_runner.run_tests(test_labels, extra_tests=extra_tests)
- if failures:
- sys.exit(bool(failures))
- # Restore the old settings.
- settings.INSTALLED_APPS = old_installed_apps
- settings.ROOT_URLCONF = old_root_urlconf
- settings.TEMPLATE_DIRS = old_template_dirs
- settings.USE_I18N = old_use_i18n
- settings.LANGUAGE_CODE = old_language_code
- settings.LOGIN_URL = old_login_url
- settings.MIDDLEWARE_CLASSES = old_middleware_classes
- if __name__ == "__main__":
- from optparse import OptionParser
- usage = "%prog [options] [model model model ...]"
- parser = OptionParser(usage=usage)
- parser.add_option('-v','--verbosity', action='store', dest='verbosity', default='1',
- type='choice', choices=['0', '1', '2', '3'],
- help='Verbosity level; 0=minimal output, 1=normal output, 2=all output')
- parser.add_option('--noinput', action='store_false', dest='interactive', default=True,
- help='Tells Django to NOT prompt the user for input of any kind.')
- parser.add_option('--failfast', action='store_true', dest='failfast', default=False,
- help='Tells Django to stop running the test suite after first failed test.')
- parser.add_option('--settings',
- help='Python path to settings module, e.g. "myproject.settings". If this isn\'t provided, the DJANGO_SETTINGS_MODULE environment variable will be used.')
- options, args = parser.parse_args()
- if options.settings:
- os.environ['DJANGO_SETTINGS_MODULE'] = options.settings
- elif "DJANGO_SETTINGS_MODULE" not in os.environ:
- parser.error("DJANGO_SETTINGS_MODULE is not set in the environment. "
- "Set it or use --settings.")
- django_tests(int(options.verbosity), options.interactive, options.failfast, args)