Implement AI orchestration wedding demo
This commit is contained in:
29
backend/tests/test_integration_flow.py
Normal file
29
backend/tests/test_integration_flow.py
Normal file
@@ -0,0 +1,29 @@
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
from app.infrastructure.database import PendingAction, SessionLocal, init_db
|
||||
from app.main import app
|
||||
|
||||
|
||||
def test_websocket_ingestion_creates_pending_action():
|
||||
init_db()
|
||||
session = SessionLocal()
|
||||
session.query(PendingAction).delete()
|
||||
session.commit()
|
||||
session.close()
|
||||
|
||||
with TestClient(app) as client:
|
||||
with client.websocket_connect("/ws/transcript") as socket:
|
||||
socket.send_json(
|
||||
{
|
||||
"speaker": "Planner",
|
||||
"timestamp": "2026-06-26T12:00:00Z",
|
||||
"text": "Create a task for Maria to confirm the photographer.",
|
||||
}
|
||||
)
|
||||
response = socket.receive_json()
|
||||
|
||||
assert response["ok"] is True
|
||||
assert response["pending_action_id"] is not None
|
||||
|
||||
dashboard = client.get("/dashboard").json()
|
||||
assert any(action["tool"] == "create_task" for action in dashboard["pending_actions"])
|
||||
Reference in New Issue
Block a user