mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-25 09:54:23 +01:00
bfbe613960
Additional changes: * Improve typing * Modularize message history memory & fix/refactor lots of things * Fix summarization * Move memory relevance calculation to MemoryItem & improve test * Fix import warnings in web_selenium.py * Remove `memory_add` ghost command * Implement overlap in `split_text` * Move memory tests into subdirectory * Remove deprecated `get_ada_embedding()` and helpers * Fix used token calculation in `chat_with_ai` * Replace Message TypedDict by dataclass * Fix AgentManager singleton issues in tests --------- Co-authored-by: Auto-GPT-Bot <github-bot@agpt.co>
75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
import pytest
|
|
|
|
from autogpt.config import Config
|
|
from autogpt.plugins import denylist_allowlist_check, inspect_zip_for_modules
|
|
|
|
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"
|
|
|
|
|
|
def test_inspect_zip_for_modules():
|
|
result = inspect_zip_for_modules(str(f"{PLUGINS_TEST_DIR}/{PLUGIN_TEST_ZIP_FILE}"))
|
|
assert result == [PLUGIN_TEST_INIT_PY]
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_denylist_allowlist_check():
|
|
class MockConfig:
|
|
"""Mock config object for testing the denylist_allowlist_check function"""
|
|
|
|
plugins_denylist = ["BadPlugin"]
|
|
plugins_allowlist = ["GoodPlugin"]
|
|
authorise_key = "y"
|
|
exit_key = "n"
|
|
|
|
return MockConfig()
|
|
|
|
|
|
def test_denylist_allowlist_check_denylist(
|
|
mock_config_denylist_allowlist_check, monkeypatch
|
|
):
|
|
# Test that the function returns False when the plugin is in the denylist
|
|
monkeypatch.setattr("builtins.input", lambda _: "y")
|
|
assert not denylist_allowlist_check(
|
|
"BadPlugin", mock_config_denylist_allowlist_check
|
|
)
|
|
|
|
|
|
def test_denylist_allowlist_check_allowlist(
|
|
mock_config_denylist_allowlist_check, monkeypatch
|
|
):
|
|
# Test that the function returns True when the plugin is in the allowlist
|
|
monkeypatch.setattr("builtins.input", lambda _: "y")
|
|
assert denylist_allowlist_check("GoodPlugin", mock_config_denylist_allowlist_check)
|
|
|
|
|
|
def test_denylist_allowlist_check_user_input_yes(
|
|
mock_config_denylist_allowlist_check, monkeypatch
|
|
):
|
|
# Test that the function returns True when the user inputs "y"
|
|
monkeypatch.setattr("builtins.input", lambda _: "y")
|
|
assert denylist_allowlist_check(
|
|
"UnknownPlugin", mock_config_denylist_allowlist_check
|
|
)
|
|
|
|
|
|
def test_denylist_allowlist_check_user_input_no(
|
|
mock_config_denylist_allowlist_check, monkeypatch
|
|
):
|
|
# Test that the function returns False when the user inputs "n"
|
|
monkeypatch.setattr("builtins.input", lambda _: "n")
|
|
assert not denylist_allowlist_check(
|
|
"UnknownPlugin", mock_config_denylist_allowlist_check
|
|
)
|
|
|
|
|
|
def test_denylist_allowlist_check_user_input_invalid(
|
|
mock_config_denylist_allowlist_check, monkeypatch
|
|
):
|
|
# Test that the function returns False when the user inputs an invalid value
|
|
monkeypatch.setattr("builtins.input", lambda _: "invalid")
|
|
assert not denylist_allowlist_check(
|
|
"UnknownPlugin", mock_config_denylist_allowlist_check
|
|
)
|