"""워커 루프 테스트 — drain→DONE, 실패→재시도→dead, reaper 회수 후 재처리, NOTIFY 깨움.
핸들러는 fake(브라우저 없이) — 워커 로직만 결정론적으로 검증한다."""

import asyncio

import pytest_asyncio
from sqlalchemy import text

from common.enums import JobStatus, JobType
from crud.job_crud import JobQueue
from worker.notify import JobListener
from worker.runner import Worker


@pytest_asyncio.fixture
async def q(db_engine):
    async with db_engine.begin() as conn:
        await conn.execute(text("TRUNCATE job"))
    return JobQueue()


async def test_worker_drains_all_to_done(q):
    for i in range(3):
        await q.enqueue(JobType.SEARCH.value, {"product_name": f"item{i}"})

    async def handler(job):
        return {"ok": True, "q": job["payload"]["product_name"]}

    processed = await Worker("w1", q, handler).drain()
    assert processed == 3
    assert (await q.counts())["DONE"] == 3


async def test_worker_failure_retries_then_dead(q):
    await q.enqueue(JobType.SEARCH.value, {"product_name": "x"}, max_attempts=2)

    async def boom(job):
        raise RuntimeError("nope")

    w = Worker("w1", q, boom, backoff_fn=lambda a: 0)  # 백오프 0 → 즉시 재시도 가능
    assert await w.process_one() is True                # 1/2 실패 → PENDING
    assert (await q.counts())["PENDING"] == 1
    assert await w.process_one() is True                # 2/2 실패 → DEAD
    counts = await q.counts()
    assert counts["DEAD"] == 1 and counts["PENDING"] == 0
    assert await w.process_one() is False               # DEAD 는 claim 대상 아님


async def test_reaper_reclaims_then_worker_reprocesses(q):
    jid = await q.enqueue(JobType.SEARCH.value, {"product_name": "x"})
    await q.claim("dead-worker", lease_sec=1)   # 점유 후 사망 흉내
    await asyncio.sleep(1.3)
    assert jid in await q.reap()                 # 회수 → PENDING

    async def handler(job):
        return {"ok": True}

    assert await Worker("w2", q, handler).process_one() is True
    assert (await q.counts())["DONE"] == 1


async def test_enqueue_notifies_listener(q):
    listener = JobListener()
    await listener.start()
    try:
        await q.enqueue(JobType.SEARCH.value, {"product_name": "x"})  # pg_notify 발생
        assert await listener.wait(3.0) is True                       # 즉시 깨어남
    finally:
        await listener.close()