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
|
# coding: utf-8
from __future__ import unicode_literals, division, absolute_import, print_function
import unittest
import re
from tests.test_ocsp_response_builder import OCSPResponseBuilderTests
from tests.test_ocsp_request_builder import OCSPRequestBuilderTests
test_classes = [OCSPResponseBuilderTests, OCSPRequestBuilderTests]
def make_suite():
"""
Constructs a unittest.TestSuite() of all tests for the package. For use
with setuptools.
:return:
A unittest.TestSuite() object
"""
loader = unittest.TestLoader()
suite = unittest.TestSuite()
for test_class in test_classes:
tests = loader.loadTestsFromTestCase(test_class)
suite.addTests(tests)
return suite
def run(matcher=None):
"""
Runs the tests
:param matcher:
A unicode string containing a regular expression to use to filter test
names by. A value of None will cause no filtering.
:return:
A bool - if the tests succeeded
"""
suite = unittest.TestSuite()
loader = unittest.TestLoader()
for test_class in test_classes:
if matcher:
names = loader.getTestCaseNames(test_class)
for name in names:
if re.search(matcher, name):
suite.addTest(test_class(name))
else:
suite.addTest(loader.loadTestsFromTestCase(test_class))
verbosity = 2 if matcher else 1
result = unittest.TextTestRunner(verbosity=verbosity).run(suite)
return result.wasSuccessful()
|
