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
|
from __future__ import annotations
from typing import TYPE_CHECKING
from cleo.ui.progress_indicator import ProgressIndicator
if TYPE_CHECKING:
from typing import Callable
from cleo.io.buffered_io import BufferedIO
def test_default_indicator(ansi_io: BufferedIO, sleep: Callable[[float], None]) -> None:
bar = ProgressIndicator(ansi_io)
bar.start("Starting...")
sleep(0.101)
bar.advance()
sleep(0.101)
bar.advance()
sleep(0.101)
bar.advance()
sleep(0.101)
bar.advance()
sleep(0.101)
bar.advance()
sleep(0.101)
bar.set_message("Advancing...")
bar.advance()
bar.finish("Done...")
bar.start("Starting Again...")
sleep(0.101)
bar.advance()
bar.finish("Done Again...")
bar.start("Starting Again...")
sleep(0.101)
bar.advance()
bar.finish("Done Again...", reset_indicator=True)
output = [
" - Starting...",
" \\ Starting...",
" | Starting...",
" / Starting...",
" - Starting...",
" \\ Starting...",
" \\ Advancing...",
" | Advancing...",
" | Done...",
]
expected = "\x0D\x1B[2K" + "\x0D\x1B[2K".join(output)
expected += "\n"
output = [" - Starting Again...", " \\ Starting Again...", " \\ Done Again..."]
expected += "\x0D\x1B[2K" + "\x0D\x1B[2K".join(output)
expected += "\n"
output = [" - Starting Again...", " \\ Starting Again...", " - Done Again..."]
expected += "\x0D\x1B[2K" + "\x0D\x1B[2K".join(output)
expected += "\n"
assert expected == ansi_io.fetch_error()
def test_explicit_format(ansi_io: BufferedIO, sleep: Callable[[float], None]) -> None:
bar = ProgressIndicator(ansi_io, ProgressIndicator.NORMAL)
bar.start("Starting...")
sleep(0.101)
bar.advance()
sleep(0.101)
bar.advance()
sleep(0.101)
bar.advance()
sleep(0.101)
bar.advance()
sleep(0.101)
bar.advance()
sleep(0.101)
bar.set_message("Advancing...")
bar.advance()
bar.finish("Done...")
bar.start("Starting Again...")
sleep(0.101)
bar.advance()
bar.finish("Done Again...")
bar.start("Starting Again...")
sleep(0.101)
bar.advance()
bar.finish("Done Again...", reset_indicator=True)
output = [
" - Starting...",
" \\ Starting...",
" | Starting...",
" / Starting...",
" - Starting...",
" \\ Starting...",
" \\ Advancing...",
" | Advancing...",
" | Done...",
]
expected = "\x0D\x1B[2K" + "\x0D\x1B[2K".join(output)
expected += "\n"
output = [" - Starting Again...", " \\ Starting Again...", " \\ Done Again..."]
expected += "\x0D\x1B[2K" + "\x0D\x1B[2K".join(output)
expected += "\n"
output = [" - Starting Again...", " \\ Starting Again...", " - Done Again..."]
expected += "\x0D\x1B[2K" + "\x0D\x1B[2K".join(output)
expected += "\n"
assert expected == ansi_io.fetch_error()
|
