refactor: restructure to make sure the Bel agents are also part of BDI.

ref: N25B-257
This commit is contained in:
Björn Otgaar
2025-11-12 12:01:37 +01:00
parent dfebe6f772
commit 9365f109ab
17 changed files with 46 additions and 40 deletions

View File

@@ -4,7 +4,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
import pytest
from spade.message import Message
from control_backend.agents.bel_agents.bel_text_extract_agent.behaviours.text_belief_extractor import ( # noqa: E501, We can't shorten this import.
from control_backend.agents.bdi_agents.bdi_text_belief_agent.behaviours.text_belief_extractor import ( # noqa: E501, We can't shorten this import.
BeliefFromText,
)
@@ -18,14 +18,14 @@ def mock_settings():
# Create a mock object that mimics the nested structure
settings_mock = MagicMock()
settings_mock.agent_settings.per_transcription_agent_name = "transcriber"
settings_mock.agent_settings.bel_collector_agent_name = "collector"
settings_mock.agent_settings.bdi_belief_collector_agent_name = "collector"
settings_mock.agent_settings.host = "fake.host"
# Use patch to replace the settings object during the test
# Adjust 'control_backend.behaviours.belief_from_text.settings' to where
# your behaviour file imports it from.
with patch(
"control_backend.agents.bel_agents.bel_text_extract_agent.behaviours.text_belief_extractor.settings",
"control_backend.agents.bdi_agents.bdi_text_belief_agent.behaviours.text_belief_extractor.settings",
settings_mock,
):
yield settings_mock
@@ -122,7 +122,7 @@ async def test_run_message_from_transcriber_demo(behavior, mock_settings, monkey
assert (
sent_msg.to
== mock_settings.agent_settings.bel_collector_agent_name
== mock_settings.agent_settings.bdi_belief_collector_agent_name
+ "@"
+ mock_settings.agent_settings.host
)
@@ -162,7 +162,7 @@ async def test_process_transcription_success(behavior, mock_settings):
# 2. Inspect the sent message
sent_msg: Message = behavior.send.call_args[0][0]
expected_to = (
mock_settings.agent_settings.bel_collector_agent_name
mock_settings.agent_settings.bdi_belief_collector_agent_name
+ "@"
+ mock_settings.agent_settings.host
)