ed/manual_slop
Private
Public Access
0
0
Fork 0
Code Activity

test(headless_service): adapt test_generate_endpoint to send_result (doeh cleanup Phase 2.5)

Browse Source
This commit is contained in:
ed
2026-06-15 13:57:47 -04:00
parent 9e89d52607
commit 81882c398e
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_headless_service.py
+2 -1
View File
@@ -8,6 +8,7 @@ from fastapi.testclient import TestClient
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
from src.app_controller import AppController from src.app_controller import AppController
from src.result_types import Result
class TestHeadlessAPI(unittest.TestCase): class TestHeadlessAPI(unittest.TestCase):
def setUp(self) -> None: def setUp(self) -> None:
@@ -55,7 +56,7 @@ class TestHeadlessAPI(unittest.TestCase):
self.assertIn("not configured", response.json()["detail"]) self.assertIn("not configured", response.json()["detail"])
def test_generate_endpoint(self) -> None: def test_generate_endpoint(self) -> None:
with patch('src.ai_client.send', return_value="AI Response"), \ with patch('src.ai_client.send_result', return_value=Result(data="AI Response")), \
patch('src.app_controller.AppController._do_generate', return_value=("md", "path", [], "stable", "disc")): patch('src.app_controller.AppController._do_generate', return_value=("md", "path", [], "stable", "disc")):
payload = {"prompt": "test prompt", "auto_add_history": False} payload = {"prompt": "test prompt", "auto_add_history": False}
response = self.client.post("/api/v1/generate", json=payload, headers=self.headers) response = self.client.post("/api/v1/generate", json=payload, headers=self.headers)