test/cli: cover selection, slash commands, and providers list

tests/test_cli/test_selection.py

@@ -0,0 +1,48 @@
+from __future__ import annotations
+
+import pytest
+
+from plyngent.cli.selection import select_model, select_provider
+from plyngent.config.models import ModelConfig, OpenAICompatibleProvider, OpenAIProvider
+
+
+def test_select_provider_preferred() -> None:
+    providers = {
+        "a": OpenAIProvider(access_key_or_token="sk"),
+        "b": OpenAICompatibleProvider(access_key_or_token="sk", url="https://x/v1"),
+    }
+    name, provider = select_provider(providers, preferred="b")
+    assert name == "b"
+    assert isinstance(provider, OpenAICompatibleProvider)
+
+
+def test_select_provider_single_auto() -> None:
+    providers = {"only": OpenAIProvider(access_key_or_token="sk")}
+    name, _ = select_provider(providers)
+    assert name == "only"
+
+
+def test_select_provider_unknown() -> None:
+    providers = {"a": OpenAIProvider(access_key_or_token="sk")}
+    with pytest.raises(Exception, match="unknown provider"):
+        _ = select_provider(providers, preferred="nope")
+
+
+def test_select_model_from_list() -> None:
+    provider = OpenAICompatibleProvider(
+        access_key_or_token="sk",
+        url="https://x/v1",
+        models={"m1": ModelConfig()},
+    )
+    assert select_model(provider) == "m1"
+    assert select_model(provider, preferred="m1") == "m1"
+
+
+def test_select_model_prompt_when_empty(monkeypatch: pytest.MonkeyPatch) -> None:
+    provider = OpenAIProvider(access_key_or_token="sk")
+
+    def _prompt(*_args: object, **_kwargs: object) -> str:
+        return "gpt-test"
+
+    monkeypatch.setattr("click.prompt", _prompt)
+    assert select_model(provider) == "gpt-test"