b3416e44da
generally useful non-testing facilities. Since extras has been around for a couple of testtools releases now, we're making this into a hard dependency of testtools. (Robert Collins) * Testtools now uses setuptools rather than distutils so that we can document the extras dependency. (Robert Collins)
109 lines
3.0 KiB
Python
109 lines
3.0 KiB
Python
# Copyright (c) 2008-2012 testtools developers. See LICENSE for details.
|
|
|
|
"""Helpers for tests."""
|
|
|
|
__all__ = [
|
|
'LoggingResult',
|
|
]
|
|
|
|
import sys
|
|
|
|
from extras import safe_hasattr
|
|
|
|
from testtools import TestResult
|
|
from testtools.content import TracebackContent
|
|
from testtools import runtest
|
|
|
|
|
|
# Importing to preserve compatibility.
|
|
safe_hasattr
|
|
|
|
# GZ 2010-08-12: Don't do this, pointlessly creates an exc_info cycle
|
|
try:
|
|
raise Exception
|
|
except Exception:
|
|
an_exc_info = sys.exc_info()
|
|
|
|
# Deprecated: This classes attributes are somewhat non deterministic which
|
|
# leads to hard to predict tests (because Python upstream are changing things.
|
|
class LoggingResult(TestResult):
|
|
"""TestResult that logs its event to a list."""
|
|
|
|
def __init__(self, log):
|
|
self._events = log
|
|
super(LoggingResult, self).__init__()
|
|
|
|
def startTest(self, test):
|
|
self._events.append(('startTest', test))
|
|
super(LoggingResult, self).startTest(test)
|
|
|
|
def stop(self):
|
|
self._events.append('stop')
|
|
super(LoggingResult, self).stop()
|
|
|
|
def stopTest(self, test):
|
|
self._events.append(('stopTest', test))
|
|
super(LoggingResult, self).stopTest(test)
|
|
|
|
def addFailure(self, test, error):
|
|
self._events.append(('addFailure', test, error))
|
|
super(LoggingResult, self).addFailure(test, error)
|
|
|
|
def addError(self, test, error):
|
|
self._events.append(('addError', test, error))
|
|
super(LoggingResult, self).addError(test, error)
|
|
|
|
def addSkip(self, test, reason):
|
|
self._events.append(('addSkip', test, reason))
|
|
super(LoggingResult, self).addSkip(test, reason)
|
|
|
|
def addSuccess(self, test):
|
|
self._events.append(('addSuccess', test))
|
|
super(LoggingResult, self).addSuccess(test)
|
|
|
|
def startTestRun(self):
|
|
self._events.append('startTestRun')
|
|
super(LoggingResult, self).startTestRun()
|
|
|
|
def stopTestRun(self):
|
|
self._events.append('stopTestRun')
|
|
super(LoggingResult, self).stopTestRun()
|
|
|
|
def done(self):
|
|
self._events.append('done')
|
|
super(LoggingResult, self).done()
|
|
|
|
def tags(self, new_tags, gone_tags):
|
|
self._events.append(('tags', new_tags, gone_tags))
|
|
super(LoggingResult, self).tags(new_tags, gone_tags)
|
|
|
|
def time(self, a_datetime):
|
|
self._events.append(('time', a_datetime))
|
|
super(LoggingResult, self).time(a_datetime)
|
|
|
|
|
|
def is_stack_hidden():
|
|
return TracebackContent.HIDE_INTERNAL_STACK
|
|
|
|
|
|
def hide_testtools_stack(should_hide=True):
|
|
result = TracebackContent.HIDE_INTERNAL_STACK
|
|
TracebackContent.HIDE_INTERNAL_STACK = should_hide
|
|
return result
|
|
|
|
|
|
def run_with_stack_hidden(should_hide, f, *args, **kwargs):
|
|
old_should_hide = hide_testtools_stack(should_hide)
|
|
try:
|
|
return f(*args, **kwargs)
|
|
finally:
|
|
hide_testtools_stack(old_should_hide)
|
|
|
|
|
|
class FullStackRunTest(runtest.RunTest):
|
|
|
|
def _run_user(self, fn, *args, **kwargs):
|
|
return run_with_stack_hidden(
|
|
False,
|
|
super(FullStackRunTest, self)._run_user, fn, *args, **kwargs)
|