|
1 | 1 | import os |
| 2 | +import asyncio |
| 3 | +import threading |
2 | 4 | import unittest |
3 | 5 | from unittest.mock import patch |
4 | 6 |
|
5 | 7 | import server |
6 | 8 | from fastapi import HTTPException |
7 | 9 | from fastapi.testclient import TestClient |
| 10 | +from providers import ProviderConfig |
8 | 11 |
|
9 | 12 |
|
10 | 13 | class ServerBoundaryTests(unittest.TestCase): |
| 14 | + def test_ollama_initialization_never_prompts_for_a_key(self): |
| 15 | + original_config = server._agent._provider_config |
| 16 | + original_provider = server._agent.llm_provider |
| 17 | + config = ProviderConfig(provider="ollama", base_url="http://127.0.0.1:11434/v1") |
| 18 | + try: |
| 19 | + with patch.object(server._agent, "detect_provider", return_value=config): |
| 20 | + with patch.object(server._agent, "get_fallback_provider", return_value=object()) as create: |
| 21 | + with patch.object(server._agent.console, "input", side_effect=AssertionError("prompted")): |
| 22 | + self.assertTrue(server._agent._prompt_and_init_deepseek(interactive=False)) |
| 23 | + create.assert_called_once_with(config) |
| 24 | + self.assertEqual(server._agent._provider_config.provider, "ollama") |
| 25 | + finally: |
| 26 | + server._agent._provider_config = original_config |
| 27 | + server._agent.llm_provider = original_provider |
| 28 | + |
| 29 | + def test_headless_missing_remote_key_enters_deterministic_degraded_state(self): |
| 30 | + original_config = server._agent._provider_config |
| 31 | + original_provider = server._agent.llm_provider |
| 32 | + config = ProviderConfig(provider="deepseek", api_key="") |
| 33 | + try: |
| 34 | + with patch.object(server._agent, "detect_provider", return_value=config): |
| 35 | + with patch.object(server._agent.console, "input", side_effect=AssertionError("prompted")): |
| 36 | + self.assertFalse(server._agent._prompt_and_init_deepseek(interactive=False)) |
| 37 | + self.assertIsNone(server._agent.llm_provider) |
| 38 | + self.assertEqual(server._agent._provider_config.provider, "deepseek") |
| 39 | + finally: |
| 40 | + server._agent._provider_config = original_config |
| 41 | + server._agent.llm_provider = original_provider |
| 42 | + |
| 43 | + def test_interactive_missing_remote_key_still_accepts_a_key(self): |
| 44 | + original_config = server._agent._provider_config |
| 45 | + original_provider = server._agent.llm_provider |
| 46 | + config = ProviderConfig(provider="deepseek", api_key="") |
| 47 | + try: |
| 48 | + with patch.object(server._agent, "detect_provider", return_value=config): |
| 49 | + with patch.object(server._agent.console, "input", return_value="sk-interactive"): |
| 50 | + with patch.object(server._agent, "save_provider_config_encrypted"): |
| 51 | + with patch.object(server._agent, "get_fallback_provider", return_value=object()): |
| 52 | + self.assertTrue(server._agent._prompt_and_init_deepseek(interactive=True)) |
| 53 | + self.assertEqual(server._agent._provider_config.api_key, "sk-interactive") |
| 54 | + finally: |
| 55 | + server._agent._provider_config = original_config |
| 56 | + server._agent.llm_provider = original_provider |
| 57 | + |
| 58 | + def test_server_startup_initializes_headlessly(self): |
| 59 | + with patch.object(server._agent, "_prompt_and_init_deepseek") as init_provider: |
| 60 | + with patch.object(server._agent, "_set_workspace_root"): |
| 61 | + with patch.object(server._agent, "_load_project_files_into_memory"): |
| 62 | + with patch.object(server, "_load_plugins"): |
| 63 | + with patch.object(server, "_trigger_hook"): |
| 64 | + with patch.object(server._task_worker, "recover", return_value=[]): |
| 65 | + with patch.object(server._scheduler, "list_jobs", return_value=[ |
| 66 | + {"payload": {"type": "task_worker"}}, |
| 67 | + ]): |
| 68 | + with patch.object(server._scheduler, "start"): |
| 69 | + errors = [] |
| 70 | + |
| 71 | + def run_startup(): |
| 72 | + try: |
| 73 | + asyncio.run(server.startup()) |
| 74 | + except Exception as exc: # pragma: no cover - assertion below reports it |
| 75 | + errors.append(exc) |
| 76 | + |
| 77 | + thread = threading.Thread(target=run_startup) |
| 78 | + thread.start() |
| 79 | + thread.join(timeout=5) |
| 80 | + self.assertFalse(thread.is_alive()) |
| 81 | + self.assertEqual(errors, []) |
| 82 | + init_provider.assert_called_once_with(interactive=False) |
| 83 | + |
11 | 84 | def test_server_profiles_keep_workspace_tools_and_gate_reset(self): |
12 | 85 | workspace = server._allowed_server_tools("mcp") |
13 | 86 | self.assertIn("run_cmd", workspace) |
|
0 commit comments