83 lines
2.0 KiB
Python
83 lines
2.0 KiB
Python
from __future__ import print_function
|
|
|
|
import colorama
|
|
import logging
|
|
|
|
from ansiblelater.utils import info, warn, abort, error, should_do_markup
|
|
|
|
|
|
def test_abort(capsys, mocker):
|
|
abort('foo')
|
|
stdout, _ = capsys.readouterr()
|
|
|
|
print('{}{}{}'.format(colorama.Fore.RED, 'FATAL: foo'.rstrip(),
|
|
colorama.Style.RESET_ALL))
|
|
x, _ = capsys.readouterr()
|
|
|
|
assert x == stdout
|
|
|
|
|
|
def test_error(capsys, mocker):
|
|
error('foo')
|
|
stdout, _ = capsys.readouterr()
|
|
|
|
print('{}{}{}'.format(colorama.Fore.RED, 'ERROR: foo'.rstrip(),
|
|
colorama.Style.RESET_ALL))
|
|
x, _ = capsys.readouterr()
|
|
|
|
assert x == stdout
|
|
|
|
|
|
def test_warn(capsys, mocker):
|
|
settings = mocker.MagicMock()
|
|
settings.log_level = getattr(logging, 'WARNING')
|
|
|
|
warn('foo', settings)
|
|
stdout, _ = capsys.readouterr()
|
|
|
|
print('{}{}{}'.format(colorama.Fore.YELLOW, 'WARN: foo'.rstrip(),
|
|
colorama.Style.RESET_ALL))
|
|
x, _ = capsys.readouterr()
|
|
|
|
assert x == stdout
|
|
|
|
|
|
def test_info(capsys, mocker):
|
|
settings = mocker.MagicMock()
|
|
settings.log_level = getattr(logging, 'INFO')
|
|
|
|
info('foo', settings)
|
|
stdout, _ = capsys.readouterr()
|
|
|
|
print('{}{}{}'.format(colorama.Fore.BLUE, 'INFO: foo'.rstrip(),
|
|
colorama.Style.RESET_ALL))
|
|
x, _ = capsys.readouterr()
|
|
|
|
assert x == stdout
|
|
|
|
|
|
def test_markup_detection_pycolors0(monkeypatch):
|
|
monkeypatch.setenv('PY_COLORS', '0')
|
|
assert not should_do_markup()
|
|
|
|
|
|
def test_markup_detection_pycolors1(monkeypatch):
|
|
monkeypatch.setenv('PY_COLORS', '1')
|
|
assert 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 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 should_do_markup()
|
|
mocker.resetall()
|
|
mocker.stopall()
|