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.
68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
"""
|
|
Testing hook for cleaning up data files created by the fixture.
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import os
|
|
import sys
|
|
|
|
from . import interface
|
|
from ..testcases import interface as testcase
|
|
from ... import errors
|
|
|
|
|
|
class CleanEveryN(interface.CustomBehavior):
|
|
"""
|
|
Restarts the fixture after it has ran 'n' tests.
|
|
On mongod-related fixtures, this will clear the dbpath.
|
|
"""
|
|
|
|
DEFAULT_N = 20
|
|
|
|
def __init__(self, hook_logger, fixture, n=DEFAULT_N):
|
|
description = "CleanEveryN (restarts the fixture after running `n` tests)"
|
|
interface.CustomBehavior.__init__(self, hook_logger, fixture, description)
|
|
|
|
# Try to isolate what test triggers the leak by restarting the fixture each time.
|
|
if "detect_leaks=1" in os.getenv("ASAN_OPTIONS", ""):
|
|
self.logger.info("ASAN_OPTIONS environment variable set to detect leaks, so restarting"
|
|
" the fixture after each test instead of after every %d.", n)
|
|
n = 1
|
|
|
|
self.n = n
|
|
self.tests_run = 0
|
|
|
|
def after_test(self, test, test_report):
|
|
self.tests_run += 1
|
|
if self.tests_run < self.n:
|
|
return
|
|
|
|
test_name = "{}:{}".format(test.short_name(), self.__class__.__name__)
|
|
self.hook_test_case = self.make_dynamic_test(testcase.TestCase, "Hook", test_name)
|
|
|
|
interface.CustomBehavior.start_dynamic_test(self.hook_test_case, test_report)
|
|
try:
|
|
self.hook_test_case.logger.info(
|
|
"%d tests have been run against the fixture, stopping it...",
|
|
self.tests_run)
|
|
self.tests_run = 0
|
|
|
|
if not self.fixture.teardown():
|
|
raise errors.ServerFailure("%s did not exit cleanly" % (self.fixture))
|
|
|
|
self.hook_test_case.logger.info("Starting the fixture back up again...")
|
|
self.fixture.setup()
|
|
self.fixture.await_ready()
|
|
except:
|
|
self.hook_test_case.logger.exception(
|
|
"Encountered an error while restarting the fixture.")
|
|
self.hook_test_case.return_code = 2
|
|
test_report.addFailure(self.hook_test_case, sys.exc_info())
|
|
raise
|
|
else:
|
|
self.hook_test_case.return_code = 0
|
|
test_report.addSuccess(self.hook_test_case)
|
|
finally:
|
|
test_report.stopTest(self.hook_test_case)
|