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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
# coding=utf-8
# flake8: noqa E302
"""
Unit testing for cmd2/ansi.py module
"""
import pytest
import cmd2.ansi as ansi
HELLO_WORLD = 'Hello, world!'
def test_strip_style():
base_str = HELLO_WORLD
ansi_str = ansi.style(base_str, fg='green')
assert base_str != ansi_str
assert base_str == ansi.strip_style(ansi_str)
def test_style_aware_wcswidth():
base_str = HELLO_WORLD
ansi_str = ansi.style(base_str, fg='green')
assert ansi.style_aware_wcswidth(ansi_str) != len(ansi_str)
def test_style_none():
base_str = HELLO_WORLD
ansi_str = base_str
assert ansi.style(base_str) == ansi_str
def test_style_fg():
base_str = HELLO_WORLD
fg_color = 'blue'
ansi_str = ansi.FG_COLORS[fg_color] + base_str + ansi.FG_RESET
assert ansi.style(base_str, fg=fg_color) == ansi_str
def test_style_bg():
base_str = HELLO_WORLD
bg_color = 'green'
ansi_str = ansi.BG_COLORS[bg_color] + base_str + ansi.BG_RESET
assert ansi.style(base_str, bg=bg_color) == ansi_str
def test_style_bold():
base_str = HELLO_WORLD
ansi_str = ansi.INTENSITY_BRIGHT + base_str + ansi.INTENSITY_NORMAL
assert ansi.style(base_str, bold=True) == ansi_str
def test_style_dim():
base_str = HELLO_WORLD
ansi_str = ansi.INTENSITY_DIM + base_str + ansi.INTENSITY_NORMAL
assert ansi.style(base_str, dim=True) == ansi_str
def test_style_underline():
base_str = HELLO_WORLD
ansi_str = ansi.UNDERLINE_ENABLE + base_str + ansi.UNDERLINE_DISABLE
assert ansi.style(base_str, underline=True) == ansi_str
def test_style_multi():
base_str = HELLO_WORLD
fg_color = 'blue'
bg_color = 'green'
ansi_str = (ansi.FG_COLORS[fg_color] + ansi.BG_COLORS[bg_color] +
ansi.INTENSITY_BRIGHT + ansi.INTENSITY_DIM + ansi.UNDERLINE_ENABLE +
base_str +
ansi.FG_RESET + ansi.BG_RESET +
ansi.INTENSITY_NORMAL + ansi.INTENSITY_NORMAL + ansi.UNDERLINE_DISABLE)
assert ansi.style(base_str, fg=fg_color, bg=bg_color, bold=True, dim=True, underline=True) == ansi_str
def test_style_color_not_exist():
base_str = HELLO_WORLD
with pytest.raises(ValueError):
ansi.style(base_str, fg='fake', bg='green')
with pytest.raises(ValueError):
ansi.style(base_str, fg='blue', bg='fake')
def test_fg_lookup_exist():
fg_color = 'green'
assert ansi.fg_lookup(fg_color) == ansi.FG_COLORS[fg_color]
def test_fg_lookup_nonexist():
with pytest.raises(ValueError):
ansi.fg_lookup('foo')
def test_bg_lookup_exist():
bg_color = 'green'
assert ansi.bg_lookup(bg_color) == ansi.BG_COLORS[bg_color]
def test_bg_lookup_nonexist():
with pytest.raises(ValueError):
ansi.bg_lookup('bar')
def test_set_title_str():
OSC = '\033]'
BEL = '\007'
title = HELLO_WORLD
assert ansi.set_title_str(title) == OSC + '2;' + title + BEL
@pytest.mark.parametrize('cols, prompt, line, cursor, msg, expected', [
(127, '(Cmd) ', 'help his', 12, ansi.style('Hello World!', fg='magenta'), '\x1b[2K\r\x1b[35mHello World!\x1b[39m'),
(127, '\n(Cmd) ', 'help ', 5, 'foo', '\x1b[2K\x1b[1A\x1b[2K\rfoo'),
(10, '(Cmd) ', 'help history of the american republic', 4, 'boo', '\x1b[3B\x1b[2K\x1b[1A\x1b[2K\x1b[1A\x1b[2K\x1b[1A\x1b[2K\x1b[1A\x1b[2K\rboo')
])
def test_async_alert_str(cols, prompt, line, cursor, msg, expected):
alert_str = ansi.async_alert_str(terminal_columns=cols, prompt=prompt, line=line, cursor_offset=cursor,
alert_msg=msg)
assert alert_str == expected
def test_fg_enum():
assert ansi.fg_lookup('bright_red') == ansi.fg_lookup(ansi.fg.bright_red)
def test_fg_enum_to_str():
assert str(ansi.fg.black) == ansi.fg_lookup('black')
def test_bg_enum():
assert ansi.bg_lookup('green') == ansi.bg_lookup(ansi.bg.green)
def test_bg_enum_to_str():
assert str(ansi.bg.blue) == ansi.bg_lookup('blue')
|
