126 lines
3.7 KiB
Python
126 lines
3.7 KiB
Python
import json
|
|
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
from fastapi import FastAPI
|
|
from fastapi.testclient import TestClient
|
|
|
|
from control_backend.api.v1.endpoints import program
|
|
from control_backend.schemas.program import Program
|
|
|
|
|
|
@pytest.fixture
|
|
def app():
|
|
"""Create a FastAPI app with the /program route and mock socket."""
|
|
app = FastAPI()
|
|
app.include_router(program.router)
|
|
return app
|
|
|
|
|
|
@pytest.fixture
|
|
def client(app):
|
|
"""Create a TestClient."""
|
|
return TestClient(app)
|
|
|
|
|
|
def make_valid_program_dict():
|
|
"""Helper to create a valid Program JSON structure."""
|
|
return {
|
|
"phases": [
|
|
{
|
|
"id": "phase1",
|
|
"label": "basephase",
|
|
"norms": [{"id": "n1", "label": "norm", "norm": "be nice"}],
|
|
"goals": [
|
|
{"id": "g1", "label": "goal", "description": "test goal", "achieved": False}
|
|
],
|
|
"triggers": [
|
|
{
|
|
"id": "t1",
|
|
"label": "trigger",
|
|
"type": "keywords",
|
|
"keywords": [
|
|
{"id": "kw1", "keyword": "keyword1"},
|
|
{"id": "kw2", "keyword": "keyword2"},
|
|
],
|
|
},
|
|
],
|
|
}
|
|
]
|
|
}
|
|
|
|
|
|
def test_receive_program_success(client):
|
|
"""Valid Program JSON should be parsed and sent through the socket."""
|
|
mock_pub_socket = AsyncMock()
|
|
client.app.state.endpoints_pub_socket = mock_pub_socket
|
|
|
|
program_dict = make_valid_program_dict()
|
|
|
|
response = client.post("/program", json=program_dict)
|
|
|
|
assert response.status_code == 202
|
|
assert response.json() == {"status": "Program parsed"}
|
|
|
|
# Verify socket call
|
|
mock_pub_socket.send_multipart.assert_awaited_once()
|
|
args, kwargs = mock_pub_socket.send_multipart.await_args
|
|
|
|
assert args[0][0] == b"program"
|
|
|
|
sent_bytes = args[0][1]
|
|
sent_obj = json.loads(sent_bytes.decode())
|
|
|
|
expected_obj = Program.model_validate(program_dict).model_dump()
|
|
assert sent_obj == expected_obj
|
|
|
|
|
|
def test_receive_program_invalid_json(client):
|
|
"""
|
|
Invalid JSON (malformed) -> FastAPI never calls endpoint.
|
|
It returns a 422 Unprocessable Entity.
|
|
"""
|
|
mock_pub_socket = AsyncMock()
|
|
client.app.state.endpoints_pub_socket = mock_pub_socket
|
|
|
|
# FastAPI only accepts valid JSON bodies, so send raw string
|
|
response = client.post("/program", content="{invalid json}")
|
|
|
|
assert response.status_code == 422
|
|
mock_pub_socket.send_multipart.assert_not_called()
|
|
|
|
|
|
def test_receive_program_invalid_deep_structure(client):
|
|
"""
|
|
Valid JSON but schema invalid -> Pydantic throws validation error -> 422.
|
|
"""
|
|
mock_pub_socket = AsyncMock()
|
|
client.app.state.endpoints_pub_socket = mock_pub_socket
|
|
|
|
# Missing "value" in norms element
|
|
bad_program = {
|
|
"phases": [
|
|
{
|
|
"id": "phase1",
|
|
"name": "deepfail",
|
|
"nextPhaseId": "phase2",
|
|
"phaseData": {
|
|
"norms": [
|
|
{"id": "n1", "name": "norm"} # INVALID: missing "value"
|
|
],
|
|
"goals": [
|
|
{"id": "g1", "name": "goal", "description": "desc", "achieved": False}
|
|
],
|
|
"triggers": [
|
|
{"id": "t1", "label": "trigger", "type": "keyword", "value": ["start"]}
|
|
],
|
|
},
|
|
}
|
|
]
|
|
}
|
|
|
|
response = client.post("/program", json=bad_program)
|
|
|
|
assert response.status_code == 422
|
|
mock_pub_socket.send_multipart.assert_not_called()
|