-
Notifications
You must be signed in to change notification settings - Fork 236
Expand file tree
/
Copy pathtest_progress.py
More file actions
258 lines (217 loc) · 7.85 KB
/
test_progress.py
File metadata and controls
258 lines (217 loc) · 7.85 KB
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
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
"""Tests for tmuxp workspace builder progress callback."""
from __future__ import annotations
import typing as t
import pytest
from libtmux.server import Server
from tmuxp import exc
from tmuxp.workspace.builder import WorkspaceBuilder
def test_builder_on_progress_callback(
server: Server,
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""WorkspaceBuilder calls on_progress at each build milestone."""
monkeypatch.delenv("TMUX", raising=False)
session_config = {
"session_name": "progress-test",
"windows": [{"window_name": "editor", "panes": [{"shell_command": []}]}],
}
calls: list[str] = []
builder = WorkspaceBuilder(
session_config=session_config,
server=server,
on_progress=calls.append,
)
builder.build()
assert any("Session created:" in c for c in calls)
assert any("Creating window:" in c for c in calls)
assert any("Creating pane:" in c for c in calls)
assert "Workspace built" in calls
def test_builder_on_before_script_not_called_without_script(
server: Server,
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""on_before_script callback is not invoked when config has no before_script key."""
monkeypatch.delenv("TMUX", raising=False)
session_config = {
"session_name": "no-script-callback-test",
"windows": [{"window_name": "main", "panes": [{"shell_command": []}]}],
}
called: list[bool] = []
builder = WorkspaceBuilder(
session_config=session_config,
server=server,
on_before_script=lambda: called.append(True),
)
builder.build()
assert called == []
def test_builder_on_script_output_not_called_without_script(
server: Server,
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""on_script_output callback is not invoked when config has no before_script key."""
monkeypatch.delenv("TMUX", raising=False)
session_config = {
"session_name": "no-script-output-test",
"windows": [{"window_name": "main", "panes": [{"shell_command": []}]}],
}
lines: list[str] = []
builder = WorkspaceBuilder(
session_config=session_config,
server=server,
on_script_output=lines.append,
)
builder.build()
assert lines == []
def test_builder_on_build_event_sequence(
server: Server,
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""on_build_event fires the full event sequence during build()."""
monkeypatch.delenv("TMUX", raising=False)
session_config = {
"session_name": "build-event-test",
"windows": [
{
"window_name": "editor",
"panes": [{"shell_command": []}, {"shell_command": []}],
},
{"window_name": "logs", "panes": [{"shell_command": []}]},
],
}
events: list[dict[str, t.Any]] = []
builder = WorkspaceBuilder(
session_config=session_config,
server=server,
on_build_event=events.append,
)
builder.build()
event_types = [e["event"] for e in events]
assert event_types[0] == "session_created"
assert event_types[-1] == "workspace_built"
assert event_types.count("window_started") == 2
assert event_types.count("window_done") == 2
assert event_types.count("pane_creating") == 3 # 2 panes + 1 pane
created = next(e for e in events if e["event"] == "session_created")
assert created["window_total"] == 2
assert created["session_pane_total"] == 3 # 2 panes + 1 pane
def test_builder_on_build_event_session_name(
server: Server,
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""session_created event carries correct session name."""
monkeypatch.delenv("TMUX", raising=False)
session_config = {
"session_name": "name-check",
"windows": [{"window_name": "main", "panes": [{"shell_command": []}]}],
}
events: list[dict[str, t.Any]] = []
builder = WorkspaceBuilder(
session_config=session_config,
server=server,
on_build_event=events.append,
)
builder.build()
created = next(e for e in events if e["event"] == "session_created")
assert created["name"] == "name-check"
assert created["window_total"] == 1
def test_builder_on_build_event_session_pane_total(
server: Server,
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""session_created event includes session_pane_total summing all windows' panes."""
monkeypatch.delenv("TMUX", raising=False)
pane: dict[str, list[object]] = {"shell_command": []}
session_config = {
"session_name": "pane-total-test",
"windows": [
{"window_name": "w1", "panes": [pane, pane]},
{"window_name": "w2", "panes": [pane]},
{"window_name": "w3", "panes": [pane, pane, pane]},
],
}
events: list[dict[str, t.Any]] = []
builder = WorkspaceBuilder(
session_config=session_config,
server=server,
on_build_event=events.append,
)
builder.build()
created = next(e for e in events if e["event"] == "session_created")
assert created["session_pane_total"] == 6 # 2 + 1 + 3
def test_builder_before_script_events(
server: Server,
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""before_script_started fires before run; before_script_done fires in finally."""
monkeypatch.delenv("TMUX", raising=False)
session_config = {
"session_name": "before-script-events-test",
"before_script": "echo hello",
"windows": [{"window_name": "main", "panes": [{"shell_command": []}]}],
}
events: list[dict[str, t.Any]] = []
builder = WorkspaceBuilder(
session_config=session_config,
server=server,
on_build_event=events.append,
)
builder.build()
event_types = [e["event"] for e in events]
assert "before_script_started" in event_types
assert "before_script_done" in event_types
bs_start_idx = event_types.index("before_script_started")
bs_done_idx = event_types.index("before_script_done")
win_idx = event_types.index("window_started")
assert bs_start_idx < bs_done_idx < win_idx
def test_builder_before_script_done_fires_on_failure(
server: Server,
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""before_script_done fires in finally even when the script fails."""
monkeypatch.delenv("TMUX", raising=False)
session_config = {
"session_name": "before-script-fail-test",
"before_script": "/bin/false",
"windows": [{"window_name": "main", "panes": [{"shell_command": []}]}],
}
events: list[dict[str, t.Any]] = []
builder = WorkspaceBuilder(
session_config=session_config,
server=server,
on_build_event=events.append,
)
with pytest.raises(exc.BeforeLoadScriptError):
builder.build()
event_types = [e["event"] for e in events]
assert "before_script_started" in event_types
assert "before_script_done" in event_types
def test_builder_on_build_event_pane_numbers(
server: Server,
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""pane_creating events carry 1-based pane_num and correct pane_total."""
monkeypatch.delenv("TMUX", raising=False)
session_config = {
"session_name": "pane-num-test",
"windows": [
{
"window_name": "main",
"panes": [
{"shell_command": []},
{"shell_command": []},
{"shell_command": []},
],
},
],
}
events: list[dict[str, t.Any]] = []
builder = WorkspaceBuilder(
session_config=session_config,
server=server,
on_build_event=events.append,
)
builder.build()
pane_events = [e for e in events if e["event"] == "pane_creating"]
assert len(pane_events) == 3
assert [e["pane_num"] for e in pane_events] == [1, 2, 3]
assert all(e["pane_total"] == 3 for e in pane_events)