91 lines
2.3 KiB
Python
91 lines
2.3 KiB
Python
# vim:fileencoding=utf-8:noet
|
|
from __future__ import (unicode_literals, division, absolute_import, print_function)
|
|
|
|
import sys
|
|
|
|
if sys.version_info < (2, 7):
|
|
from unittest2 import TestCase as _TestCase # NOQA
|
|
from unittest2 import main as _main # NOQA
|
|
from unittest2.case import SkipTest # NOQA
|
|
else:
|
|
from unittest import TestCase as _TestCase # NOQA
|
|
from unittest import main as _main # NOQA
|
|
from unittest.case import SkipTest # NOQA
|
|
|
|
|
|
class PowerlineDummyTest(object):
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def __exit__(self, *args):
|
|
pass
|
|
|
|
def fail(self, *args, **kwargs):
|
|
pass
|
|
|
|
def exception(self, *args, **kwargs):
|
|
pass
|
|
|
|
|
|
class PowerlineTestSuite(object):
|
|
def __init__(self, name):
|
|
self.name = name
|
|
|
|
def __enter__(self):
|
|
self.saved_current_suite = os.environ['POWERLINE_CURRENT_SUITE']
|
|
os.environ['POWERLINE_CURRENT_SUITE'] = (
|
|
self.saved_current_suite + '/' + self.name)
|
|
self.suite = self.saved_current_suite + '/' + self.name
|
|
return self
|
|
|
|
def __exit__(self, exc_type, exc_value, traceback):
|
|
if exc_type is not None:
|
|
self.exception(
|
|
'suite_noexcept',
|
|
'Exception while running test suite: {0!r}'.format(exc_value),
|
|
)
|
|
os.environ['POWERLINE_CURRENT_SUITE'] = self.saved_current_suite
|
|
|
|
def record_test_failure(self, fail_char, test_name, message, allow_failure=False):
|
|
if allow_failure:
|
|
fail_char = 'A' + fail_char
|
|
full_msg = '{fail_char} {suite}|{test_name} :: {message}'.format(
|
|
fail_char=fail_char,
|
|
suite=self.suite,
|
|
test_name=test_name,
|
|
message=message,
|
|
)
|
|
with open(os.environ['FAILURES_FILE'], 'a') as ffd:
|
|
ffd.write(full_msg + '\n')
|
|
return False
|
|
|
|
def exception(self, test_name, message, allow_failure=False):
|
|
return self.record_test_failure('E', test_name, message, allow_failure)
|
|
|
|
def fail(self, test_name, message, allow_failure=False):
|
|
return self.record_test_failure('F', test_name, message, allow_failure)
|
|
|
|
def test(self, name, attempts_left=0):
|
|
if not attempts_left:
|
|
return PowerlineSingleTest(self, name)
|
|
else:
|
|
return PowerlineDummyTest()
|
|
|
|
def subsuite(self, name):
|
|
return PowerlineTestSuite(name)
|
|
|
|
|
|
suite = None
|
|
|
|
|
|
def main(*args, **kwargs):
|
|
global suite
|
|
suite = PowerlineTestSuite(sys.argv[0])
|
|
_main(*args, **kwargs)
|
|
|
|
|
|
class TestCase(_TestCase):
|
|
def fail(self, *args, **kwargs):
|
|
super(TestCase, self).fail(*args, **kwargs)
|
|
suite.fail(self.__class__.__name__, 'Failed test')
|