chore: MagicMock in test_workforce_single_agent.py (#3135)

Author: Wendong-Fan

camel/environments/tic_tac_toe.py

@@ -483,7 +483,7 @@ def check_winner(board: List[str]) -> Optional[Literal["X", "O", "draw"]]:
         # Check all win combinations.
         for a, b, c in TicTacToeEnv.WIN_COMBINATIONS:
             if board[a] != " " and board[a] == board[b] == board[c]:
-                return board[a]
+                return board[a]  # type: ignore[return-value]
         # Check for draw.
         if all(cell != " " for cell in board):
             return "draw"

test/workforce/test_workforce_single_agent.py

@@ -13,6 +13,7 @@
 # ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
 import time
 from typing import List
+from unittest.mock import patch
 
 import pytest
 
@@ -102,7 +103,11 @@ async def test_graceful_shutdown_one_second_timeout():
 
 
 @pytest.mark.asyncio
-async def test_get_dep_tasks_info():
+@patch('camel.tasks.task.Task.decompose')
+@patch(
+    'camel.societies.workforce.single_agent_worker.SingleAgentWorker._process_task'
+)
+async def test_get_dep_tasks_info(mock_process_task, mock_decompose):
     """Original test for backwards compatibility."""
     sys_msg = BaseMessage.make_assistant_message(
         role_name="programmer",
@@ -115,7 +120,18 @@ async def test_get_dep_tasks_info():
         id='0',
     )
 
-    subtasks = human_task.decompose(agent)  # TODO: use MagicMock
-    await test_worker._process_task(
-        human_task, subtasks
-    )  # TODO: use MagicMock
+    # Configure the mocks
+    mock_subtasks = [
+        Task(content="Task 1", id="1"),
+        Task(content="Task 2", id="2"),
+    ]
+    mock_decompose.return_value = mock_subtasks
+    mock_process_task.return_value = TaskState.DONE
+
+    # Execute the test
+    subtasks = human_task.decompose(agent)
+    await test_worker._process_task(human_task, subtasks)
+
+    # Verify the mocks were called
+    mock_decompose.assert_called_once_with(agent)
+    mock_process_task.assert_called_once_with(human_task, mock_subtasks)