aboutsummaryrefslogtreecommitdiff
path: root/tests/integration/test_setup.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/integration/test_setup.py')
-rw-r--r--tests/integration/test_setup.py73
1 files changed, 0 insertions, 73 deletions
diff --git a/tests/integration/test_setup.py b/tests/integration/test_setup.py
deleted file mode 100644
index b74eebafc..000000000
--- a/tests/integration/test_setup.py
+++ /dev/null
@@ -1,73 +0,0 @@
-from unittest.mock import patch
-
-import pytest
-
-from autogpt.config.ai_config import AIConfig
-from autogpt.setup import generate_aiconfig_automatic, prompt_user
-
-
-@pytest.mark.vcr
-@pytest.mark.requires_openai_api_key
-def test_generate_aiconfig_automatic_default(patched_api_requestor, config):
- user_inputs = [""]
- with patch("autogpt.utils.session.prompt", side_effect=user_inputs):
- ai_config = prompt_user(config)
-
- assert isinstance(ai_config, AIConfig)
- assert ai_config.ai_name is not None
- assert ai_config.ai_role is not None
- assert 1 <= len(ai_config.ai_goals) <= 5
-
-
-@pytest.mark.vcr
-@pytest.mark.requires_openai_api_key
-def test_generate_aiconfig_automatic_typical(patched_api_requestor, config):
- user_prompt = "Help me create a rock opera about cybernetic giraffes"
- ai_config = generate_aiconfig_automatic(user_prompt, config)
-
- assert isinstance(ai_config, AIConfig)
- assert ai_config.ai_name is not None
- assert ai_config.ai_role is not None
- assert 1 <= len(ai_config.ai_goals) <= 5
-
-
-@pytest.mark.vcr
-@pytest.mark.requires_openai_api_key
-def test_generate_aiconfig_automatic_fallback(patched_api_requestor, config):
- user_inputs = [
- "T&GF£OIBECC()!*",
- "Chef-GPT",
- "an AI designed to browse bake a cake.",
- "Purchase ingredients",
- "Bake a cake",
- "",
- "",
- ]
- with patch("autogpt.utils.session.prompt", side_effect=user_inputs):
- ai_config = prompt_user(config)
-
- assert isinstance(ai_config, AIConfig)
- assert ai_config.ai_name == "Chef-GPT"
- assert ai_config.ai_role == "an AI designed to browse bake a cake."
- assert ai_config.ai_goals == ["Purchase ingredients", "Bake a cake"]
-
-
-@pytest.mark.vcr
-@pytest.mark.requires_openai_api_key
-def test_prompt_user_manual_mode(patched_api_requestor, config):
- user_inputs = [
- "--manual",
- "Chef-GPT",
- "an AI designed to browse bake a cake.",
- "Purchase ingredients",
- "Bake a cake",
- "",
- "",
- ]
- with patch("autogpt.utils.session.prompt", side_effect=user_inputs):
- ai_config = prompt_user(config)
-
- assert isinstance(ai_config, AIConfig)
- assert ai_config.ai_name == "Chef-GPT"
- assert ai_config.ai_role == "an AI designed to browse bake a cake."
- assert ai_config.ai_goals == ["Purchase ingredients", "Bake a cake"]