mirror of
https://git.yoctoproject.org/poky
synced 2026-02-06 00:38:45 +01:00
b40f78a5c7
Some of the decorators need proper cleanup, such as OETimeout which sets a signal handler that needs to be cleared via teardown. If this is not done then the signal gets called later with unpredictable effects. This can be seen if there's a test that is skipped via a decorator and sets a timeout at the same time: the timeout isn't cleared, and is invoked later in a completely unrelated context. The test case for this is added in the next commit. (From OE-Core rev: f42a08e1aabf1ca57e0c09d69fb69cc717c7f156) Signed-off-by: Alexander Kanavin <alex.kanavin@gmail.com> Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>
106 lines
3.3 KiB
Python
106 lines
3.3 KiB
Python
#
|
|
# Copyright (C) 2016 Intel Corporation
|
|
#
|
|
# SPDX-License-Identifier: MIT
|
|
#
|
|
|
|
import base64
|
|
import zlib
|
|
import unittest
|
|
|
|
from oeqa.core.exception import OEQAMissingVariable
|
|
|
|
def _validate_td_vars(td, td_vars, type_msg):
|
|
if td_vars:
|
|
for v in td_vars:
|
|
if not v in td:
|
|
raise OEQAMissingVariable("Test %s need %s variable but"\
|
|
" isn't into td" % (type_msg, v))
|
|
|
|
class OETestCase(unittest.TestCase):
|
|
# TestContext and Logger instance set by OETestLoader.
|
|
tc = None
|
|
logger = None
|
|
|
|
# td has all the variables needed by the test cases
|
|
# is the same across all the test cases.
|
|
td = None
|
|
|
|
# td_vars has the variables needed by a test class
|
|
# or test case instance, if some var isn't into td a
|
|
# OEQAMissingVariable exception is raised
|
|
td_vars = None
|
|
|
|
@classmethod
|
|
def _oeSetUpClass(clss):
|
|
_validate_td_vars(clss.td, clss.td_vars, "class")
|
|
if hasattr(clss, 'setUpHooker') and callable(getattr(clss, 'setUpHooker')):
|
|
clss.setUpHooker()
|
|
clss.setUpClassMethod()
|
|
|
|
@classmethod
|
|
def _oeTearDownClass(clss):
|
|
clss.tearDownClassMethod()
|
|
|
|
def _oeSetUp(self):
|
|
try:
|
|
for d in self.decorators:
|
|
d.setUpDecorator()
|
|
except:
|
|
for d in self.decorators:
|
|
d.tearDownDecorator()
|
|
raise
|
|
self.setUpMethod()
|
|
|
|
def _oeTearDown(self):
|
|
for d in self.decorators:
|
|
d.tearDownDecorator()
|
|
self.tearDownMethod()
|
|
|
|
class OEPTestResultTestCase:
|
|
"""
|
|
Mix-in class to provide functions to make interacting with extraresults for
|
|
the purposes of storing ptestresult data.
|
|
"""
|
|
@staticmethod
|
|
def _compress_log(log):
|
|
logdata = log.encode("utf-8") if isinstance(log, str) else log
|
|
logdata = zlib.compress(logdata)
|
|
logdata = base64.b64encode(logdata).decode("utf-8")
|
|
return {"compressed" : logdata}
|
|
|
|
def ptest_rawlog(self, log):
|
|
if not hasattr(self, "extraresults"):
|
|
self.extraresults = {"ptestresult.sections" : {}}
|
|
self.extraresults["ptestresult.rawlogs"] = {"log" : self._compress_log(log)}
|
|
|
|
def ptest_section(self, section, duration = None, log = None, logfile = None, exitcode = None):
|
|
if not hasattr(self, "extraresults"):
|
|
self.extraresults = {"ptestresult.sections" : {}}
|
|
|
|
sections = self.extraresults.get("ptestresult.sections")
|
|
if section not in sections:
|
|
sections[section] = {}
|
|
|
|
if log is not None:
|
|
sections[section]["log"] = self._compress_log(log)
|
|
elif logfile is not None:
|
|
with open(logfile, "rb") as f:
|
|
sections[section]["log"] = self._compress_log(f.read())
|
|
|
|
if duration is not None:
|
|
sections[section]["duration"] = duration
|
|
if exitcode is not None:
|
|
sections[section]["exitcode"] = exitcode
|
|
|
|
def ptest_result(self, section, test, result):
|
|
if not hasattr(self, "extraresults"):
|
|
self.extraresults = {"ptestresult.sections" : {}}
|
|
|
|
sections = self.extraresults.get("ptestresult.sections")
|
|
if section not in sections:
|
|
sections[section] = {}
|
|
resultname = "ptestresult.{}.{}".format(section, test)
|
|
self.extraresults[resultname] = {"status" : result}
|
|
|