Merge remote-tracking branch 'origin/main' into better-whitebox
# Conflicts: # strix/agents/StrixAgent/strix_agent.py # strix/agents/StrixAgent/system_prompt.jinja
This commit is contained in:
94
tests/tools/test_tool_registration_modes.py
Normal file
94
tests/tools/test_tool_registration_modes.py
Normal file
@@ -0,0 +1,94 @@
|
||||
import importlib
|
||||
import sys
|
||||
from types import ModuleType
|
||||
from typing import Any
|
||||
|
||||
from strix.config import Config
|
||||
from strix.tools.registry import clear_registry
|
||||
|
||||
|
||||
def _empty_config_load(_cls: type[Config]) -> dict[str, dict[str, str]]:
|
||||
return {"env": {}}
|
||||
|
||||
|
||||
def _reload_tools_module() -> ModuleType:
|
||||
clear_registry()
|
||||
|
||||
for name in list(sys.modules):
|
||||
if name == "strix.tools" or name.startswith("strix.tools."):
|
||||
sys.modules.pop(name, None)
|
||||
|
||||
return importlib.import_module("strix.tools")
|
||||
|
||||
|
||||
def test_non_sandbox_registers_agents_graph_but_not_browser_or_web_search_when_disabled(
|
||||
monkeypatch: Any,
|
||||
) -> None:
|
||||
monkeypatch.setenv("STRIX_SANDBOX_MODE", "false")
|
||||
monkeypatch.setenv("STRIX_DISABLE_BROWSER", "true")
|
||||
monkeypatch.delenv("PERPLEXITY_API_KEY", raising=False)
|
||||
monkeypatch.setattr(Config, "load", classmethod(_empty_config_load))
|
||||
|
||||
tools = _reload_tools_module()
|
||||
names = set(tools.get_tool_names())
|
||||
|
||||
assert "create_agent" in names
|
||||
assert "browser_action" not in names
|
||||
assert "web_search" not in names
|
||||
|
||||
|
||||
def test_sandbox_registers_sandbox_tools_but_not_non_sandbox_tools(
|
||||
monkeypatch: Any,
|
||||
) -> None:
|
||||
monkeypatch.setenv("STRIX_SANDBOX_MODE", "true")
|
||||
monkeypatch.setenv("STRIX_DISABLE_BROWSER", "true")
|
||||
monkeypatch.delenv("PERPLEXITY_API_KEY", raising=False)
|
||||
monkeypatch.setattr(Config, "load", classmethod(_empty_config_load))
|
||||
|
||||
tools = _reload_tools_module()
|
||||
names = set(tools.get_tool_names())
|
||||
|
||||
assert "terminal_execute" in names
|
||||
assert "python_action" in names
|
||||
assert "list_requests" in names
|
||||
assert "create_agent" not in names
|
||||
assert "finish_scan" not in names
|
||||
assert "load_skill" not in names
|
||||
assert "browser_action" not in names
|
||||
assert "web_search" not in names
|
||||
|
||||
|
||||
def test_load_skill_import_does_not_register_create_agent_in_sandbox(
|
||||
monkeypatch: Any,
|
||||
) -> None:
|
||||
monkeypatch.setenv("STRIX_SANDBOX_MODE", "true")
|
||||
monkeypatch.setenv("STRIX_DISABLE_BROWSER", "true")
|
||||
monkeypatch.delenv("PERPLEXITY_API_KEY", raising=False)
|
||||
monkeypatch.setattr(Config, "load", classmethod(_empty_config_load))
|
||||
|
||||
clear_registry()
|
||||
for name in list(sys.modules):
|
||||
if name == "strix.tools" or name.startswith("strix.tools."):
|
||||
sys.modules.pop(name, None)
|
||||
|
||||
load_skill_module = importlib.import_module("strix.tools.load_skill.load_skill_actions")
|
||||
registry = importlib.import_module("strix.tools.registry")
|
||||
|
||||
names_before = set(registry.get_tool_names())
|
||||
assert "load_skill" not in names_before
|
||||
assert "create_agent" not in names_before
|
||||
|
||||
state_type = type(
|
||||
"DummyState",
|
||||
(),
|
||||
{
|
||||
"agent_id": "agent_test",
|
||||
"context": {},
|
||||
"update_context": lambda self, key, value: self.context.__setitem__(key, value),
|
||||
},
|
||||
)
|
||||
result = load_skill_module.load_skill(state_type(), "nmap")
|
||||
|
||||
names_after = set(registry.get_tool_names())
|
||||
assert "create_agent" not in names_after
|
||||
assert result["success"] is False
|
||||
Reference in New Issue
Block a user