|
6 | 6 | import pathlib |
7 | 7 | import time |
8 | 8 | import typing as t |
| 9 | +from typing import cast |
9 | 10 |
|
10 | 11 | import pytest |
11 | 12 |
|
12 | 13 | from libtmux import exc |
13 | 14 | from libtmux._internal.engines.base import ExitStatus |
14 | | -from libtmux._internal.engines.control_mode import ControlModeEngine |
| 15 | +from libtmux._internal.engines.control_mode import ControlModeEngine, _ControlProcess |
15 | 16 | from libtmux._internal.engines.control_protocol import ControlProtocol |
16 | 17 | from libtmux.server import Server |
17 | 18 |
|
@@ -106,7 +107,7 @@ def poll(self) -> int | None: # pragma: no cover - simple stub |
106 | 107 | def fake_start(server_args: t.Sequence[str | int] | None) -> None: |
107 | 108 | engine.tmux_bin = "tmux" |
108 | 109 | engine._server_args = tuple(server_args or ()) |
109 | | - engine.process = fake_process |
| 110 | + engine.process = cast(_ControlProcess, fake_process) |
110 | 111 |
|
111 | 112 | monkeypatch.setattr(engine, "_start_process", fake_start) |
112 | 113 |
|
@@ -144,7 +145,7 @@ def poll(self) -> int | None: |
144 | 145 | def fake_start(server_args: t.Sequence[str | int] | None) -> None: |
145 | 146 | engine.tmux_bin = "tmux" |
146 | 147 | engine._server_args = tuple(server_args or ()) |
147 | | - engine.process = FakeProcess() |
| 148 | + engine.process = cast(_ControlProcess, FakeProcess()) |
148 | 149 |
|
149 | 150 | monkeypatch.setattr(engine, "_start_process", fake_start) |
150 | 151 |
|
@@ -265,7 +266,7 @@ def poll(self) -> int | None: |
265 | 266 | return 0 |
266 | 267 |
|
267 | 268 | engine = ControlModeEngine() |
268 | | - engine.process = FakeProcess() |
| 269 | + engine.process = cast(_ControlProcess, FakeProcess()) |
269 | 270 |
|
270 | 271 | with pytest.raises(case.should_raise): |
271 | 272 | engine._write_line("list-sessions", server_args=()) |
|
0 commit comments