92 lines
2.2 KiB
Python
92 lines
2.2 KiB
Python
|
# coding: utf-8
|
||
|
|
from __future__ import unicode_literals, division, absolute_import, print_function
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
import re
|
||
|
|
import sys
|
||
|
|
import warnings
|
||
|
|
|
||
|
|
from . import requires_oscrypto
|
||
|
|
from ._import import _preload
|
||
|
|
|
||
|
|
from tests import test_classes
|
||
|
|
|
||
|
|
if sys.version_info < (3,):
|
||
|
|
range = xrange # noqa
|
||
|
|
from cStringIO import StringIO
|
||
|
|
else:
|
||
|
|
from io import StringIO
|
||
|
|
|
||
|
|
|
||
|
|
run_args = [
|
||
|
|
{
|
||
|
|
'name': 'regex',
|
||
|
|
'kwarg': 'matcher',
|
||
|
|
},
|
||
|
|
{
|
||
|
|
'name': 'repeat_count',
|
||
|
|
'kwarg': 'repeat',
|
||
|
|
'cast': 'int',
|
||
|
|
},
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def run(matcher=None, repeat=1, ci=False):
|
||
|
|
"""
|
||
|
|
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.
|
||
|
|
|
||
|
|
:param repeat:
|
||
|
|
An integer - the number of times to run the tests
|
||
|
|
|
||
|
|
:param ci:
|
||
|
|
A bool, indicating if the tests are being run as part of CI
|
||
|
|
|
||
|
|
:return:
|
||
|
|
A bool - if the tests succeeded
|
||
|
|
"""
|
||
|
|
|
||
|
|
_preload(requires_oscrypto, not ci)
|
||
|
|
|
||
|
|
warnings.filterwarnings("error")
|
||
|
|
|
||
|
|
loader = unittest.TestLoader()
|
||
|
|
# We have to manually track the list of applicable tests because for
|
||
|
|
# some reason with Python 3.4 on Windows, the tests in a suite are replaced
|
||
|
|
# with None after being executed. This breaks the repeat functionality.
|
||
|
|
test_list = []
|
||
|
|
for test_class in test_classes():
|
||
|
|
if matcher:
|
||
|
|
names = loader.getTestCaseNames(test_class)
|
||
|
|
for name in names:
|
||
|
|
if re.search(matcher, name):
|
||
|
|
test_list.append(test_class(name))
|
||
|
|
else:
|
||
|
|
test_list.append(loader.loadTestsFromTestCase(test_class))
|
||
|
|
|
||
|
|
stream = sys.stdout
|
||
|
|
verbosity = 1
|
||
|
|
if matcher and repeat == 1:
|
||
|
|
verbosity = 2
|
||
|
|
elif repeat > 1:
|
||
|
|
stream = StringIO()
|
||
|
|
|
||
|
|
for _ in range(0, repeat):
|
||
|
|
suite = unittest.TestSuite()
|
||
|
|
for test in test_list:
|
||
|
|
suite.addTest(test)
|
||
|
|
result = unittest.TextTestRunner(stream=stream, verbosity=verbosity).run(suite)
|
||
|
|
|
||
|
|
if len(result.errors) > 0 or len(result.failures) > 0:
|
||
|
|
if repeat > 1:
|
||
|
|
print(stream.getvalue())
|
||
|
|
return False
|
||
|
|
|
||
|
|
if repeat > 1:
|
||
|
|
stream.truncate(0)
|
||
|
|
|
||
|
|
return True
|