tests/test_browser_screenshot_artifact_safety.py

Rama dev · réplica de lectura del repositorio original.

import asyncio
import json

import pytest
from src.agent_tools.web_tools import PrivateBrowserTool


@pytest.mark.parametrize("name", ["output.html", "source.svg", "report.pdf", "notes.txt"])
def test_screenshot_cannot_overwrite_nonimage_artifact(monkeypatch, tmp_path, name):
    source = tmp_path / name
    source.write_bytes(b"original artifact")
    monkeypatch.setattr("src.tool_execution.get_active_workspace", lambda: str(tmp_path))
    monkeypatch.setattr(PrivateBrowserTool, "_resolve_workspace_path", staticmethod(lambda value: source))

    async def forbidden(*args, **kwargs):
        raise AssertionError("browser must not launch with nonimage destination")

    monkeypatch.setattr(asyncio, "create_subprocess_exec", forbidden)
    result = asyncio.run(PrivateBrowserTool().execute(
        json.dumps({"action": "screenshot", "path": "/workspace/" + name}),
        {"session_id": "artifact-safety"},
    ))
    assert result["exit_code"] == 1
    assert result["failure_kind"] == "browser_page_authority_unavailable"
    assert result["executed"] is False
    assert source.read_bytes() == b"original artifact"