Skip to content

Commit 93b1642

Browse files
committed
add missing Moc value
1 parent 2ee09f2 commit 93b1642

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/coordinator/test_coordinator_core.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -244,22 +244,26 @@ async def test_process_join_game_action_success(initialized_coordinator):
244244
# Minimal working state
245245
initialized_coordinator._starting_positions_per_role = {"Attacker": MagicMock()}
246246
initialized_coordinator._goal_description_per_role = {"Attacker": "Goal"}
247+
initialized_coordinator._win_conditions_per_role = {"Attacker": MagicMock()}
247248
initialized_coordinator._steps_limit_per_role = {"Attacker": 10}
248249
initialized_coordinator._CONFIG_FILE_HASH = "abc123"
249250
initialized_coordinator._min_required_players = 1
250251
initialized_coordinator._agent_status = {agent: MagicMock()}
251252
initialized_coordinator._episode_start_event.set() # Prevent wait
252253

253254
action = MagicMock()
254-
action.parameters = {"agent_info": MagicMock(name="AgentX", role="Attacker")}
255+
agent_info = MagicMock()
256+
agent_info.name = "AgentX"
257+
agent_info.role = "Attacker"
258+
action.parameters = {"agent_info": agent_info}
255259
observation = SimpleNamespace(
256260
state=SimpleNamespace(as_dict={}), # empty dict works here
257261
reward=0,
258262
end=False,
259263
info={}
260264
)
261265

262-
with patch.object(initialized_coordinator, "register_agent", new_callable=AsyncMock, return_value=MagicMock()), \
266+
with patch.object(initialized_coordinator, "register_agent", new_callable=AsyncMock, return_value=(MagicMock(),MagicMock())), \
263267
patch.object(initialized_coordinator, "_initialize_new_player", return_value=observation), \
264268
patch.object(initialized_coordinator.logger, "info"), \
265269
patch.object(initialized_coordinator.logger, "debug"):

0 commit comments

Comments
 (0)