# MIT License
# Copyright (c) 2024
"""最小端到端测试：Dummy LLM 推理。"""

from autodemo.llm import DummyLLM
from autodemo.schema import ControlSnapshot, EventRecord, Rect


def test_dummy_llm_generate() -> None:
    llm = DummyLLM()
    ev = EventRecord(
        kind="mouse_click",
        timestamp=1.0,
        data={"x": 1, "y": 2},
        control=ControlSnapshot(
            AutomationId="btn1", Name="按钮", ClassName="Button", ControlType="Button", BoundingRectangle=Rect(left=0, top=0, right=10, bottom=10)
        ),
    )
    spec = llm.generate([ev])
    assert spec.steps[0]["action"] == "click"
    assert spec.steps[0]["target"]["AutomationId"] == "btn1"