shell bypass 403
# Copyright (c) 2016, Matt Layman
try:
from cStringIO import StringIO
except ImportError: # pragma: no cover
from io import StringIO
import sys
from py.io import TerminalWriter
import pytest
from tap.formatter import format_as_diagnostics
from tap.i18n import _
from tap.tracker import Tracker
# Because of how pytest hooks work, there is not much choice
# except to use module level state. Ugh.
tracker = Tracker()
def pytest_addoption(parser):
"""Include all the command line options."""
group = parser.getgroup('terminal reporting', after='general')
group.addoption(
'--tap-stream', default=False, action='store_true', help=_(
'Stream TAP output instead of the default test runner output.'))
group.addoption(
'--tap-files', default=False, action='store_true', help=_(
'Store all TAP test results into individual files per test case.'))
group.addoption(
'--tap-combined', default=False, action='store_true', help=_(
'Store all TAP test results into a combined output file.'))
group.addoption('--tap-outdir', metavar='path', help=_(
'An optional output directory to write TAP files to. '
'If the directory does not exist, it will be created.'))
@pytest.mark.trylast
def pytest_configure(config):
"""Set all the options before the test run."""
tracker.outdir = config.option.tap_outdir
tracker.combined = config.option.tap_combined
if config.option.tap_stream:
reporter = config.pluginmanager.getplugin('terminalreporter')
config.pluginmanager.unregister(reporter)
tracker.streaming = True
tracker.stream = sys.stdout
def pytest_runtest_logreport(report):
"""Add a test result to the tracker."""
if report.when != 'call':
return
description = report.location[2]
testcase = description.split('.', 1)[0]
if report.outcome == 'passed':
tracker.add_ok(testcase, description)
elif report.outcome == 'failed':
diagnostics = _make_as_diagnostics(report)
tracker.add_not_ok(testcase, description, diagnostics=diagnostics)
elif report.outcome == 'skipped':
reason = report.longrepr[2].split(':', 1)[1].strip()
tracker.add_skip(testcase, description, reason)
def _make_as_diagnostics(report):
"""Format a report as TAP diagnostic output."""
out = StringIO()
tw = TerminalWriter(file=out)
report.toterminal(tw)
lines = out.getvalue().splitlines(True)
return format_as_diagnostics(lines)
def pytest_unconfigure(config):
"""Dump the results."""
if (
config.option.tap_stream or
config.option.tap_files or
config.option.tap_combined
):
tracker.generate_tap_reports()