freeipa/ipatests/pytest_plugins/nose_compat.py
Petr Viktorin 48de128571 Add local pytest plugin for --with-xunit and --logging-level
The --with-xunit option ihas the same behavior as in nosetests:
it's an alias for pytest's --junitxml=nosetests.py

The --logging-level option enables direct IPA logging to stdout.

https://fedorahosted.org/freeipa/ticket/4610

Reviewed-By: Tomas Babej <tbabej@redhat.com>
2014-11-21 12:14:44 +01:00

74 lines
2.9 KiB
Python

# Authors:
# Petr Viktorin <pviktori@redhat.com>
#
# Copyright (C) 2014 Red Hat
# see file 'COPYING' for use and warranty information
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""Provides command-line options for very limited Nose compatibility"""
import os
import sys
import logging
from ipapython.ipa_log_manager import log_mgr
def pytest_addoption(parser):
group = parser.getgroup("IPA nosetests compatibility shim")
group.addoption('--with-xunit', action="store_const",
dest="xmlpath", metavar="path", default=None,
const=os.environ.get('IPATEST_XUNIT_PATH', './nosetests.xml'),
help="create junit-xml style report file at $IPATEST_XUNIT_PATH,"
"or nosetests.xml by default")
group.addoption('--logging-level', action="store",
dest="logging_level", metavar="level", default='CRITICAL',
help="level for logging to stderr. "
"Bypasses pytest logging redirection."
"May be used to show progress of long-running tests.")
def pytest_configure(config):
if config.getoption('logging_level'):
# Forward IPA logging to a normal Python logger. Nose's logcapture plugin
# can't work with IPA-managed loggers
class LogHandler(logging.Handler):
name = 'forwarding log handler'
logger = logging.getLogger('IPA')
def emit(self, record):
capture = config.pluginmanager.getplugin('capturemanager')
orig_stdout, orig_stderr = sys.stdout, sys.stderr
if capture:
capture._capturing.suspend_capturing()
sys.stderr.write(self.format(record))
sys.stderr.write('\n')
if capture:
capture._capturing.resume_capturing()
sys.stdout, sys.stderr = orig_stdout, orig_stderr
log_mgr.configure(
{
'default_level': config.getoption('logging_level'),
'handlers': [{'log_handler': LogHandler(),
'format': '[%(name)s] %(message)s',
'level': 'debug'},
{'level': 'debug',
'name': 'real_stderr',
'stream': sys.stderr}]},
configure_state='tests')