kolega-code 0.1.0__py3-none-any.whl

kolega_code/cli/tests/test_cli_config.py

@@ -0,0 +1,171 @@
+from pathlib import Path
+
+import pytest
+
+from kolega_code.config import ModelProvider
+from kolega_code.cli.config import (
+    DEFAULT_EDIT_MODEL,
+    DEFAULT_FAST_MODEL,
+    DEFAULT_LONG_MODEL,
+    DEFAULT_THINKING_MODEL,
+    CliConfigError,
+    CliConfigOverrides,
+    build_agent_config,
+    config_summary,
+)
+from kolega_code.cli.provider_registry import DEEPSEEK_DEFAULT_MODEL, UI_DEFAULT_MODEL, UI_DEFAULT_PROVIDER
+from kolega_code.cli.settings import CliSettings
+
+
+def test_build_agent_config_defaults_to_latest_anthropic_models(tmp_path: Path) -> None:
+    config = build_agent_config(tmp_path, env={"ANTHROPIC_API_KEY": "test-key"})
+
+    assert config.long_context_config.provider == ModelProvider.ANTHROPIC
+    assert config.long_context_config.model == DEFAULT_LONG_MODEL
+    assert config.fast_config.model == DEFAULT_FAST_MODEL
+    assert config.edit_model_config.model == DEFAULT_EDIT_MODEL
+    assert config.thinking_config.model == DEFAULT_THINKING_MODEL
+    assert config.thinking_config.thinking_tokens == 1024
+
+
+def test_build_agent_config_env_overrides(tmp_path: Path) -> None:
+    config = build_agent_config(
+        tmp_path,
+        env={
+            "ANTHROPIC_API_KEY": "test-key",
+            "KOLEGA_CODE_MODEL": "claude-sonnet-4-6",
+            "KOLEGA_CODE_THINKING_TOKENS": "2048",
+        },
+    )
+
+    assert config.long_context_config.model == "claude-sonnet-4-6"
+    assert config.thinking_config.thinking_tokens == 2048
+
+
+def test_build_agent_config_flags_override_env(tmp_path: Path) -> None:
+    config = build_agent_config(
+        tmp_path,
+        CliConfigOverrides(model="claude-opus-4-7", thinking_tokens=4096),
+        env={
+            "ANTHROPIC_API_KEY": "test-key",
+            "KOLEGA_CODE_MODEL": "claude-sonnet-4-6",
+            "KOLEGA_CODE_THINKING_TOKENS": "2048",
+        },
+    )
+
+    assert config.long_context_config.model == "claude-opus-4-7"
+    assert config.thinking_config.thinking_tokens == 4096
+
+
+def test_build_agent_config_requires_api_key(tmp_path: Path) -> None:
+    with pytest.raises(CliConfigError, match="ANTHROPIC_API_KEY"):
+        build_agent_config(tmp_path, env={})
+
+
+def test_build_agent_config_rejects_unknown_model(tmp_path: Path) -> None:
+    with pytest.raises(CliConfigError, match="not supported"):
+        build_agent_config(tmp_path, CliConfigOverrides(model="claude-not-real"), env={"ANTHROPIC_API_KEY": "key"})
+
+
+def test_config_summary_excludes_api_keys(tmp_path: Path) -> None:
+    config = build_agent_config(tmp_path, env={"ANTHROPIC_API_KEY": "secret-value"})
+
+    summary = config_summary(config)
+
+    assert summary["long_model"] == DEFAULT_LONG_MODEL
+    assert "secret-value" not in str(summary)
+    assert "api_key" not in str(summary).lower()
+
+
+def test_build_agent_config_uses_stored_kimi_for_all_model_slots(tmp_path: Path) -> None:
+    settings = CliSettings(active_provider=UI_DEFAULT_PROVIDER, active_model=UI_DEFAULT_MODEL)
+    settings.set_api_key(UI_DEFAULT_PROVIDER, "moonshot-key")
+
+    config = build_agent_config(tmp_path, settings=settings, env={})
+
+    assert config.long_context_config.provider == ModelProvider.MOONSHOT
+    assert config.long_context_config.model == UI_DEFAULT_MODEL
+    assert config.fast_config.provider == ModelProvider.MOONSHOT
+    assert config.fast_config.model == UI_DEFAULT_MODEL
+    assert config.edit_model_config.provider == ModelProvider.MOONSHOT
+    assert config.edit_model_config.model == UI_DEFAULT_MODEL
+    assert config.thinking_config.provider == ModelProvider.MOONSHOT
+    assert config.thinking_config.model == UI_DEFAULT_MODEL
+    assert config.moonshot_api_key == "moonshot-key"
+
+
+def test_build_agent_config_uses_stored_deepseek_for_all_model_slots(tmp_path: Path) -> None:
+    settings = CliSettings(active_provider=ModelProvider.DEEPSEEK.value, active_model=DEEPSEEK_DEFAULT_MODEL)
+    settings.set_api_key(ModelProvider.DEEPSEEK.value, "deepseek-key")
+
+    config = build_agent_config(tmp_path, settings=settings, env={})
+
+    assert config.long_context_config.provider == ModelProvider.DEEPSEEK
+    assert config.long_context_config.model == DEEPSEEK_DEFAULT_MODEL
+    assert config.fast_config.provider == ModelProvider.DEEPSEEK
+    assert config.fast_config.model == DEEPSEEK_DEFAULT_MODEL
+    assert config.edit_model_config.provider == ModelProvider.DEEPSEEK
+    assert config.edit_model_config.model == DEEPSEEK_DEFAULT_MODEL
+    assert config.thinking_config.provider == ModelProvider.DEEPSEEK
+    assert config.thinking_config.model == DEEPSEEK_DEFAULT_MODEL
+    assert config.deepseek_api_key == "deepseek-key"
+
+
+def test_build_agent_config_accepts_moonshot_cli_active_model(tmp_path: Path) -> None:
+    config = build_agent_config(
+        tmp_path,
+        CliConfigOverrides(provider=UI_DEFAULT_PROVIDER, model=UI_DEFAULT_MODEL),
+        env={"MOONSHOT_API_KEY": "moonshot-key"},
+    )
+
+    assert config.long_context_config.provider == ModelProvider.MOONSHOT
+    assert config.fast_config.provider == ModelProvider.MOONSHOT
+    assert config.edit_model_config.provider == ModelProvider.MOONSHOT
+    assert config.thinking_config.provider == ModelProvider.MOONSHOT
+
+
+def test_build_agent_config_accepts_deepseek_cli_active_model(tmp_path: Path) -> None:
+    config = build_agent_config(
+        tmp_path,
+        CliConfigOverrides(provider=ModelProvider.DEEPSEEK.value, model=DEEPSEEK_DEFAULT_MODEL),
+        env={"DEEPSEEK_API_KEY": "deepseek-key"},
+    )
+
+    assert config.long_context_config.provider == ModelProvider.DEEPSEEK
+    assert config.fast_config.provider == ModelProvider.DEEPSEEK
+    assert config.edit_model_config.provider == ModelProvider.DEEPSEEK
+    assert config.thinking_config.provider == ModelProvider.DEEPSEEK
+
+
+def test_env_provider_model_overrides_stored_settings(tmp_path: Path) -> None:
+    settings = CliSettings(active_provider=UI_DEFAULT_PROVIDER, active_model=UI_DEFAULT_MODEL)
+    settings.set_api_key(UI_DEFAULT_PROVIDER, "moonshot-key")
+
+    config = build_agent_config(
+        tmp_path,
+        settings=settings,
+        env={
+            "ANTHROPIC_API_KEY": "anthropic-key",
+            "KOLEGA_CODE_PROVIDER": "anthropic",
+            "KOLEGA_CODE_MODEL": "claude-sonnet-4-6",
+        },
+    )
+
+    assert config.long_context_config.provider == ModelProvider.ANTHROPIC
+    assert config.long_context_config.model == "claude-sonnet-4-6"
+    assert config.fast_config.provider == ModelProvider.ANTHROPIC
+    assert config.fast_config.model == "claude-sonnet-4-6"
+
+
+def test_stored_kimi_settings_require_moonshot_key(tmp_path: Path) -> None:
+    settings = CliSettings(active_provider=UI_DEFAULT_PROVIDER, active_model=UI_DEFAULT_MODEL)
+
+    with pytest.raises(CliConfigError, match="MOONSHOT_API_KEY"):
+        build_agent_config(tmp_path, settings=settings, env={})
+
+
+def test_stored_deepseek_settings_require_deepseek_key(tmp_path: Path) -> None:
+    settings = CliSettings(active_provider=ModelProvider.DEEPSEEK.value, active_model=DEEPSEEK_DEFAULT_MODEL)
+
+    with pytest.raises(CliConfigError, match="DEEPSEEK_API_KEY"):
+        build_agent_config(tmp_path, settings=settings, env={})

kolega_code/cli/tests/test_connection.py

@@ -0,0 +1,26 @@
+import pytest
+
+from kolega_code.events import AgentEvent
+from kolega_code.cli.connection import CliConnectionManager
+
+
+@pytest.mark.asyncio
+async def test_cli_connection_manager_queues_broadcast_events() -> None:
+    manager = CliConnectionManager()
+    event = AgentEvent(event_type="log_message", sender="test", content={"text": "hello"})
+
+    await manager.broadcast_event(event, "workspace", "thread")
+
+    assert await manager.next_event() == event
+
+
+@pytest.mark.asyncio
+async def test_cli_connection_manager_counts_connections() -> None:
+    manager = CliConnectionManager()
+
+    await manager.connect(object(), "workspace", "thread", "chat")
+    await manager.connect(object(), "workspace", "thread", "chat")
+    await manager.connect(object(), "workspace", "thread", "logs")
+    manager.disconnect(object(), "workspace", "thread", "chat")
+
+    assert manager.get_connection_count("workspace", "thread") == {"chat": 1, "logs": 1}

kolega_code/cli/tests/test_file_index.py

@@ -0,0 +1,103 @@
+"""Tests for the workspace file index behind @ mention autocomplete."""
+
+from pathlib import Path
+
+from kolega_code.cli.file_index import WorkspaceFileIndex, fuzzy_score
+
+
+def _paths(entries) -> list[str]:
+    return [entry.path for entry in entries]
+
+
+def test_index_lists_files_and_directories(tmp_path: Path) -> None:
+    (tmp_path / "src").mkdir()
+    (tmp_path / "src" / "main.py").write_text("", encoding="utf-8")
+    (tmp_path / "README.md").write_text("", encoding="utf-8")
+    index = WorkspaceFileIndex(tmp_path)
+    paths = _paths(index.entries())
+    assert "src/" in paths
+    assert "src/main.py" in paths
+    assert "README.md" in paths
+
+
+def test_index_prunes_excluded_directories(tmp_path: Path) -> None:
+    (tmp_path / "node_modules" / "pkg").mkdir(parents=True)
+    (tmp_path / "node_modules" / "pkg" / "index.js").write_text("", encoding="utf-8")
+    (tmp_path / "app.js").write_text("", encoding="utf-8")
+    index = WorkspaceFileIndex(tmp_path)
+    paths = _paths(index.entries())
+    assert paths == ["app.js"]
+
+
+def test_index_respects_gitignore(tmp_path: Path) -> None:
+    (tmp_path / ".gitignore").write_text("ignored.txt\nsecrets/\n", encoding="utf-8")
+    (tmp_path / "ignored.txt").write_text("", encoding="utf-8")
+    (tmp_path / "kept.txt").write_text("", encoding="utf-8")
+    (tmp_path / "secrets").mkdir()
+    (tmp_path / "secrets" / "key.pem").write_text("", encoding="utf-8")
+    index = WorkspaceFileIndex(tmp_path)
+    paths = _paths(index.entries())
+    assert "kept.txt" in paths
+    assert "ignored.txt" not in paths
+    assert all(not path.startswith("secrets") for path in paths)
+
+
+def test_index_skips_binary_extensions(tmp_path: Path) -> None:
+    (tmp_path / "image.png").write_bytes(b"png")
+    (tmp_path / "code.py").write_text("", encoding="utf-8")
+    index = WorkspaceFileIndex(tmp_path)
+    paths = _paths(index.entries())
+    assert "code.py" in paths
+    assert "image.png" not in paths
+
+
+def test_index_caps_total_entries(tmp_path: Path, monkeypatch) -> None:
+    monkeypatch.setattr(WorkspaceFileIndex, "MAX_FILES", 10)
+    for index_num in range(25):
+        (tmp_path / f"f{index_num:03}.txt").write_text("", encoding="utf-8")
+    index = WorkspaceFileIndex(tmp_path)
+    assert len(index.entries()) == 10
+
+
+def test_index_refreshes_after_ttl(tmp_path: Path, monkeypatch) -> None:
+    monkeypatch.setattr(WorkspaceFileIndex, "TTL_SECONDS", 0.0)
+    index = WorkspaceFileIndex(tmp_path)
+    assert index.entries() == []
+    (tmp_path / "late.txt").write_text("", encoding="utf-8")
+    assert _paths(index.entries()) == ["late.txt"]
+
+
+def test_search_ranks_substring_above_subsequence(tmp_path: Path) -> None:
+    (tmp_path / "main.py").write_text("", encoding="utf-8")
+    (tmp_path / "mandatory_ai_notes.txt").write_text("", encoding="utf-8")
+    index = WorkspaceFileIndex(tmp_path)
+    results = _paths(index.search("main"))
+    assert results[0] == "main.py"
+
+
+def test_search_empty_query_prefers_shallow_paths(tmp_path: Path) -> None:
+    (tmp_path / "deep" / "nest").mkdir(parents=True)
+    (tmp_path / "deep" / "nest" / "leaf.txt").write_text("", encoding="utf-8")
+    (tmp_path / "top.txt").write_text("", encoding="utf-8")
+    index = WorkspaceFileIndex(tmp_path)
+    results = _paths(index.search(""))
+    assert results[0] == "top.txt"
+
+
+def test_fuzzy_score_no_match_returns_none() -> None:
+    assert fuzzy_score("zzz", "src/main.py") is None
+
+
+def test_fuzzy_score_subsequence_matches() -> None:
+    assert fuzzy_score("smain", "src/main.py") is not None
+    assert fuzzy_score("apppy", "kolega_code/cli/app.py") is not None
+
+
+def test_search_dense_basename_match_beats_scattered_path_match(tmp_path: Path) -> None:
+    (tmp_path / "cli").mkdir()
+    (tmp_path / "agent").mkdir()
+    (tmp_path / "cli" / "app.py").write_text("", encoding="utf-8")
+    (tmp_path / "agent" / "prompts.py").write_text("", encoding="utf-8")
+    index = WorkspaceFileIndex(tmp_path)
+    results = _paths(index.search("apppy"))
+    assert results[0] == "cli/app.py"