import logging
import sys
import os
import colors
import re
import pytest
from _pytest.capture import safe_text_dupfile
from ..step import steps
logging.basicConfig(
level=logging.INFO,
format='%(levelname)7s: %(message)s',
stream=sys.stderr,
)
[docs]class StepReporter:
[docs] def __init__(self, terminalreporter, *, rewrite=False):
self.tr = terminalreporter
# copy the original stdout for use with CaptureFixture
self.tr._tw._file = safe_text_dupfile(self.tr._tw._file, mode=self.tr._tw._file.mode)
self.rewrite = rewrite
self.__reset()
steps.subscribe(self.notify)
def __reset(self):
self.cur_step = None
self.cur_resource = None
self.elements = []
def __commit(self):
if self.cur_step and self.rewrite:
self.tr._tw.write('\n')
self.tr._tw._lastlen = 0
self.__reset()
def __merge_element(self, key, value):
if self.elements and self.elements[-1][0] == key:
self.elements.pop()
if value is not None:
self.elements.append((key, value))
def __format_elements(self):
return [
"{}={}".format(colors.color(k, style='underline'), repr(v)) for k, v in self.elements if v is not None
]
[docs] def notify(self, event):
# ignore tagged events
if event.step.tag:
return
new = not self.rewrite
if self.cur_step is not event.step:
new = True
if self.cur_resource is not event.resource:
new = True
self.cur_resource = event.resource
if new:
self.__commit()
self.cur_step = event.step
self.cur_resource = event.resource
self.elements = []
for k, v in event.data.items():
self.__merge_element(k, v)
self._line_format(event)
def _line_format(self, event):
indent = ' '*event.step.level
line = [indent, colors.color(event.step.title, style='bold')]
if event.resource:
line.append(event.resource)
line.extend(self.__format_elements())
if self.rewrite:
self.tr._tw.reline(" ".join(line))
else:
self.tr._tw.line(" ".join(line))
[docs] @pytest.hookimpl(hookwrapper=True, trylast=True)
def pytest_runtest_logstart(self):
yield
self.tr._tw.write('\n')
self.tr._tw._lastlen = 0
[docs] @pytest.hookimpl(tryfirst=True)
def pytest_runtest_logreport(self, report):
if report.when == "setup":
self.__reset()
elif report.when == "call":
self.__commit()
else:
pass
[docs]class ColoredStepReporter(StepReporter):
EVENT_COLORS_DARK = {
'expect$': 8, # dark gray (a lot of output, blend into background)
'run': 10, # green
'state_': 51, # light blue
'transition$': 45, # blue
'cycle$|on$|off$': 246, # light gray
}
EVENT_COLORS_LIGHT = {
'expect$': 250, # light gray (a lot of output, blend into background)
'run': 10, # green
'state_': 51, # light blue
'transition$': 45, # blue
'cycle$|on$|off$': 8, # dark gray
}
[docs] def __init__(self, terminalreporter, *, rewrite=False):
super().__init__(terminalreporter, rewrite=rewrite)
scheme = os.environ.get('LG_COLOR_SCHEME', 'dark')
if scheme == 'light':
self.color_scheme = ColoredStepReporter.EVENT_COLORS_LIGHT
else:
self.color_scheme = ColoredStepReporter.EVENT_COLORS_DARK
def __event_color(self, event):
for pattern, color in self.color_scheme.items():
if re.match(pattern, event.step.title):
return color
else:
return 'default'
def __format_elements(self, color):
return [
colors.color(k, fg=color, style='underline') + colors.color('={}'.format(repr(v)), fg=color) for k, v in self.elements if v is not None
]
def _line_format(self, event):
indent = ' '*event.step.level
color = self.__event_color(event)
line = [indent, colors.color(event.step.title, fg=color, style='bold')]
if event.resource:
line.append(event.resource)
line.extend(self.__format_elements(color))
if self.rewrite:
self.tr._tw.reline(" ".join(line))
else:
self.tr._tw.line(" ".join(line))