| """Unit tests for the commands module""" |
| from unittest.mock import MagicMock, patch |
|
|
| import pytest |
|
|
| import autogpt.agent.agent_manager as agent_manager |
| from autogpt.app import execute_command, list_agents, start_agent |
|
|
|
|
| @pytest.mark.integration_test |
| def test_make_agent() -> None: |
| """Test the make_agent command""" |
| with patch("openai.ChatCompletion.create") as mock: |
| obj = MagicMock() |
| obj.response.choices[0].messages[0].content = "Test message" |
| mock.return_value = obj |
| start_agent("Test Agent", "chat", "Hello, how are you?", "gpt2") |
| agents = list_agents() |
| assert "List of agents:\n0: chat" == agents |
| start_agent("Test Agent 2", "write", "Hello, how are you?", "gpt2") |
| agents = list_agents() |
| assert "List of agents:\n0: chat\n1: write" == agents |
|
|
