2c451638c2
Exposes an additional logger that uses the test's root logger as the parent for any TestCase instances constructor by the hook. The fixture logger still continues to be the parent of the hook logger. Also changes the CleanEveryN hook to explicitly get marked as a failure if a ServerFailure exception is thrown, rather than relying on TestReport.combine() to do so implicitly.
90 lines
2.5 KiB
Python
90 lines
2.5 KiB
Python
"""
|
|
Interface for customizing the behavior of a test fixture.
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
|
|
from ...logging import loggers
|
|
from ...utils import registry
|
|
|
|
|
|
_HOOKS = {}
|
|
|
|
|
|
def make_custom_behavior(class_name, *args, **kwargs):
|
|
"""
|
|
Factory function for creating CustomBehavior instances.
|
|
"""
|
|
|
|
if class_name not in _HOOKS:
|
|
raise ValueError("Unknown custom behavior class '%s'" % (class_name))
|
|
|
|
return _HOOKS[class_name](*args, **kwargs)
|
|
|
|
|
|
class CustomBehavior(object):
|
|
"""
|
|
The common interface all CustomBehaviors will inherit from.
|
|
"""
|
|
|
|
__metaclass__ = registry.make_registry_metaclass(_HOOKS)
|
|
|
|
REGISTERED_NAME = registry.LEAVE_UNREGISTERED
|
|
|
|
@staticmethod
|
|
def start_dynamic_test(hook_test_case, test_report):
|
|
"""
|
|
If a CustomBehavior wants to add a test case that will show up
|
|
in the test report, it should use this method to add it to the
|
|
report, since we will need to count it as a dynamic test to get
|
|
the stats in the summary information right.
|
|
"""
|
|
test_report.startTest(hook_test_case, dynamic=True)
|
|
|
|
def __init__(self, hook_logger, fixture, description):
|
|
"""
|
|
Initializes the CustomBehavior with the specified fixture.
|
|
"""
|
|
|
|
if not isinstance(hook_logger, loggers.HookLogger):
|
|
raise TypeError("logger must be a HookLogger instance")
|
|
|
|
self.logger = hook_logger
|
|
self.fixture = fixture
|
|
self.description = description
|
|
self.hook_test_case = None
|
|
|
|
def make_dynamic_test(self, test_case_class, *args, **kwargs):
|
|
"""
|
|
Returns an instance of 'test_case_class' configured to use the
|
|
appropriate logger.
|
|
"""
|
|
return test_case_class(self.logger.test_case_logger, *args, **kwargs)
|
|
|
|
def before_suite(self, test_report):
|
|
"""
|
|
The test runner calls this exactly once before they start
|
|
running the suite.
|
|
"""
|
|
pass
|
|
|
|
def after_suite(self, test_report):
|
|
"""
|
|
The test runner calls this exactly once after all tests have
|
|
finished executing. Be sure to reset the behavior back to its
|
|
original state so that it can be run again.
|
|
"""
|
|
pass
|
|
|
|
def before_test(self, test, test_report):
|
|
"""
|
|
Each test will call this before it executes.
|
|
"""
|
|
pass
|
|
|
|
def after_test(self, test, test_report):
|
|
"""
|
|
Each test will call this after it executes.
|
|
"""
|
|
pass
|