| from django.core.management.base import BaseCommand |
| from optparse import make_option |
| import sys |
| |
| class Command(BaseCommand): |
| option_list = BaseCommand.option_list + ( |
| make_option('--verbosity', action='store', dest='verbosity', default='1', |
| type='choice', choices=['0', '1', '2'], |
| help='Verbosity level; 0=minimal output, 1=normal output, 2=all output'), |
| make_option('--noinput', action='store_false', dest='interactive', default=True, |
| help='Tells Django to NOT prompt the user for input of any kind.'), |
| ) |
| help = 'Runs the test suite for the specified applications, or the entire site if no apps are specified.' |
| args = '[appname ...]' |
| |
| requires_model_validation = False |
| |
| def handle(self, *test_labels, **options): |
| from django.conf import settings |
| |
| verbosity = int(options.get('verbosity', 1)) |
| interactive = options.get('interactive', True) |
| |
| test_path = settings.TEST_RUNNER.split('.') |
| # Allow for Python 2.5 relative paths |
| if len(test_path) > 1: |
| test_module_name = '.'.join(test_path[:-1]) |
| else: |
| test_module_name = '.' |
| test_module = __import__(test_module_name, {}, {}, test_path[-1]) |
| test_runner = getattr(test_module, test_path[-1]) |
| |
| failures = test_runner(test_labels, verbosity=verbosity, interactive=interactive) |
| if failures: |
| sys.exit(failures) |