4ec06ece56
- Implemented a new dependency pipeline for plugins to manage Python package dependencies, including conflict detection and automatic installation of missing dependencies. - Introduced an HTML rendering service that utilizes existing browsers to render HTML content as PNG images, with support for various configurations and error handling.
97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
"""插件运行时浏览器渲染能力测试。"""
|
|
|
|
from typing import Optional
|
|
|
|
import pytest
|
|
|
|
from src.plugin_runtime.integration import PluginRuntimeManager
|
|
from src.plugin_runtime.host.supervisor import PluginSupervisor
|
|
from src.services.html_render_service import HtmlRenderRequest, HtmlRenderResult
|
|
|
|
|
|
class _FakeRenderService:
|
|
"""用于替代真实浏览器渲染服务的测试桩。"""
|
|
|
|
def __init__(self) -> None:
|
|
"""初始化测试桩。"""
|
|
|
|
self.last_request: Optional[HtmlRenderRequest] = None
|
|
|
|
async def render_html_to_png(self, request: HtmlRenderRequest) -> HtmlRenderResult:
|
|
"""记录请求并返回固定的渲染结果。
|
|
|
|
Args:
|
|
request: 当前渲染请求。
|
|
|
|
Returns:
|
|
HtmlRenderResult: 固定的测试渲染结果。
|
|
"""
|
|
|
|
self.last_request = request
|
|
return HtmlRenderResult(
|
|
image_base64="ZmFrZS1pbWFnZQ==",
|
|
mime_type="image/png",
|
|
width=640,
|
|
height=480,
|
|
render_ms=12,
|
|
)
|
|
|
|
|
|
def test_render_capability_is_registered() -> None:
|
|
"""Host 注册能力时应包含 render.html2png。"""
|
|
|
|
manager = PluginRuntimeManager()
|
|
supervisor = PluginSupervisor(plugin_dirs=[])
|
|
|
|
manager._register_capability_impls(supervisor)
|
|
|
|
assert "render.html2png" in supervisor.capability_service.list_capabilities()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_render_capability_forwards_request(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
"""render.html2png 应将请求透传给浏览器渲染服务。"""
|
|
|
|
from src.plugin_runtime.capabilities import render as render_capability_module
|
|
|
|
fake_service = _FakeRenderService()
|
|
monkeypatch.setattr(render_capability_module, "get_html_render_service", lambda: fake_service)
|
|
|
|
manager = PluginRuntimeManager()
|
|
result = await manager._cap_render_html2png(
|
|
"demo.plugin",
|
|
"render.html2png",
|
|
{
|
|
"html": "<body><div id='card'>hello</div></body>",
|
|
"selector": "#card",
|
|
"viewport": {"width": 1024, "height": 768},
|
|
"device_scale_factor": 1.5,
|
|
"full_page": False,
|
|
"omit_background": True,
|
|
"wait_until": "networkidle",
|
|
"wait_for_selector": "#card",
|
|
"wait_for_timeout_ms": 150,
|
|
"timeout_ms": 3000,
|
|
"allow_network": True,
|
|
},
|
|
)
|
|
|
|
assert result == {
|
|
"success": True,
|
|
"result": {
|
|
"image_base64": "ZmFrZS1pbWFnZQ==",
|
|
"mime_type": "image/png",
|
|
"width": 640,
|
|
"height": 480,
|
|
"render_ms": 12,
|
|
},
|
|
}
|
|
assert fake_service.last_request is not None
|
|
assert fake_service.last_request.selector == "#card"
|
|
assert fake_service.last_request.viewport_width == 1024
|
|
assert fake_service.last_request.viewport_height == 768
|
|
assert fake_service.last_request.device_scale_factor == 1.5
|
|
assert fake_service.last_request.omit_background is True
|
|
assert fake_service.last_request.wait_until == "networkidle"
|
|
assert fake_service.last_request.allow_network is True
|