diff options
author | Martin Matusiak <numerodix@gmail.com> | 2021-05-05 10:51:12 +1000 |
---|---|---|
committer | Martin Matusiak <numerodix@gmail.com> | 2021-05-05 10:51:12 +1000 |
commit | ac75cfe3b1f008afe1c1c1e2d8feaa4c73126c8c (patch) | |
tree | 610b2afb433ac287b66a5f07482174343f906a95 | |
parent | f50e32985be723a42e84ceb03a604cb3e6fb1eb5 (diff) | |
download | ansicolor-ac75cfe3b1f008afe1c1c1e2d8feaa4c73126c8c.tar.gz |
add tests for disabled state
-rw-r--r-- | tests/test_colors.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/tests/test_colors.py b/tests/test_colors.py index 5bfc523..c71e7a1 100644 --- a/tests/test_colors.py +++ b/tests/test_colors.py @@ -1,3 +1,5 @@ +import contextlib + from ansicolor import Colors from ansicolor import blue from ansicolor import colordiff @@ -10,6 +12,19 @@ from ansicolor import strip_escapes from ansicolor import wrap_string from ansicolor import get_code_v2 from ansicolor import colorize_v2 +import ansicolor + + +@contextlib.contextmanager +def disabled_state(): + ansicolor.ansicolor._disabled = True + + try: + # run the test with colors disabled + yield + + finally: + ansicolor.ansicolor._disabled = False def test_codes(): @@ -35,6 +50,10 @@ def test_codes(): # bold + reverse color assert '\033[1;7;31m' == get_code(Colors.Red, bold=True, reverse=True) +def test_codes_disabled(): + with disabled_state(): + assert '' == get_code(Colors.Black) + def test_codes_v2(): # reset code assert '\033[0m' == get_code_v2(None) @@ -70,6 +89,10 @@ def test_codes_v2(): assert '\033[1;5;7;31m' == get_code_v2(Colors.Red, bold=True, blink=True, reverse=True) assert '\033[1;4;5;7;31m' == get_code_v2(Colors.Red, bold=True, underline=True, blink=True, reverse=True) +def test_codes_v2_disabled(): + with disabled_state(): + assert '' == get_code_v2(Colors.Black) + def test_coloring(): assert '\033[0;0;31m' + 'hi' + '\033[0;0m' == red('hi') |