alihan/OpenHands-swe-agent
1
0
Fork 0
mirror of https://github.com/All-Hands-AI/OpenHands.git synced 2024-08-29 01:18:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

(test) Fix test_agent_controller.py mock exceptions (#3577)

Browse Source 
* fix test_agent_controller.py mock exceptions

* revert change to agent_controller.py
This commit is contained in:
tobitege
2024-08-28 19:05:22 +02:00
committed by GitHub
parent 9c39f07430
commit f8c4d1df45
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/unit/test_agent_controller.py
View File

@@ -49,6 +49,7 @@ async def test_set_agent_state(mock_agent, mock_event_stream):
await controller.set_agent_state_to(AgentState.PAUSED)
assert controller.get_agent_state() == AgentState.PAUSED
await controller.close()
@pytest.mark.asyncio
@@ -65,6 +66,7 @@ async def test_on_event_message_action(mock_agent, mock_event_stream):
message_action = MessageAction(content='Test message')
await controller.on_event(message_action)
assert controller.get_agent_state() == AgentState.RUNNING
await controller.close()
@pytest.mark.asyncio
@@ -81,6 +83,7 @@ async def test_on_event_change_agent_state_action(mock_agent, mock_event_stream)
change_state_action = ChangeAgentStateAction(agent_state=AgentState.PAUSED)
await controller.on_event(change_state_action)
assert controller.get_agent_state() == AgentState.PAUSED
await controller.close()
@pytest.mark.asyncio
@@ -97,6 +100,7 @@ async def test_report_error(mock_agent, mock_event_stream):
await controller.report_error(error_message)
assert controller.state.last_error == error_message
controller.event_stream.add_event.assert_called_once()
await controller.close()
@pytest.mark.asyncio
@@ -116,6 +120,7 @@ async def test_step_with_exception(mock_agent, mock_event_stream):
# Verify that report_error was called with the correct error message
controller.report_error.assert_called_once_with('Malformed action')
await controller.close()
@pytest.mark.asyncio
@@ -134,6 +139,7 @@ async def test_step_max_iterations(mock_agent, mock_event_stream):
await controller._step()
assert controller.state.traffic_control_state == TrafficControlState.THROTTLING
assert controller.state.agent_state == AgentState.PAUSED
await controller.close()
@pytest.mark.asyncio
@@ -153,6 +159,7 @@ async def test_step_max_iterations_headless(mock_agent, mock_event_stream):
assert controller.state.traffic_control_state == TrafficControlState.THROTTLING
# In headless mode, throttling results in an error
assert controller.state.agent_state == AgentState.ERROR
await controller.close()
@pytest.mark.asyncio
@@ -172,6 +179,7 @@ async def test_step_max_budget(mock_agent, mock_event_stream):
await controller._step()
assert controller.state.traffic_control_state == TrafficControlState.THROTTLING
assert controller.state.agent_state == AgentState.PAUSED
await controller.close()
@pytest.mark.asyncio
@@ -192,3 +200,4 @@ async def test_step_max_budget_headless(mock_agent, mock_event_stream):
assert controller.state.traffic_control_state == TrafficControlState.THROTTLING
# In headless mode, throttling results in an error
assert controller.state.agent_state == AgentState.ERROR
await controller.close()