mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-23 00:44:22 +01:00
113 lines
3.1 KiB
Python
113 lines
3.1 KiB
Python
import pytest
|
|
|
|
from autogpt.config import Config
|
|
from autogpt.plugins import (
|
|
blacklist_whitelist_check,
|
|
inspect_zip_for_module,
|
|
scan_plugins,
|
|
)
|
|
|
|
PLUGINS_TEST_DIR = "tests/unit/data/test_plugins"
|
|
PLUGIN_TEST_ZIP_FILE = "Auto-GPT-Plugin-Test-master.zip"
|
|
PLUGIN_TEST_INIT_PY = "Auto-GPT-Plugin-Test-master/src/auto_gpt_vicuna/__init__.py"
|
|
PLUGIN_TEST_OPENAI = "https://weathergpt.vercel.app/"
|
|
|
|
|
|
def test_inspect_zip_for_module():
|
|
result = inspect_zip_for_module(str(f"{PLUGINS_TEST_DIR}/{PLUGIN_TEST_ZIP_FILE}"))
|
|
assert result == PLUGIN_TEST_INIT_PY
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_blacklist_whitelist_check():
|
|
class MockConfig:
|
|
plugins_blacklist = ["BadPlugin"]
|
|
plugins_whitelist = ["GoodPlugin"]
|
|
|
|
return MockConfig()
|
|
|
|
|
|
def test_blacklist_whitelist_check_blacklist(
|
|
mock_config_blacklist_whitelist_check, monkeypatch
|
|
):
|
|
monkeypatch.setattr("builtins.input", lambda _: "y")
|
|
assert not blacklist_whitelist_check(
|
|
"BadPlugin", mock_config_blacklist_whitelist_check
|
|
)
|
|
|
|
|
|
def test_blacklist_whitelist_check_whitelist(
|
|
mock_config_blacklist_whitelist_check, monkeypatch
|
|
):
|
|
monkeypatch.setattr("builtins.input", lambda _: "y")
|
|
assert blacklist_whitelist_check(
|
|
"GoodPlugin", mock_config_blacklist_whitelist_check
|
|
)
|
|
|
|
|
|
def test_blacklist_whitelist_check_user_input_yes(
|
|
mock_config_blacklist_whitelist_check, monkeypatch
|
|
):
|
|
monkeypatch.setattr("builtins.input", lambda _: "y")
|
|
assert blacklist_whitelist_check(
|
|
"UnknownPlugin", mock_config_blacklist_whitelist_check
|
|
)
|
|
|
|
|
|
def test_blacklist_whitelist_check_user_input_no(
|
|
mock_config_blacklist_whitelist_check, monkeypatch
|
|
):
|
|
monkeypatch.setattr("builtins.input", lambda _: "n")
|
|
assert not blacklist_whitelist_check(
|
|
"UnknownPlugin", mock_config_blacklist_whitelist_check
|
|
)
|
|
|
|
|
|
def test_blacklist_whitelist_check_user_input_invalid(
|
|
mock_config_blacklist_whitelist_check, monkeypatch
|
|
):
|
|
monkeypatch.setattr("builtins.input", lambda _: "invalid")
|
|
assert not blacklist_whitelist_check(
|
|
"UnknownPlugin", mock_config_blacklist_whitelist_check
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def config_with_plugins():
|
|
cfg = Config()
|
|
cfg.plugins_dir = PLUGINS_TEST_DIR
|
|
cfg.plugins_openai = ["https://weathergpt.vercel.app/"]
|
|
return cfg
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_openai_plugin():
|
|
class MockConfig:
|
|
plugins_dir = PLUGINS_TEST_DIR
|
|
plugins_openai = [PLUGIN_TEST_OPENAI]
|
|
plugins_blacklist = ["AutoGPTPVicuna"]
|
|
plugins_whitelist = [PLUGIN_TEST_OPENAI]
|
|
|
|
return MockConfig()
|
|
|
|
|
|
def test_scan_plugins_openai(mock_config_openai_plugin):
|
|
result = scan_plugins(mock_config_openai_plugin, debug=True)
|
|
assert len(result) == 1
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_generic_plugin():
|
|
class MockConfig:
|
|
plugins_dir = PLUGINS_TEST_DIR
|
|
plugins_openai = []
|
|
plugins_blacklist = []
|
|
plugins_whitelist = ["AutoGPTPVicuna"]
|
|
|
|
return MockConfig()
|
|
|
|
|
|
def test_scan_plugins_generic(mock_config_generic_plugin):
|
|
result = scan_plugins(mock_config_generic_plugin, debug=True)
|
|
assert len(result) == 1
|