test: make tests work again after changing Program schema

ref: N25B-380

test/unit/agents/bdi/test_bdi_program_manager.py

@@ -1,6 +1,6 @@
 import asyncio
-import json
 import sys
+import uuid
 from unittest.mock import AsyncMock
 
 import pytest
@@ -8,31 +8,45 @@ import pytest
 from control_backend.agents.bdi.bdi_program_manager import BDIProgramManager
 from control_backend.core.agent_system import InternalMessage
 from control_backend.schemas.belief_message import BeliefMessage
-from control_backend.schemas.program import Program
+from control_backend.schemas.program import BasicNorm, Goal, Phase, Plan, Program
 
 # Fix Windows Proactor loop for zmq
 if sys.platform.startswith("win"):
     asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
 
 
-def make_valid_program_json(norm="N1", goal="G1"):
-    return json.dumps(
-        {
-            "phases": [
-                {
-                    "id": "phase1",
-                    "label": "Phase 1",
-                    "triggers": [],
-                    "norms": [{"id": "n1", "label": "Norm 1", "norm": norm}],
-                    "goals": [
-                        {"id": "g1", "label": "Goal 1", "description": goal, "achieved": False}
-                    ],
-                }
-            ]
-        }
-    )
+def make_valid_program_json(norm="N1", goal="G1") -> str:
+    return Program(
+        phases=[
+            Phase(
+                id=uuid.uuid4(),
+                name="Basic Phase",
+                norms=[
+                    BasicNorm(
+                        id=uuid.uuid4(),
+                        name=norm,
+                        norm=norm,
+                    ),
+                ],
+                goals=[
+                    Goal(
+                        id=uuid.uuid4(),
+                        name=goal,
+                        plan=Plan(
+                            id=uuid.uuid4(),
+                            name="Goal Plan",
+                            steps=[],
+                        ),
+                        can_fail=False,
+                    ),
+                ],
+                triggers=[],
+            ),
+        ],
+    ).model_dump_json()
 
 
+@pytest.mark.skip(reason="Functionality being rebuilt.")
 @pytest.mark.asyncio
 async def test_send_to_bdi():
     manager = BDIProgramManager(name="program_manager_test")
@@ -73,5 +87,5 @@ async def test_receive_programs_valid_and_invalid():
     # Only valid Program should have triggered _send_to_bdi
     assert manager._send_to_bdi.await_count == 1
     forwarded: Program = manager._send_to_bdi.await_args[0][0]
-    assert forwarded.phases[0].norms[0].norm == "N1"
-    assert forwarded.phases[0].goals[0].description == "G1"
+    assert forwarded.phases[0].norms[0].name == "N1"
+    assert forwarded.phases[0].goals[0].name == "G1"

test/unit/agents/bdi/test_text_extractor.py

@@ -45,10 +45,10 @@ async def test_handle_message_from_transcriber(agent, mock_settings):
 
 
 @pytest.mark.asyncio
-async def test_process_transcription_demo(agent, mock_settings):
+async def test_process_user_said(agent, mock_settings):
     transcription = "this is a test"
 
-    await agent._process_transcription_demo(transcription)
+    await agent._user_said(transcription)
 
     agent.send.assert_awaited_once()  # noqa  # `agent.send` has no such property, but we mock it.
     sent: InternalMessage = agent.send.call_args.args[0]  # noqa
@@ -56,10 +56,3 @@ async def test_process_transcription_demo(agent, mock_settings):
     assert sent.thread == "beliefs"
     parsed = json.loads(sent.body)
     assert parsed["beliefs"]["user_said"] == [transcription]
-
-
-@pytest.mark.asyncio
-async def test_setup_initializes_beliefs(agent):
-    """Covers the setup method and ensures beliefs are initialized."""
-    await agent.setup()
-    assert agent.beliefs == {"mood": ["X"], "car": ["Y"]}

test/unit/agents/llm/test_llm_agent.py

@@ -66,7 +66,7 @@ async def test_llm_processing_success(mock_httpx_client, mock_settings):
     # "Hello world." constitutes one sentence/chunk based on punctuation split
     # The agent should call send once with the full sentence
     assert agent.send.called
-    args = agent.send.call_args[0][0]
+    args = agent.send.call_args_list[0][0][0]
     assert args.to == mock_settings.agent_settings.bdi_core_name
     assert "Hello world." in args.body