configs/arch-config/.config/calibre/plugins/DeACSM/modules/asn1crypto/dev/tests.py

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