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
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""Test runner for sqlparse."""
import fnmatch
import optparse
import os
import sys
import unittest
test_mod = os.path.abspath(os.path.join(os.path.dirname(__file__), '../'))
if test_mod not in sys.path:
sys.path.insert(1, test_mod)
parser = optparse.OptionParser()
parser.add_option('-P', '--profile',
help='Create hotshot profile.',
action='store_true', default=False)
def _path_matches(path, *patterns):
if not patterns:
return True
for pattern in patterns:
if fnmatch.fnmatch(path, pattern):
return True
return False
def _collect_test_modules(base=None, *patterns):
for root, dirnames, filenames in os.walk(os.path.dirname(__file__)):
print root, filenames
for fname in filenames:
if not fname.endswith('.py'):
continue
elif not _path_matches(os.path.join(root, fname), *patterns):
continue
if not root in sys.path:
sys.path.append(root)
modname = os.path.splitext(fname)[0]
yield __import__(modname)
def main(args):
"""Create a TestSuite and run it."""
loader = unittest.TestLoader()
suite = unittest.TestSuite()
for mod in _collect_test_modules(*args):
suite.addTests(loader.loadTestsFromModule(mod))
return unittest.TextTestRunner(verbosity=2).run(suite)
if __name__ == '__main__':
opts, args = parser.parse_args()
if opts.profile:
import hotshot
prof = hotshot.Profile("sqlparse.prof")
prof.runcall(main, args)
prof.close()
else:
result = main(args)
if not result.wasSuccessful():
return_code = 1
else:
return_code = 0
sys.exit(return_code)
|
