WIP: PAIN
This commit is contained in:
@@ -1,176 +1,112 @@
|
||||
import sys
|
||||
import unittest
|
||||
from unittest.mock import patch, MagicMock
|
||||
import gui_2
|
||||
import pytest
|
||||
import importlib
|
||||
from pathlib import Path
|
||||
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
|
||||
|
||||
class TestHeadlessAPI(unittest.TestCase):
|
||||
def setUp(self) -> None:
|
||||
with patch('src.models.load_config', return_value={'ai': {'provider': 'gemini', 'model': 'gemini-2.5-flash-lite'}, 'projects': {}, 'gui': {'show_windows': {}}}), \
|
||||
patch('gui_2.session_logger.open_session'), \
|
||||
patch('gui_2.ai_client.set_provider'), \
|
||||
patch('gui_2.PerformanceMonitor'), \
|
||||
patch('gui_2.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.app_instance = gui_2.App()
|
||||
# Set a default API key for tests
|
||||
self.test_api_key = "test-secret-key"
|
||||
self.app_instance.config["headless"] = {"api_key": self.test_api_key}
|
||||
self.headers = {"X-API-KEY": self.test_api_key}
|
||||
# Clear any leftover state
|
||||
self.app_instance._pending_actions = {}
|
||||
self.app_instance._pending_dialog = None
|
||||
self.api = self.app_instance.create_api()
|
||||
self.client = TestClient(self.api)
|
||||
def setUp(self) -> None:
|
||||
with patch('src.models.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.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:
|
||||
if hasattr(self, 'app_instance'):
|
||||
self.app_instance.shutdown()
|
||||
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_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:
|
||||
with patch.dict(self.app_instance.config, {"headless": {"api_key": "some-required-key"}}):
|
||||
response = self.client.get("/status")
|
||||
self.assertEqual(response.status_code, 403)
|
||||
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:
|
||||
headers = {"X-API-KEY": "test-secret-key"}
|
||||
with patch.dict(self.app_instance.config, {"headless": {"api_key": "test-secret-key"}}):
|
||||
response = self.client.get("/status", headers=headers)
|
||||
self.assertEqual(response.status_code, 200)
|
||||
def test_status_endpoint_authorized(self) -> None:
|
||||
response = self.client.get("/status", headers=self.headers)
|
||||
self.assertEqual(response.status_code, 200)
|
||||
data = response.json()
|
||||
self.assertIn("status", data)
|
||||
self.assertIn("provider", data)
|
||||
|
||||
def test_generate_endpoint(self) -> None:
|
||||
payload = {
|
||||
"prompt": "Hello AI"
|
||||
}
|
||||
# Mock ai_client.send and get_comms_log
|
||||
with patch('gui_2.ai_client.send') as mock_send, \
|
||||
patch('gui_2.ai_client.get_comms_log') as mock_log:
|
||||
mock_send.return_value = "Hello from Mock AI"
|
||||
mock_log.return_value = [{
|
||||
"kind": "response",
|
||||
"payload": {
|
||||
"usage": {"input_tokens": 10, "output_tokens": 5}
|
||||
}
|
||||
}]
|
||||
response = self.client.post("/api/v1/generate", json=payload, headers=self.headers)
|
||||
self.assertEqual(response.status_code, 200)
|
||||
data = response.json()
|
||||
self.assertEqual(data["text"], "Hello from Mock AI")
|
||||
self.assertIn("metadata", data)
|
||||
self.assertEqual(data["usage"]["input_tokens"], 10)
|
||||
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_pending_actions_endpoint(self) -> None:
|
||||
with patch('gui_2.uuid.uuid4', return_value="test-action-id"):
|
||||
dialog = gui_2.ConfirmDialog("dir", ".")
|
||||
self.app_instance._pending_actions[dialog._uid] = dialog
|
||||
response = self.client.get("/api/v1/pending_actions", headers=self.headers)
|
||||
self.assertEqual(response.status_code, 200)
|
||||
data = response.json()
|
||||
self.assertEqual(len(data), 1)
|
||||
self.assertEqual(data[0]["action_id"], "test-action-id")
|
||||
def test_generate_endpoint(self) -> None:
|
||||
with patch('src.ai_client.send', return_value="AI Response"), \
|
||||
patch('src.app_controller.AppController._do_generate', return_value=("md", "path", [], "stable", "disc")):
|
||||
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)
|
||||
self.assertEqual(response.json()["text"], "AI Response")
|
||||
|
||||
def test_confirm_action_endpoint(self) -> None:
|
||||
with patch('gui_2.uuid.uuid4', return_value="test-confirm-id"):
|
||||
dialog = gui_2.ConfirmDialog("dir", ".")
|
||||
self.app_instance._pending_actions[dialog._uid] = dialog
|
||||
payload = {"approved": True}
|
||||
response = self.client.post("/api/v1/confirm/test-confirm-id", json=payload, headers=self.headers)
|
||||
self.assertEqual(response.status_code, 200)
|
||||
self.assertTrue(dialog._done)
|
||||
self.assertTrue(dialog._approved)
|
||||
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_list_sessions_endpoint(self) -> None:
|
||||
Path("logs").mkdir(exist_ok=True)
|
||||
# Create a dummy log
|
||||
dummy_log = Path("logs/test_session_api.log")
|
||||
dummy_log.write_text("dummy content")
|
||||
try:
|
||||
response = self.client.get("/api/v1/sessions", headers=self.headers)
|
||||
self.assertEqual(response.status_code, 200)
|
||||
data = response.json()
|
||||
self.assertIn("test_session_api.log", data)
|
||||
finally:
|
||||
if dummy_log.exists():
|
||||
dummy_log.unlink()
|
||||
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_get_context_endpoint(self) -> None:
|
||||
response = self.client.get("/api/v1/context", headers=self.headers)
|
||||
self.assertEqual(response.status_code, 200)
|
||||
data = response.json()
|
||||
self.assertIn("files", data)
|
||||
self.assertIn("screenshots", data)
|
||||
self.assertIn("files_base_dir", data)
|
||||
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_endpoint_no_api_key_configured(self) -> None:
|
||||
with patch.dict(self.app_instance.config, {"headless": {"api_key": ""}}):
|
||||
response = self.client.get("/status", headers=self.headers)
|
||||
self.assertEqual(response.status_code, 403)
|
||||
self.assertEqual(response.json()["detail"], "API Key not configured on server")
|
||||
def test_get_context_endpoint(self) -> None:
|
||||
with patch('src.app_controller.AppController._do_generate', return_value=("md", "path", [], "stable", "disc")):
|
||||
response = self.client.get("/api/v1/context", headers=self.headers)
|
||||
self.assertEqual(response.status_code, 200)
|
||||
data = response.json()
|
||||
self.assertEqual(data["markdown"], "md")
|
||||
|
||||
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('gui_2.immapp.run')
|
||||
@patch('gui_2.api_hooks.HookServer')
|
||||
@patch('gui_2.save_config')
|
||||
@patch('gui_2.ai_client.cleanup')
|
||||
@patch('gui_2.PerformanceMonitor')
|
||||
@patch('uvicorn.run') # Mock uvicorn.run to prevent hanging
|
||||
def test_headless_flag_prevents_gui_run(self, mock_uvicorn_run: MagicMock, mock_perf: MagicMock, mock_cleanup: MagicMock, mock_save_config: MagicMock, mock_hook_server: MagicMock, mock_immapp_run: MagicMock) -> None:
|
||||
test_args = ["gui_2.py", "--headless"]
|
||||
with patch.object(sys, 'argv', test_args):
|
||||
with patch('gui_2.session_logger.close_session'), \
|
||||
patch('gui_2.session_logger.open_session'):
|
||||
app = gui_2.App()
|
||||
# Mock _fetch_models to avoid network calls
|
||||
app._fetch_models = MagicMock()
|
||||
app.run()
|
||||
# Expectation: immapp.run should NOT be called in headless mode
|
||||
mock_immapp_run.assert_not_called()
|
||||
# Expectation: uvicorn.run SHOULD be called
|
||||
mock_uvicorn_run.assert_called_once()
|
||||
app.shutdown()
|
||||
|
||||
@patch('gui_2.immapp.run')
|
||||
@patch('gui_2.PerformanceMonitor')
|
||||
def test_normal_startup_calls_gui_run(self, mock_perf: MagicMock, mock_immapp_run: MagicMock) -> None:
|
||||
test_args = ["gui_2.py"]
|
||||
with patch.object(sys, 'argv', test_args):
|
||||
# In normal mode, it should still call immapp.run
|
||||
with patch('gui_2.api_hooks.HookServer'), \
|
||||
patch('gui_2.save_config'), \
|
||||
patch('gui_2.ai_client.cleanup'), \
|
||||
patch('gui_2.session_logger.close_session'), \
|
||||
patch('gui_2.session_logger.open_session'):
|
||||
app = gui_2.App()
|
||||
app._fetch_models = MagicMock()
|
||||
app.run()
|
||||
mock_immapp_run.assert_called_once()
|
||||
app.shutdown()
|
||||
|
||||
def test_fastapi_installed() -> None:
|
||||
"""Verify that fastapi is installed."""
|
||||
try:
|
||||
importlib.import_module("fastapi")
|
||||
except ImportError:
|
||||
pytest.fail("fastapi is not installed")
|
||||
|
||||
def test_uvicorn_installed() -> None:
|
||||
"""Verify that uvicorn is installed."""
|
||||
try:
|
||||
importlib.import_module("uvicorn")
|
||||
except ImportError:
|
||||
pytest.fail("uvicorn is not installed")
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
@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()
|
||||
|
||||
Reference in New Issue
Block a user