99 lines
2.3 KiB
Python
99 lines
2.3 KiB
Python
"""Test logging module."""
|
|
|
|
from __future__ import print_function
|
|
|
|
import colorama
|
|
|
|
from ansiblelater import logger
|
|
|
|
|
|
def test_flag_extra():
|
|
extra = {}
|
|
extra.update(foo="bar")
|
|
|
|
flagged = logger.flag_extra(extra)
|
|
|
|
assert flagged.get("later_foo") == "bar"
|
|
|
|
|
|
def test_critical(capsys, mocker):
|
|
log = logger.get_logger("test_critical")
|
|
log.critical("foo")
|
|
_, stderr = capsys.readouterr()
|
|
|
|
print(
|
|
f"{colorama.Fore.RED}{colorama.Style.BRIGHT}CRITICAL:{colorama.Style.NORMAL} foo\n"
|
|
f"{colorama.Style.RESET_ALL}"
|
|
)
|
|
x, _ = capsys.readouterr()
|
|
|
|
assert x == stderr
|
|
|
|
|
|
def test_error(capsys, mocker):
|
|
log = logger.get_logger("test_error")
|
|
log.error("foo")
|
|
_, stderr = capsys.readouterr()
|
|
|
|
print(
|
|
f"{colorama.Fore.RED}{colorama.Style.BRIGHT}ERROR:{colorama.Style.NORMAL} foo\n"
|
|
f"{colorama.Style.RESET_ALL}"
|
|
)
|
|
x, _ = capsys.readouterr()
|
|
|
|
assert x == stderr
|
|
|
|
|
|
def test_warn(capsys, mocker):
|
|
log = logger.get_logger("test_warn")
|
|
log.warning("foo")
|
|
stdout, _ = capsys.readouterr()
|
|
|
|
print(
|
|
f"{colorama.Fore.YELLOW}{colorama.Style.BRIGHT}WARNING:{colorama.Style.NORMAL} foo\n"
|
|
f"{colorama.Style.RESET_ALL}"
|
|
)
|
|
x, _ = capsys.readouterr()
|
|
|
|
assert x == stdout
|
|
|
|
|
|
def test_info(capsys, mocker):
|
|
log = logger.get_logger("test_info")
|
|
log.info("foo")
|
|
stdout, _ = capsys.readouterr()
|
|
|
|
print(
|
|
f"{colorama.Fore.BLUE}{colorama.Style.BRIGHT}INFO:{colorama.Style.NORMAL} foo\n"
|
|
f"{colorama.Style.RESET_ALL}"
|
|
)
|
|
x, _ = capsys.readouterr()
|
|
|
|
assert x == stdout
|
|
|
|
|
|
def test_markup_detection_pycolors0(monkeypatch):
|
|
monkeypatch.setenv("PY_COLORS", "0")
|
|
assert not logger._should_do_markup()
|
|
|
|
|
|
def test_markup_detection_pycolors1(monkeypatch):
|
|
monkeypatch.setenv("PY_COLORS", "1")
|
|
assert logger._should_do_markup()
|
|
|
|
|
|
def test_markup_detection_tty_yes(mocker):
|
|
mocker.patch("sys.stdout.isatty", return_value=True)
|
|
mocker.patch("os.environ", {"TERM": "xterm"})
|
|
assert logger._should_do_markup()
|
|
mocker.resetall()
|
|
mocker.stopall()
|
|
|
|
|
|
def test_markup_detection_tty_no(mocker):
|
|
mocker.patch("os.environ", {})
|
|
mocker.patch("sys.stdout.isatty", return_value=False)
|
|
assert not logger._should_do_markup()
|
|
mocker.resetall()
|
|
mocker.stopall()
|