blob: 3a8f7236baf2b16a19801f3334bee282fb926ee3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
from unittest.mock import MagicMock, patch
import colorama
import pytest
from hypothesis_auto import auto_pytest_magic
import isort.format
auto_pytest_magic(isort.format.show_unified_diff, auto_allow_exceptions_=(UnicodeEncodeError,))
def test_ask_whether_to_apply_changes_to_file():
with patch("isort.format.input", MagicMock(return_value="y")):
assert isort.format.ask_whether_to_apply_changes_to_file("")
with patch("isort.format.input", MagicMock(return_value="n")):
assert not isort.format.ask_whether_to_apply_changes_to_file("")
with patch("isort.format.input", MagicMock(return_value="q")):
with pytest.raises(SystemExit):
assert isort.format.ask_whether_to_apply_changes_to_file("")
def test_basic_printer(capsys):
printer = isort.format.create_terminal_printer(color=False)
printer.success("All good!")
out, _ = capsys.readouterr()
assert out == "SUCCESS: All good!\n"
printer.error("Some error")
out, _ = capsys.readouterr()
assert out == "ERROR: Some error\n"
def test_colored_printer_success(capsys):
printer = isort.format.create_terminal_printer(color=True)
printer.success("All good!")
out, _ = capsys.readouterr()
assert "SUCCESS" in out
assert "All good!" in out
assert colorama.Fore.GREEN in out
def test_colored_printer_error(capsys):
printer = isort.format.create_terminal_printer(color=True)
printer.error("Some error")
out, _ = capsys.readouterr()
assert "ERROR" in out
assert "Some error" in out
assert colorama.Fore.RED in out
@patch("isort.format.colorama_unavailable", True)
def test_colorama_not_available_handled_gracefully(capsys):
with pytest.raises(SystemExit) as system_exit:
_ = isort.format.create_terminal_printer(color=True)
assert system_exit.value.code > 0
_, err = capsys.readouterr()
assert "colorama" in err
assert "colors extra" in err
|
