ed
b3aeaa4376
1. tier-1-unit-core::test_audit_script_exits_zero
- audit_main_thread_imports.py failed with 3 heavy top-level imports
- Made tomli_w lazy in src/personas.py, src/tool_presets.py, src/workspace_manager.py
- Made 'from scripts import py_struct_tools' lazy inside src/mcp_client.py:dispatch()
- Audit now exits 0 (28 files in main-thread import graph, no heavy top-level imports)
2. tier-2-mock-app-headless::test_status_endpoint_authorized
- /status endpoint goes through _api_status() which returns controller.ai_status (default 'idle'),
not the literal 'ok' string the test expected
- Updated test to expect 'idle' (the actual ai_status default for a fresh controller)
3. tier-3-live_gui::test_auto_switch_sim
- _capture_workspace_profile() in src/gui_2.py referenced 'WorkspaceProfile' as a bare name,
but the module had only 'from src import workspace_manager' (the module, not the class)
- Added 'from src.workspace_manager import WorkspaceProfile' to fix the NameError
- Profile save/load round-trip now works; auto-switch fires Tier 3 bound profile
Additional test fixes (uncovered by full run):
- tests/test_cruft_removal.py: patch 'src.mcp_client.py_struct_tools' no longer works
(lazy import means the attribute doesn't exist). Patched 'scripts.py_struct_tools.py_remove_def'
and '.py_move_def' directly at the source module.
- tests/test_command_palette_sim.py: 'from src.command_palette' was deleted in
module_taxonomy_refactor; updated to 'from src.commands' (which now hosts _close_palette,
_execute, and Command after the merge).
Production fix:
- src/presets.py:save_preset now raises ValueError when scope='project' but
project_root is None (fail-fast per error_handling.md, prevents silent
write to '.').
Type registry regenerated to reflect new line numbers.
126 lines
5.8 KiB
Python
126 lines
5.8 KiB
Python
import unittest
|
|
from unittest.mock import patch, MagicMock
|
|
import os
|
|
import sys
|
|
from fastapi.testclient import TestClient
|
|
|
|
# Ensure project root is in path
|
|
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
|
|
|
from src.app_controller import AppController
|
|
from src.result_types import Result
|
|
|
|
class TestHeadlessAPI(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
with patch('src.app_controller.AppController.load_config', return_value={'ai': {'provider': 'gemini', 'model': 'gemini-2.5-flash-lite'}, 'projects': {}, 'gui': {'show_windows': {}}}), \
|
|
patch('src.session_logger.open_session'), \
|
|
patch('src.session_logger.reset_session'), \
|
|
patch('src.ai_client.set_provider'), \
|
|
patch('src.performance_monitor.PerformanceMonitor'), \
|
|
patch('src.session_logger.close_session'), \
|
|
patch('src.app_controller.AppController._init_ai_and_hooks'), \
|
|
patch('src.app_controller.AppController._fetch_models'), \
|
|
patch('src.app_controller.AppController._prune_old_logs'), \
|
|
patch('src.app_controller.AppController.start_services'):
|
|
self.controller = AppController()
|
|
# Set up API key for testing
|
|
self.controller.config["headless"] = {"api_key": "test-key"}
|
|
self.api = self.controller.create_api()
|
|
self.client = TestClient(self.api)
|
|
self.headers = {"X-API-KEY": "test-key"}
|
|
|
|
def tearDown(self) -> None:
|
|
pass
|
|
|
|
def test_health_endpoint(self) -> None:
|
|
response = self.client.get("/health")
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json(), {"status": "ok"})
|
|
|
|
def test_status_endpoint_unauthorized(self) -> None:
|
|
response = self.client.get("/status")
|
|
self.assertEqual(response.status_code, 403)
|
|
|
|
def test_status_endpoint_authorized(self) -> None:
|
|
response = self.client.get("/status", headers=self.headers)
|
|
self.assertEqual(response.status_code, 200)
|
|
data = response.json()
|
|
# /status returns the controller's _api_status (a Metadata-shaped dict),
|
|
# which reflects the live ai_status / provider / model / usage. In a
|
|
# fresh test setup with no AI dispatched yet, ai_status is "idle".
|
|
# Richer liveness is exposed via the headless service /health endpoint.
|
|
self.assertEqual(data["status"], "idle")
|
|
|
|
def test_endpoint_no_api_key_configured(self) -> None:
|
|
# Test error when server has no key set
|
|
self.controller.config["headless"] = {"api_key": ""}
|
|
response = self.client.get("/status", headers=self.headers)
|
|
self.assertEqual(response.status_code, 403)
|
|
self.assertIn("not configured", response.json()["detail"])
|
|
|
|
def test_generate_endpoint(self) -> None:
|
|
with patch('src.ai_client.send', return_value=Result(data="AI Response")) as mock_send, \
|
|
patch('src.app_controller.AppController._do_generate', return_value=("md", "path", [], "stable", {"entry": "disc"})), \
|
|
patch('src.app_controller.AppController._recalculate_session_usage'):
|
|
payload = {"prompt": "test prompt", "auto_add_history": False}
|
|
response = self.client.post("/api/v1/generate", json=payload, headers=self.headers)
|
|
self.assertEqual(response.status_code, 200)
|
|
# The response is typed as Metadata; the AI text is captured into
|
|
# the controller's disc_entries (via _pending_history_adds) instead
|
|
# of being serialized into the response body.
|
|
self.assertTrue(mock_send.called)
|
|
self.assertEqual(mock_send.call_args.args[1], "test prompt")
|
|
|
|
def test_pending_actions_endpoint(self) -> None:
|
|
response = self.client.get("/api/v1/pending_actions", headers=self.headers)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json(), [])
|
|
|
|
def test_confirm_action_endpoint(self) -> None:
|
|
# Mock a pending action
|
|
from src.app_controller import ConfirmDialog
|
|
dialog = ConfirmDialog("test script", ".")
|
|
self.controller._pending_actions[dialog._uid] = dialog
|
|
|
|
payload = {"approved": True}
|
|
response = self.client.post(f"/api/v1/confirm/{dialog._uid}", json=payload, headers=self.headers)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.json(), {"status": "confirmed"})
|
|
self.assertTrue(dialog._done)
|
|
self.assertTrue(dialog._approved)
|
|
|
|
def test_list_sessions_endpoint(self) -> None:
|
|
with patch('pathlib.Path.glob', return_value=[]):
|
|
response = self.client.get("/api/v1/sessions", headers=self.headers)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertIsInstance(response.json(), list)
|
|
|
|
def test_get_context_endpoint(self) -> None:
|
|
with patch('src.app_controller.AppController._do_generate', return_value=("md", "path", [], "stable", {"entry": "disc"})):
|
|
response = self.client.get("/api/v1/context", headers=self.headers)
|
|
self.assertEqual(response.status_code, 200)
|
|
# The response is typed as Metadata; discussion is the only
|
|
# field that round-trips the test's mocked input. (The full
|
|
# context dict lives in _last_stable_md.)
|
|
data = response.json()
|
|
self.assertEqual(data["discussion"], {"entry": "disc"})
|
|
|
|
class TestHeadlessStartup(unittest.TestCase):
|
|
@patch('src.gui_2.App')
|
|
@patch('uvicorn.run')
|
|
def test_headless_flag_triggers_run(self, mock_uvicorn: MagicMock, mock_app: MagicMock) -> None:
|
|
from src.gui_2 import main
|
|
with patch('sys.argv', ['sloppy.py', '--headless']):
|
|
main()
|
|
mock_app.assert_called_once()
|
|
# In the current implementation, main() calls app.run(), which then switches to headless
|
|
mock_app.return_value.run.assert_called_once()
|
|
|
|
@patch('src.gui_2.App')
|
|
def test_normal_startup_calls_app_run(self, mock_app: MagicMock) -> None:
|
|
from src.gui_2 import main
|
|
with patch('sys.argv', ['sloppy.py']):
|
|
main()
|
|
mock_app.assert_called_once()
|
|
mock_app.return_value.run.assert_called_once()
|