mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-15 17:30:13 +08:00
101 lines
2.8 KiB
Python
101 lines
2.8 KiB
Python
import json
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from astrbot.core.agent.run_context import ContextWrapper
|
|
from astrbot.core.tools.computer_tools.shipyard_neo.browser import BrowserExecTool
|
|
from astrbot.core.tools.computer_tools.shipyard_neo.neo_skills import (
|
|
GetExecutionHistoryTool,
|
|
)
|
|
|
|
|
|
class _FakeBrowser:
|
|
async def exec(self, **kwargs):
|
|
return {
|
|
"ok": True,
|
|
"cmd": kwargs["cmd"],
|
|
}
|
|
|
|
|
|
class _FakeSandbox:
|
|
async def get_execution_history(self, **kwargs):
|
|
return {
|
|
"items": [],
|
|
"limit": kwargs["limit"],
|
|
}
|
|
|
|
|
|
def _make_run_context(require_admin: bool, role: str = "member") -> ContextWrapper:
|
|
config_holder = SimpleNamespace(
|
|
get_config=lambda umo: { # noqa: ARG005
|
|
"provider_settings": {
|
|
"computer_use_require_admin": require_admin,
|
|
}
|
|
}
|
|
)
|
|
event = SimpleNamespace(
|
|
role=role,
|
|
unified_msg_origin="qq_official:friend:user-1",
|
|
get_sender_id=lambda: "user-1",
|
|
)
|
|
astr_ctx = SimpleNamespace(context=config_holder, event=event)
|
|
return ContextWrapper(context=astr_ctx)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_browser_tool_allows_non_admin_when_admin_requirement_disabled(
|
|
monkeypatch,
|
|
):
|
|
async def _fake_get_booter(_ctx, _session_id):
|
|
return SimpleNamespace(browser=_FakeBrowser())
|
|
|
|
monkeypatch.setattr(
|
|
"astrbot.core.tools.computer_tools.shipyard_neo.browser.get_booter",
|
|
_fake_get_booter,
|
|
)
|
|
|
|
result = await BrowserExecTool().call(
|
|
_make_run_context(require_admin=False),
|
|
cmd="open https://example.com",
|
|
)
|
|
|
|
assert json.loads(result)["ok"] is True # type: ignore[arg-type]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_neo_skill_tool_allows_non_admin_when_admin_requirement_disabled(
|
|
monkeypatch,
|
|
):
|
|
async def _fake_get_booter(_ctx, _session_id):
|
|
return SimpleNamespace(
|
|
bay_client=object(),
|
|
sandbox=_FakeSandbox(),
|
|
)
|
|
|
|
monkeypatch.setattr(
|
|
"astrbot.core.tools.computer_tools.shipyard_neo.neo_skills.get_booter",
|
|
_fake_get_booter,
|
|
)
|
|
|
|
result = await GetExecutionHistoryTool().call(
|
|
_make_run_context(require_admin=False),
|
|
limit=5,
|
|
)
|
|
|
|
payload = json.loads(result) # type: ignore[arg-type]
|
|
assert payload["items"] == []
|
|
assert payload["limit"] == 5
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_browser_tool_still_denies_non_admin_when_admin_requirement_enabled():
|
|
result = await BrowserExecTool().call(
|
|
_make_run_context(require_admin=True),
|
|
cmd="open https://example.com",
|
|
)
|
|
|
|
assert "Permission denied" in result
|
|
assert "Using browser tools is only allowed for admin users" in result
|
|
assert "User's ID is: user-1" in result
|