Fix inheritance problem in test/hook.py.

Subclasses cannot modify the signature of their constructors.

     def __init__(self, test):
-        super(PollHook, self).__init__(test.logger)
+        super(PollHook, self).__init__(test)

Change-Id: I764df8871128f9198a03fac4ec2f45528547467a
Signed-off-by: Paul Vinciguerra <pvinci@vinciconsulting.com>
This commit is contained in:
Paul Vinciguerra
2019-01-08 20:37:40 -08:00
committed by Ole Trøan
parent 2baf9422cf
commit 895e2f8506
2 changed files with 17 additions and 18 deletions

View File

@ -17,8 +17,9 @@ class Hook(object):
Generic hooks before/after API/CLI calls
"""
def __init__(self, logger):
self.logger = logger
def __init__(self, test):
self.test = test
self.logger = test.logger
def before_api(self, api_name, api_args):
"""
@ -79,13 +80,12 @@ class VppDiedError(Exception):
class PollHook(Hook):
""" Hook which checks if the vpp subprocess is alive """
def __init__(self, testcase):
super(PollHook, self).__init__(testcase.logger)
self.testcase = testcase
def __init__(self, test):
super(PollHook, self).__init__(test)
def on_crash(self, core_path):
self.logger.error("Core file present, debug with: gdb %s %s" %
(self.testcase.vpp_bin, core_path))
(self.test.vpp_bin, core_path))
check_core_path(self.logger, core_path)
self.logger.error("Running `file %s':" % core_path)
try:
@ -101,27 +101,27 @@ class PollHook(Hook):
Poll the vpp status and throw an exception if it's not running
:raises VppDiedError: exception if VPP is not running anymore
"""
if self.testcase.vpp_dead:
if self.test.vpp_dead:
# already dead, nothing to do
return
self.testcase.vpp.poll()
if self.testcase.vpp.returncode is not None:
self.test.vpp.poll()
if self.test.vpp.returncode is not None:
signaldict = dict(
(k, v) for v, k in reversed(sorted(signal.__dict__.items()))
if v.startswith('SIG') and not v.startswith('SIG_'))
if self.testcase.vpp.returncode in signaldict:
s = signaldict[abs(self.testcase.vpp.returncode)]
if self.test.vpp.returncode in signaldict:
s = signaldict[abs(self.test.vpp.returncode)]
else:
s = "unknown"
msg = "VPP subprocess died unexpectedly with returncode %d [%s]." \
% (self.testcase.vpp.returncode, s)
% (self.test.vpp.returncode, s)
self.logger.critical(msg)
core_path = get_core_path(self.testcase.tempdir)
core_path = get_core_path(self.test.tempdir)
if os.path.isfile(core_path):
self.on_crash(core_path)
self.testcase.vpp_dead = True
self.test.vpp_dead = True
raise VppDiedError(msg)
def before_api(self, api_name, api_args):
@ -151,11 +151,11 @@ class PollHook(Hook):
class StepHook(PollHook):
""" Hook which requires user to press ENTER before doing any API/CLI """
def __init__(self, testcase):
def __init__(self, test):
self.skip_stack = None
self.skip_num = None
self.skip_count = 0
super(StepHook, self).__init__(testcase)
super(StepHook, self).__init__(test)
def skip(self):
if self.skip_stack is None: