Ziheng Zhang
da74c325d6
* fix(channel): scope stop and restore resume history * refactor(channel): simplify stop and resume patch * Delete PR_MESSAGE.md * fix(channel): address review feedback * fix(channel): address remaining review bugs * fix(channel): clean up stopped request handling * fix(channel): preserve resolved replies and sync tui commands --------- Co-authored-by: Xi Zhang <106144707+X-iZhang@users.noreply.github.com>
91 lines
2.6 KiB
Python
91 lines
2.6 KiB
Python
"""Tests for TUI command-completion state sync."""
|
|
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from EvoScientist.commands.base import CommandContext
|
|
from tests.conftest import run_async as _run
|
|
|
|
pytest.importorskip("textual")
|
|
|
|
|
|
class _Loader:
|
|
def __init__(self) -> None:
|
|
self.adopt_calls: list[object] = []
|
|
|
|
def adopt(self, agent: object) -> None:
|
|
self.adopt_calls.append(agent)
|
|
|
|
|
|
class _StubApp:
|
|
def __init__(self) -> None:
|
|
self._agent_loader = _Loader()
|
|
self._conversation_tid = "thread-1"
|
|
self.model_updates: list[tuple[str, str | None]] = []
|
|
self.refresh_calls: list[bool] = []
|
|
|
|
def update_status_after_model_change(
|
|
self,
|
|
new_model: str,
|
|
new_provider: str | None = None,
|
|
) -> None:
|
|
self.model_updates.append((new_model, new_provider))
|
|
|
|
async def _refresh_status_snapshot(
|
|
self,
|
|
*,
|
|
reset_streaming_text: bool = True,
|
|
) -> None:
|
|
self.refresh_calls.append(reset_streaming_text)
|
|
|
|
|
|
def test_sync_tui_command_completion_adopts_agent_swap(monkeypatch):
|
|
import EvoScientist.cli.tui_interactive as tui_mod
|
|
from EvoScientist import EvoScientist as evosci_mod
|
|
|
|
app = _StubApp()
|
|
ctx = CommandContext(
|
|
agent="new-agent",
|
|
thread_id="thread-1",
|
|
ui=SimpleNamespace(),
|
|
)
|
|
cmd = SimpleNamespace(name="/model")
|
|
|
|
monkeypatch.setattr(
|
|
evosci_mod,
|
|
"_ensure_config",
|
|
lambda: SimpleNamespace(model="gpt-5.5", provider="openai"),
|
|
)
|
|
monkeypatch.setattr(tui_mod, "_channels_is_running", lambda: True)
|
|
monkeypatch.setattr(tui_mod._ch_mod, "_cli_agent", "old-agent", raising=False)
|
|
monkeypatch.setattr(tui_mod._ch_mod, "_cli_thread_id", "old-thread", raising=False)
|
|
|
|
_run(tui_mod._sync_tui_command_completion(app, ctx, "old-agent", cmd))
|
|
|
|
assert app._agent_loader.adopt_calls == ["new-agent"]
|
|
assert app.model_updates == [("gpt-5.5", "openai")]
|
|
assert app.refresh_calls == [True]
|
|
assert tui_mod._ch_mod._cli_agent == "new-agent"
|
|
assert tui_mod._ch_mod._cli_thread_id == "thread-1"
|
|
|
|
|
|
def test_sync_tui_command_completion_refreshes_without_agent_swap(monkeypatch):
|
|
import EvoScientist.cli.tui_interactive as tui_mod
|
|
|
|
app = _StubApp()
|
|
ctx = CommandContext(
|
|
agent="same-agent",
|
|
thread_id="thread-1",
|
|
ui=SimpleNamespace(),
|
|
)
|
|
cmd = SimpleNamespace(name="/compact")
|
|
|
|
monkeypatch.setattr(tui_mod, "_channels_is_running", lambda: False)
|
|
|
|
_run(tui_mod._sync_tui_command_completion(app, ctx, "same-agent", cmd))
|
|
|
|
assert app._agent_loader.adopt_calls == []
|
|
assert app.model_updates == []
|
|
assert app.refresh_calls == [True]
|