forked from coleifer/peewee
/
runtests.py
executable file
·199 lines (182 loc) · 8.53 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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
#!/usr/bin/env python
import optparse
import os
import shutil
import sys
import unittest
def collect():
import tests
runtests(tests, 1)
def runtests(suite, verbosity):
results = unittest.TextTestRunner(verbosity=verbosity).run(suite)
return results.failures, results.errors
def get_option_parser():
parser = optparse.OptionParser()
basic = optparse.OptionGroup(parser, 'Basic test options')
basic.add_option(
'-e',
'--engine',
dest='engine',
help=('Database engine to test, one of '
'[sqlite, postgres, mysql, apsw, sqlcipher, berkeleydb]'))
basic.add_option('-v', '--verbosity', dest='verbosity', default=1, type='int', help='Verbosity of output')
suite = optparse.OptionGroup(parser, 'Simple test suite options')
suite.add_option('-a', '--all', dest='all', default=False, action='store_true', help='Run all tests, including extras')
suite.add_option('-x', '--extra', dest='extra', default=False, action='store_true', help='Run only extras tests')
cases = optparse.OptionGroup(parser, 'Individual test module options')
cases.add_option('--apsw', dest='apsw', default=False, action='store_true', help='apsw tests (requires apsw)')
cases.add_option('--berkeleydb', dest='berkeleydb', default=False, action='store_true', help='berkeleydb tests (requires pysqlite compiled against berkeleydb)')
cases.add_option('--csv', dest='csv', default=False, action='store_true', help='csv tests')
cases.add_option('--dataset', dest='dataset', default=False, action='store_true', help='dataset tests')
cases.add_option('--db-url', dest='db_url', default=False, action='store_true', help='db url tests')
cases.add_option('--djpeewee', dest='djpeewee', default=False, action='store_true', help='djpeewee tests')
cases.add_option('--flask', dest='flask', default=False, action='store_true', help='flask utils tests')
cases.add_option('--gfk', dest='gfk', default=False, action='store_true', help='gfk tests')
cases.add_option('--hybrid', dest='hybrid', default=False, action='store_true', help='hybrid property/method tests')
cases.add_option('--kv', dest='kv', default=False, action='store_true', help='key/value store tests')
cases.add_option('--manytomany', dest='manytomany', default=False, action='store_true', help='manytomany field tests')
cases.add_option('--migrations', dest='migrations', default=False, action='store_true', help='migration helper tests (requires psycopg2)')
cases.add_option('--pool', dest='pool', default=False, action='store_true', help='connection pool tests')
cases.add_option('--postgres-ext', dest='postgres_ext', default=False, action='store_true', help='postgres_ext tests (requires psycopg2)')
cases.add_option('--pwiz', dest='pwiz', default=False, action='store_true', help='pwiz, model code generator')
cases.add_option('--read-slave', dest='read_slave', default=False, action='store_true', help='read_slave tests')
cases.add_option('--reflection', dest='reflection', default=False, action='store_true', help='reflection schema introspector')
cases.add_option('--signals', dest='signals', default=False, action='store_true', help='signals tests')
cases.add_option('--shortcuts', dest='shortcuts', default=False, action='store_true', help='shortcuts tests')
cases.add_option('--sqlcipher-ext', dest='sqlcipher', default=False, action='store_true', help='sqlcipher_ext tests (requires pysqlcipher)')
cases.add_option('--sqlite-ext', dest='sqlite_ext', default=False, action='store_true', help='sqlite_ext tests')
cases.add_option('--test-utils', dest='test_utils', default=False, action='store_true', help='test_utils tests')
parser.add_option_group(basic)
parser.add_option_group(suite)
parser.add_option_group(cases)
return parser
def collect_modules(options):
modules = []
xtra = lambda op: op or options.extra or options.all
if xtra(options.apsw):
try:
from playhouse.tests import test_apsw
modules.append(test_apsw)
except ImportError:
print_('Unable to import apsw tests, skipping')
if xtra(options.berkeleydb):
try:
from playhouse.tests import test_berkeleydb
modules.append(test_berkeleydb)
except ImportError:
print_('Unable to import berkeleydb tests, skipping')
if xtra(options.csv):
from playhouse.tests import test_csv_utils
modules.append(test_csv_utils)
if xtra(options.dataset):
from playhouse.tests import test_dataset
modules.append(test_dataset)
if xtra(options.db_url):
from playhouse.tests import test_db_url
modules.append(test_db_url)
if xtra(options.djpeewee):
from playhouse.tests import test_djpeewee
modules.append(test_djpeewee)
if xtra(options.flask):
try:
import flask
except ImportError:
print_('Unable to import Flask tests, Flask is not installed.')
else:
from playhouse.tests import test_flask_utils
modules.append(test_flask_utils)
if xtra(options.gfk):
from playhouse.tests import test_gfk
modules.append(test_gfk)
if xtra(options.hybrid):
from playhouse.tests import test_hybrid
modules.append(test_hybrid)
if xtra(options.kv):
from playhouse.tests import test_kv
modules.append(test_kv)
if xtra(options.manytomany):
from playhouse.tests import test_manytomany
modules.append(test_manytomany)
if xtra(options.migrations):
try:
from playhouse.tests import test_migrate
modules.append(test_migrate)
except ImportError:
print_('Unable to import migration tests, skipping')
if xtra(options.pool):
try:
from playhouse.tests import test_pool
modules.append(test_pool)
except ImportError:
print_('Unable to import connection pool tests, skipping')
if xtra(options.postgres_ext):
try:
from playhouse.tests import test_postgres
modules.append(test_postgres)
except ImportError:
print_('Unable to import postgres-ext tests, skipping')
if xtra(options.pwiz):
from playhouse.tests import test_pwiz
modules.append(test_pwiz)
if xtra(options.read_slave):
from playhouse.tests import test_read_slave
modules.append(test_read_slave)
if xtra(options.reflection):
from playhouse.tests import test_reflection
modules.append(test_reflection)
if xtra(options.signals):
from playhouse.tests import test_signals
modules.append(test_signals)
if xtra(options.shortcuts):
from playhouse.tests import test_manytomany
from playhouse.tests import test_shortcuts
modules.append(test_shortcuts)
if test_manytomany not in modules:
modules.append(test_manytomany)
if xtra(options.sqlcipher):
try:
from playhouse.tests import test_sqlcipher_ext
modules.append(test_sqlcipher_ext)
except ImportError:
print_('Unable to import pysqlcipher tests, skipping')
if xtra(options.sqlite_ext):
from playhouse.tests import test_sqlite_ext
modules.append(test_sqlite_ext)
if xtra(options.test_utils):
from playhouse.tests import test_test_utils
modules.append(test_test_utils)
if not modules or options.all:
import tests
modules.insert(0, tests)
return modules
if __name__ == '__main__':
parser = get_option_parser()
options, args = parser.parse_args()
if options.engine:
os.environ['PEEWEE_TEST_BACKEND'] = options.engine
os.environ['PEEWEE_TEST_VERBOSITY'] = str(options.verbosity)
from peewee import print_
suite = unittest.TestSuite()
for module in collect_modules(options):
print_('Adding tests for "%s"' % module.__name__)
module_suite = unittest.TestLoader().loadTestsFromModule(module)
suite.addTest(module_suite)
failures, errors = runtests(suite, options.verbosity)
files_to_delete = [
'peewee_test.db',
'peewee_test',
'tmp.db',
'peewee_test.bdb.db',
'peewee_test.cipher.db']
paths_to_delete = ['peewee_test.bdb.db-journal']
for filename in files_to_delete:
if os.path.exists(filename):
os.unlink(filename)
for path in paths_to_delete:
if os.path.exists(path):
shutil.rmtree(path)
if errors:
sys.exit(2)
elif failures:
sys.exit(1)
sys.exit(0)