AutoGPT/tests/unit/test_commands.py

"""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