wehub-resource-sync
93 lines
2.8 KiB
Python
93 lines
2.8 KiB
Python
"""Unit tests for the REPL's ``/compact`` slash command."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
from omnigent_ui_sdk import RichBlockFormatter
|
|
|
|
from omnigent.repl._repl import COMMANDS, handle_slash_command
|
|
from tests.repl.helpers import CapturingHost
|
|
|
|
|
|
class _CompactSession:
|
|
"""Minimal session stub for ``/compact`` tests."""
|
|
|
|
def __init__(self, *, is_streaming: bool = False, fail: Exception | None = None) -> None:
|
|
self.is_streaming = is_streaming
|
|
self.calls = 0
|
|
self._fail = fail
|
|
|
|
async def compact(self) -> None:
|
|
"""Record the compaction request or raise the configured failure."""
|
|
self.calls += 1
|
|
if self._fail is not None:
|
|
raise self._fail
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_compact_command_registered() -> None:
|
|
"""``/compact`` appears in the slash-command registry."""
|
|
assert "/compact" in COMMANDS
|
|
assert "compact" in COMMANDS["/compact"][0].lower()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_compact_invokes_session_compact() -> None:
|
|
"""The command delegates to the session's explicit compaction hook.
|
|
|
|
Progress messages ("Compacting…" / "Compaction complete.") are now
|
|
delivered via the session SSE stream (response.compaction.in_progress /
|
|
response.compaction.completed events) rather than emitted directly by
|
|
the command handler, so we only assert that the session hook was called.
|
|
"""
|
|
host = CapturingHost()
|
|
session = _CompactSession()
|
|
|
|
await handle_slash_command(
|
|
"/compact",
|
|
session, # type: ignore[arg-type]
|
|
None, # type: ignore[arg-type]
|
|
host,
|
|
RichBlockFormatter(), # type: ignore[arg-type]
|
|
)
|
|
|
|
assert session.calls == 1
|
|
# No direct output — progress comes from the SSE stream.
|
|
assert host.text == ""
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_compact_refuses_while_streaming() -> None:
|
|
"""The command does not request compaction during an active response."""
|
|
host = CapturingHost()
|
|
session = _CompactSession(is_streaming=True)
|
|
|
|
await handle_slash_command(
|
|
"/compact",
|
|
session, # type: ignore[arg-type]
|
|
None, # type: ignore[arg-type]
|
|
host,
|
|
RichBlockFormatter(), # type: ignore[arg-type]
|
|
)
|
|
|
|
assert session.calls == 0
|
|
assert "Cannot compact while a response is running" in host.text
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_compact_surfaces_failure() -> None:
|
|
"""Compaction errors render inline instead of crashing the REPL."""
|
|
host = CapturingHost()
|
|
session = _CompactSession(fail=RuntimeError("boom"))
|
|
|
|
await handle_slash_command(
|
|
"/compact",
|
|
session, # type: ignore[arg-type]
|
|
None, # type: ignore[arg-type]
|
|
host,
|
|
RichBlockFormatter(), # type: ignore[arg-type]
|
|
)
|
|
|
|
assert session.calls == 1
|
|
assert "Compaction failed: boom" in host.text
|