forked from hovel/pybbm
-
Notifications
You must be signed in to change notification settings - Fork 0
/
runtests.py
118 lines (106 loc) · 3.83 KB
/
runtests.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#!/usr/bin/env python
import django
import sys
import os
from os.path import dirname, abspath
from optparse import OptionParser
from django.conf import settings
try:
import south
south_installed = True
except ImportError:
south_installed = False
TEMPLATE_CONTEXT_PROCESSORS = [
'django.contrib.auth.context_processors.auth',
'django.core.context_processors.debug',
'django.core.context_processors.i18n',
'django.core.context_processors.media',
'django.core.context_processors.static',
'django.core.context_processors.request',
'pybb.context_processors.processor',
'django.core.context_processors.tz'
]
MIDDLEWARE_CLASSES = (
'django.middleware.common.CommonMiddleware',
'django.contrib.sessions.middleware.SessionMiddleware',
'django.middleware.csrf.CsrfViewMiddleware',
'django.contrib.auth.middleware.AuthenticationMiddleware',
'django.contrib.messages.middleware.MessageMiddleware',
'django.middleware.clickjacking.XFrameOptionsMiddleware',
'pybb.middleware.PybbMiddleware',
)
# For convenience configure settings if they are not pre-configured or if we
# haven't been provided settings to use by environment variable.
if not settings.configured and not os.environ.get('DJANGO_SETTINGS_MODULE'):
DATABASES = {
'default': {
'ENGINE': 'django.db.backends.sqlite3',
'NAME': ':memory:',
'TEST_CHARSET': 'utf8',
}
}
test_db = os.environ.get('DB', 'sqlite')
if test_db == 'mysql':
DATABASES['default'].update({
'ENGINE': 'django.db.backends.mysql',
'NAME': 'pybbm',
'USER': 'root',
'TEST_COLLATION': 'utf8_general_ci',
})
elif test_db == 'postgres':
DATABASES['default'].update({
'ENGINE': 'django.db.backends.postgresql_psycopg2',
'USER': 'postgres',
'NAME': 'pybbm',
'OPTIONS': {
'autocommit': True,
}
})
INSTALLED_APPS = [
'django.contrib.auth',
'django.contrib.admin',
'django.contrib.contenttypes',
'django.contrib.sessions',
'django.contrib.sites',
'test.test_project',
'pybb',
]
if django.VERSION[:2] < (1, 7) and south_installed:
INSTALLED_APPS.append('south')
settings.configure(
DATABASES=DATABASES,
INSTALLED_APPS=INSTALLED_APPS,
ROOT_URLCONF='test.test_project.test_urls',
DEBUG=False,
SITE_ID=1,
STATIC_URL='/static/',
TEMPLATE_DIRS=(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'test/test_project/templates'), ),
PYBB_ATTACHMENT_ENABLE=True,
MIDDLEWARE_CLASSES=MIDDLEWARE_CLASSES,
TEMPLATE_CONTEXT_PROCESSORS=TEMPLATE_CONTEXT_PROCESSORS,
AUTH_USER_MODEL='test_project.CustomUser',
LOGIN_URL='/'
)
try:
from django.test.runner import DiscoverRunner as Runner
except ImportError:
from django.test.simple import DjangoTestSuiteRunner as Runner
def runtests(*test_args, **kwargs):
if django.VERSION[:2] >= (1, 7):
django.setup()
if 'south' in settings.INSTALLED_APPS and south_installed:
from south.management.commands import patch_for_test_db_setup
patch_for_test_db_setup()
if not test_args:
test_args = ['pybb']
parent = dirname(abspath(__file__))
sys.path.insert(0, parent)
test_runner = Runner(verbosity=kwargs.get('verbosity', 1), interactive=kwargs.get('interactive', False),
failfast=kwargs.get('failfast'))
failures = test_runner.run_tests(test_args)
sys.exit(failures)
if __name__ == '__main__':
parser = OptionParser()
parser.add_option('--failfast', action='store_true', default=False, dest='failfast')
(options, args) = parser.parse_args()
runtests(failfast=options.failfast, *args)