236 lines
7.6 KiB
Python
236 lines
7.6 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
import re
|
|
import os
|
|
import subprocess
|
|
from collections import OrderedDict
|
|
import xml.etree.ElementTree as ET
|
|
|
|
result_re = re.compile(".*(PASS|FAIL|SKIP) - (test_)?(.*) in")
|
|
|
|
class Harness:
|
|
GCOV_START = "GCOV_COVERAGE_DUMP_START"
|
|
GCOV_END = "GCOV_COVERAGE_DUMP_END"
|
|
FAULT = "ZEPHYR FATAL ERROR"
|
|
RUN_PASSED = "PROJECT EXECUTION SUCCESSFUL"
|
|
RUN_FAILED = "PROJECT EXECUTION FAILED"
|
|
|
|
def __init__(self):
|
|
self.state = None
|
|
self.type = None
|
|
self.regex = []
|
|
self.matches = OrderedDict()
|
|
self.ordered = True
|
|
self.repeat = 1
|
|
self.tests = {}
|
|
self.id = None
|
|
self.fail_on_fault = True
|
|
self.fault = False
|
|
self.capture_coverage = False
|
|
self.next_pattern = 0
|
|
self.record = None
|
|
self.recording = []
|
|
self.fieldnames = []
|
|
self.ztest = False
|
|
self.is_pytest = False
|
|
|
|
def configure(self, instance):
|
|
config = instance.testcase.harness_config
|
|
self.id = instance.testcase.id
|
|
if "ignore_faults" in instance.testcase.tags:
|
|
self.fail_on_fault = False
|
|
|
|
if config:
|
|
self.type = config.get('type', None)
|
|
self.regex = config.get('regex', [])
|
|
self.repeat = config.get('repeat', 1)
|
|
self.ordered = config.get('ordered', True)
|
|
self.record = config.get('record', {})
|
|
|
|
def process_test(self, line):
|
|
|
|
if self.RUN_PASSED in line:
|
|
if self.fault:
|
|
self.state = "failed"
|
|
else:
|
|
self.state = "passed"
|
|
|
|
if self.RUN_FAILED in line:
|
|
self.state = "failed"
|
|
|
|
if self.fail_on_fault:
|
|
if self.FAULT == line:
|
|
self.fault = True
|
|
|
|
if self.GCOV_START in line:
|
|
self.capture_coverage = True
|
|
elif self.GCOV_END in line:
|
|
self.capture_coverage = False
|
|
|
|
class Console(Harness):
|
|
|
|
def configure(self, instance):
|
|
super(Console, self).configure(instance)
|
|
if self.type == "one_line":
|
|
self.pattern = re.compile(self.regex[0])
|
|
elif self.type == "multi_line":
|
|
self.patterns = []
|
|
for r in self.regex:
|
|
self.patterns.append(re.compile(r))
|
|
|
|
def handle(self, line):
|
|
if self.type == "one_line":
|
|
if self.pattern.search(line):
|
|
self.state = "passed"
|
|
elif self.type == "multi_line" and self.ordered:
|
|
if (self.next_pattern < len(self.patterns) and
|
|
self.patterns[self.next_pattern].search(line)):
|
|
self.next_pattern += 1
|
|
if self.next_pattern >= len(self.patterns):
|
|
self.state = "passed"
|
|
elif self.type == "multi_line" and not self.ordered:
|
|
for i, pattern in enumerate(self.patterns):
|
|
r = self.regex[i]
|
|
if pattern.search(line) and not r in self.matches:
|
|
self.matches[r] = line
|
|
if len(self.matches) == len(self.regex):
|
|
self.state = "passed"
|
|
|
|
if self.fail_on_fault:
|
|
if self.FAULT in line:
|
|
self.fault = True
|
|
|
|
if self.GCOV_START in line:
|
|
self.capture_coverage = True
|
|
elif self.GCOV_END in line:
|
|
self.capture_coverage = False
|
|
|
|
|
|
if self.record:
|
|
pattern = re.compile(self.record.get("regex", ""))
|
|
match = pattern.search(line)
|
|
if match:
|
|
csv = []
|
|
if not self.fieldnames:
|
|
for k,v in match.groupdict().items():
|
|
self.fieldnames.append(k)
|
|
|
|
for k,v in match.groupdict().items():
|
|
csv.append(v.strip())
|
|
self.recording.append(csv)
|
|
|
|
self.process_test(line)
|
|
|
|
if self.state == "passed":
|
|
self.tests[self.id] = "PASS"
|
|
else:
|
|
self.tests[self.id] = "FAIL"
|
|
|
|
class Pytest(Harness):
|
|
def configure(self, instance):
|
|
super(Pytest, self).configure(instance)
|
|
self.running_dir = instance.build_dir
|
|
self.source_dir = instance.testcase.source_dir
|
|
self.pytest_root = 'pytest'
|
|
self.is_pytest = True
|
|
config = instance.testcase.harness_config
|
|
|
|
if config:
|
|
self.pytest_root = config.get('pytest_root', 'pytest')
|
|
|
|
def handle(self, line):
|
|
''' Test cases that make use of pytest more care about results given
|
|
by pytest tool which is called in pytest_run(), so works of this
|
|
handle is trying to give a PASS or FAIL to avoid timeout, nothing
|
|
is writen into handler.log
|
|
'''
|
|
self.state = "passed"
|
|
self.tests[self.id] = "PASS"
|
|
|
|
def pytest_run(self, log_file):
|
|
''' To keep artifacts of pytest in self.running_dir, pass this directory
|
|
by "--cmdopt". On pytest end, add a command line option and provide
|
|
the cmdopt through a fixture function
|
|
If pytest harness report failure, twister will direct user to see
|
|
handler.log, this method writes test result in handler.log
|
|
'''
|
|
cmd = [
|
|
'pytest',
|
|
'-s',
|
|
os.path.join(self.source_dir, self.pytest_root),
|
|
'--cmdopt',
|
|
self.running_dir,
|
|
'--junit-xml',
|
|
os.path.join(self.running_dir, 'report.xml'),
|
|
'-q'
|
|
]
|
|
|
|
log = open(log_file, "a")
|
|
outs = []
|
|
errs = []
|
|
|
|
with subprocess.Popen(cmd,
|
|
stdout = subprocess.PIPE,
|
|
stderr = subprocess.PIPE) as proc:
|
|
try:
|
|
outs, errs = proc.communicate()
|
|
tree = ET.parse(os.path.join(self.running_dir, "report.xml"))
|
|
root = tree.getroot()
|
|
for child in root:
|
|
if child.tag == 'testsuite':
|
|
if child.attrib['failures'] != '0':
|
|
self.state = "failed"
|
|
elif child.attrib['skipped'] != '0':
|
|
self.state = "skipped"
|
|
elif child.attrib['errors'] != '0':
|
|
self.state = "errors"
|
|
else:
|
|
self.state = "passed"
|
|
except subprocess.TimeoutExpired:
|
|
proc.kill()
|
|
self.state = "failed"
|
|
except ET.ParseError:
|
|
self.state = "failed"
|
|
except IOError:
|
|
log.write("Can't access report.xml\n")
|
|
self.state = "failed"
|
|
|
|
if self.state == "passed":
|
|
self.tests[self.id] = "PASS"
|
|
log.write("Pytest cases passed\n")
|
|
elif self.state == "skipped":
|
|
self.tests[self.id] = "SKIP"
|
|
log.write("Pytest cases skipped\n")
|
|
log.write("Please refer report.xml for detail")
|
|
else:
|
|
self.tests[self.id] = "FAIL"
|
|
log.write("Pytest cases failed\n")
|
|
|
|
log.write("\nOutput from pytest:\n")
|
|
log.write(outs.decode('UTF-8'))
|
|
log.write(errs.decode('UTF-8'))
|
|
log.close()
|
|
|
|
|
|
class Test(Harness):
|
|
RUN_PASSED = "PROJECT EXECUTION SUCCESSFUL"
|
|
RUN_FAILED = "PROJECT EXECUTION FAILED"
|
|
|
|
def handle(self, line):
|
|
match = result_re.match(line)
|
|
if match and match.group(2):
|
|
name = "{}.{}".format(self.id, match.group(3))
|
|
self.tests[name] = match.group(1)
|
|
self.ztest = True
|
|
|
|
self.process_test(line)
|
|
|
|
if not self.ztest and self.state:
|
|
if self.state == "passed":
|
|
self.tests[self.id] = "PASS"
|
|
else:
|
|
self.tests[self.id] = "FAIL"
|
|
|
|
|
|
class Ztest(Test):
|
|
pass
|