84 lines
2.2 KiB
Python
84 lines
2.2 KiB
Python
|
"""
|
||
|
|
Interface for customizing the behavior of a test fixture.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from __future__ import absolute_import
|
||
|
|
|
||
|
|
from ... import logging
|
||
|
|
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, logging.Logger):
|
||
|
|
raise TypeError("logger must be a Logger instance")
|
||
|
|
|
||
|
|
self.logger = hook_logger
|
||
|
|
self.fixture = fixture
|
||
|
|
self.hook_test_case = None
|
||
|
|
self.logger_name = self.__class__.__name__
|
||
|
|
self.description = description
|
||
|
|
|
||
|
|
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
|