plyngent/tests/test_cli/test_readline_setup.py

from __future__ import annotations

import importlib.util

import pytest
import tomlkit

from plyngent.cli.readline_setup import (
    build_completer,
    filter_prefix,
    history_path,
    slash_commands,
)
from plyngent.cli.state import ReplState
from plyngent.config.models import ModelConfig, OpenAICompatibleProvider, OpenAIProvider
from plyngent.config.store import ConfigStore

pytestmark = pytest.mark.skipif(
    importlib.util.find_spec("readline") is None,
    reason="readline not available (e.g. Windows/Wine)",
)


def _minimal_state(tmp_path: object) -> ReplState:
    from pathlib import Path
    from unittest.mock import MagicMock

    assert isinstance(tmp_path, Path)
    provider = OpenAICompatibleProvider(
        access_key_or_token="sk",
        url="https://example.com/v1",
        models={"alpha": ModelConfig(), "beta": ModelConfig()},
    )
    config = ConfigStore(path=tmp_path / "plyngent.toml", document=tomlkit.document())
    config.providers = {
        "local": OpenAIProvider(access_key_or_token="sk"),
        "remote": provider,
    }
    # Avoid real client/network: build ReplState pieces manually via object.__new__
    state = object.__new__(ReplState)
    state.config = config
    state.provider = provider
    state.provider_name = "remote"
    state.model = "alpha"
    state.tools_enabled = True
    state.memory = MagicMock()
    state.workspace = tmp_path
    state.session_id = None
    state.client = MagicMock()
    state.agent = MagicMock()
    return state


def test_filter_prefix() -> None:
    assert filter_prefix("/he", ["/help", "/quit"]) == ["/help"]
    assert filter_prefix("", ["a", "b"]) == ["a", "b"]


def test_history_path_under_user_data() -> None:
    path = history_path()
    assert path.name == "repl_history"
    assert "plyngent" in str(path)


def test_completer_commands(tmp_path: object, monkeypatch: object) -> None:
    import readline
    from pathlib import Path

    import pytest

    assert isinstance(tmp_path, Path)
    assert isinstance(monkeypatch, pytest.MonkeyPatch)

    state = _minimal_state(tmp_path)
    completer = build_completer(state)
    monkeypatch.setattr(readline, "get_line_buffer", lambda: "/")
    monkeypatch.setattr(readline, "get_begidx", lambda: 0)

    found: list[str] = []
    index = 0
    while True:
        item = completer("/", index)
        if item is None:
            break
        found.append(item)
        index += 1
    names = slash_commands()
    assert "/help" in found
    assert "/stream" in names
    assert "/verbose" in names
    assert set(found) <= set(names)


def test_bind_tab_complete_runs() -> None:
    from plyngent.cli.readline_setup import bind_tab_complete

    class FakeReadline:
        binds: list[str]

        def __init__(self) -> None:
            self.binds = []

        def parse_and_bind(self, s: str) -> None:
            self.binds.append(s)

    fake = FakeReadline()
    bind_tab_complete(fake)
    assert any("complete" in b or "rl_complete" in b for b in fake.binds)


def test_completer_provider_args(tmp_path: object, monkeypatch: object) -> None:
    import readline
    from pathlib import Path

    import pytest

    assert isinstance(tmp_path, Path)
    assert isinstance(monkeypatch, pytest.MonkeyPatch)

    state = _minimal_state(tmp_path)
    completer = build_completer(state)
    monkeypatch.setattr(readline, "get_line_buffer", lambda: "/provider ")
    monkeypatch.setattr(readline, "get_begidx", lambda: len("/provider "))

    first = completer("r", 0)
    assert first == "remote"


def test_completer_help_commands(tmp_path: object, monkeypatch: object) -> None:
    import readline
    from pathlib import Path

    import pytest

    assert isinstance(tmp_path, Path)
    assert isinstance(monkeypatch, pytest.MonkeyPatch)

    state = _minimal_state(tmp_path)
    completer = build_completer(state)
    monkeypatch.setattr(readline, "get_line_buffer", lambda: "/help ")
    monkeypatch.setattr(readline, "get_begidx", lambda: len("/help "))

    found: list[str] = []
    index = 0
    while True:
        item = completer("c", index)
        if item is None:
            break
        found.append(item)
        index += 1
    assert "compact" in found
    assert "clear" in found


def test_completer_stream_on_off(tmp_path: object, monkeypatch: object) -> None:
    import readline
    from pathlib import Path

    import pytest

    assert isinstance(tmp_path, Path)
    assert isinstance(monkeypatch, pytest.MonkeyPatch)

    state = _minimal_state(tmp_path)
    completer = build_completer(state)
    monkeypatch.setattr(readline, "get_line_buffer", lambda: "/stream ")
    monkeypatch.setattr(readline, "get_begidx", lambda: len("/stream "))

    assert completer("o", 0) == "on"
    assert completer("o", 1) == "off"


def test_complete_slash_args_from_registry(tmp_path: object) -> None:
    from pathlib import Path

    from plyngent.cli.slash import complete_slash_args

    assert isinstance(tmp_path, Path)
    state = _minimal_state(tmp_path)
    assert complete_slash_args(state, "/provider", "r") == ["remote"]
    assert complete_slash_args(state, "/model", "a") == ["alpha"]
    assert complete_slash_args(state, "/export", "j") == ["json"]
    assert complete_slash_args(state, "/help", "st") == ["status", "stream"]