from app.api.views.agent_run_sse import encode_events, recovery_events


def test_first_connection_emits_unicode_deltas_only_after_commit():
    content = "基金😀\n结果" * 100
    events = recovery_events(run_id="r", trace_id="t", status="succeeded", error_code=None,
                             content=content, tool_calls=None, replay=False, chunk_size=11)
    assert events[0][0] == "start" and events[-1][0] == "done"
    assert "".join(payload["content"] for name, payload in events if name == "delta") == content
    assert all(len(payload["content"]) <= 11 for name, payload in events if name == "delta")
    ids = [line for event in encode_events("r", events)
           for line in event.splitlines() if line.startswith("id:")]
    assert len(set(ids)) == len(ids)
    pending = recovery_events(run_id="r", trace_id="t", status="running", error_code=None,
                              content=content, tool_calls=None, replay=False)
    assert [name for name, _ in pending] == ["start"]


def test_empty_result_still_emits_delta():
    events = recovery_events(run_id="r", trace_id="t", status="succeeded", error_code=None,
                             content="", tool_calls=None, replay=False)
    assert [name for name, _ in events] == ["start", "delta", "done"]


def test_pending_run_never_emits_result_events() -> None:
    events = recovery_events(
        run_id="run",
        trace_id="trace",
        status="running",
        error_code=None,
        content=None,
        tool_calls=None,
    )
    assert [name for name, _ in events] == ["start"]


def test_succeeded_run_emits_tools_and_result_after_terminal_state() -> None:
    events = recovery_events(
        run_id="run",
        trace_id="trace",
        status="succeeded",
        error_code=None,
        content="answer",
        tool_calls={"calls": []},
    )
    assert [name for name, _ in events] == ["start", "tools", "replace", "done"]


def test_failed_run_emits_error_without_result() -> None:
    events = recovery_events(
        run_id="run",
        trace_id="trace",
        status="failed",
        error_code="MODEL_TIMEOUT",
        content=None,
        tool_calls=None,
    )
    assert [name for name, _ in events] == ["start", "error"]


def test_client_disconnect_does_not_change_recoverable_terminal_events() -> None:
    events = recovery_events(
        run_id="run",
        trace_id="trace",
        status="succeeded",
        error_code=None,
        content="answer",
        tool_calls=None,
    )
    stream = encode_events("run", events)
    first = next(stream)
    assert "event: start" in first
    del stream
    recovered = list(
        encode_events(
            "run",
            recovery_events(
                run_id="run",
                trace_id="trace",
                status="succeeded",
                error_code=None,
                content="answer",
                tool_calls=None,
            ),
        )
    )
    assert any("event: replace" in item for item in recovered)
    assert any("event: done" in item for item in recovered)