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
|
#!/usr/bin/python
import sys
import unittest
class TestLoader(unittest.TestLoader):
"""Custom TestLoader to extend the stock python one."""
suiteClass = unittest.TestSuite
def loadTestsFromModuleNames(self, names):
result = self.suiteClass()
for name in names:
result.addTests(self.loadTestsFromModuleName(name))
return result
def loadTestsFromModuleName(self, name):
result = self.suiteClass()
__import__(name, globals(), locals(), [])
module = sys.modules[name]
result.addTests(self.loadTestsFromModule(module))
return result
class TestProgram(unittest.TestProgram):
def __init__(self, module='__main__'):
# Tell unittest to call createTests
unittest.TestProgram.__init__(self, module, testLoader=TestLoader(),
# Tell unittest to call createTests
defaultTest=True,
)
def createTests(self):
suite = self.testLoader.suiteClass()
std_tests = self.testLoader.loadTestsFromModule(self.module)
suite.addTests(std_tests)
suite.addTests(self.testLoader.loadTestsFromModuleNames([
'udd.tests',
'udd.tests.test_circuit_breaker',
'udd.tests.test_config',
'udd.tests.test_commit_database',
'udd.tests.test_import_list',
'udd.tests.test_import_package',
'udd.tests.test_package_database',
'udd.tests.test_package_to_import',
'udd.tests.test_revid_database',
'udd.tests.test_status_database',
'udd.tests.test_threads',
]))
self.test = suite
if __name__ == '__main__':
TestProgram(module=__name__)
|