1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
# Copyright (C) 2020 Konsulko Group
# Author: Edi Feschiyan
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import pytest
import argparse
import time
class LavaAction(argparse.Action):
def __init__(self, option_strings, dest, nargs=0, **kwargs):
if nargs != 0:
raise ValueError("nargs not allowed")
super(LavaAction, self).__init__(option_strings, dest, nargs=nargs, **kwargs)
def __call__(self, parser, namespace, values, option_string=None):
setattr(namespace, self.dest, True)
setattr(namespace, 'color', 'no')
def pytest_addoption(parser):
parser.addoption('-L', '--lava', action=LavaAction, help='enable LAVA signals')
def pytest_configure(config):
# Force normal progress and verbose output off when doing LAVA output
terminal = config.pluginmanager.getplugin('terminal')
class QuietReporter(terminal.TerminalReporter):
def _determine_show_progress_info(self):
return False
@property
def verbosity(self):
return 0
@property
def showlongtestinfo(self):
return False
@property
def showfspath(self):
return False
if config.getoption('lava'):
terminal.TerminalReporter = QuietReporter
def lava_result_convert(pytest_outcome):
""" Convert the pytest outcome to the string expected by LAVA."""
if pytest_outcome == 'passed':
return 'pass'
elif pytest_outcome == 'skipped':
return 'skip'
elif pytest_outcome == 'xfailed':
return 'pass'
else:
return 'fail'
def pytest_report_teststatus(config, report):
""" Insert strings that LAVA expects to capture test results."""
done = False
if config.getoption('lava'):
# Convert pytest test file and name into a LAVA test name
test_file = report.location[0].split('/')[-1]
test_file = test_file.replace('test_', '', 1)
if test_file.endswith('.py'):
test_file = test_file[:-3]
test_name = test_file + '_' + report.location[2][5:]
test_result = lava_result_convert(report.outcome)
# Generate expected LAVA testcase output
if report.when == 'setup':
if report.outcome == 'skipped':
done = True
elif report.when == 'call':
done = True
if report.outcome == 'failed':
print(f'<LAVA_SIGNAL_STARTTC {test_name}>')
print('ERROR:\n')
print(report.longrepr)
print(f'<LAVA_SIGNAL_ENDTC {test_name}>')
if done:
print(f'<LAVA_SIGNAL_TESTCASE TEST_CASE_ID={test_name} RESULT={test_result}>\n')
# Delay to slow down serial output for LAVA
time.sleep(0.25)
# Quiet short result output
category, short, verbose = '', '', ''
if hasattr(report, 'wasxfail'):
if report.skipped:
category = 'xfailed'
elif report.passed:
category = 'xpassed'
return (category, short, verbose)
elif report.when in ('setup', 'teardown'):
if report.failed:
category = 'error'
elif report.skipped:
category = 'skipped'
return (category, short, verbose)
category = report.outcome
return (category, short, verbose)
|
