"""Sentence completion over the API, from the user's sent mail."""

from __future__ import annotations

from typing import Any

import pytest

from app.api import completion as api_completion
from app.completion import build_index_job
from tests.api.conftest import HDR, MailCtx, raw_mail


async def _suggest(ctx: MailCtx, before: str) -> str | None:
    resp = await ctx.client.post("/api/completion", json={"before": before}, headers=HDR)
    assert resp.status_code == 200, resp.text
    return resp.json()["suggestion"]


async def test_suggests_from_sent_mail(mail_ctx: MailCtx, monkeypatch: pytest.MonkeyPatch) -> None:
    queued: list[tuple[Any, ...]] = []

    async def fake_enqueue(name: str, *args: Any, **kwargs: Any) -> None:
        queued.append((name, *args))

    monkeypatch.setattr(api_completion, "enqueue_job", fake_enqueue)
    body = "Hi Anna,\nThe renders will be ready by Thursday evening.\n\nOn Mon Anna wrote:\n> ok"
    for uid in (1, 2):
        await mail_ctx.deliver(
            raw_mail("Renders", sender="me@example.com").replace(b"Renders body", body.encode()),
            uid,
            folder="sent",
        )

    # No index yet: a build is asked for, and the starter phrases still answer.
    assert await _suggest(mail_ctx, "I look forward ") == "to hearing from you."
    assert queued and queued[0][0] == "build_index_job"

    assert await build_index_job({}, str(mail_ctx.user_id)) > 0
    assert await _suggest(mail_ctx, "The renders will be re") == "ady by Thursday evening."
    assert await _suggest(mail_ctx, "> ok") is None  # quoted text was not learned