tests/test_eval_progress.py - master - Vibek/Agent_GG

tests/test_eval_progress.py

66 строк2 KB

Vibek

прогресс бар оценки добавлен и работает корректно

24 апр 2026, 19:10

24 апр 2026, 19:101e17055

from __future__ import annotations

import sys
from pathlib import Path

# Корень репозитория на path для `import agent`
_ROOT = Path(__file__).resolve().parents[1]
if str(_ROOT) not in sys.path:
    sys.path.insert(0, str(_ROOT))

from agent.tools.eval_progress import (  # noqa: E402
    EvaluationProgressTracker,
    bind_eval_progress,
    get_active_eval_progress_id,
)


def test_eval_progress_tracker_contract_payload() -> None:
    tracker = EvaluationProgressTracker()
    payload = tracker.get("missing-id")
    assert set(payload.keys()) == {
        "active",
        "status",
        "percent",
        "stage",
        "detail",
        "created_at",
        "updated_at",
    }
    assert payload["active"] is False
    assert payload["status"] == "idle"
    assert payload["percent"] == 0


def test_eval_progress_tracker_lifecycle() -> None:
    tracker = EvaluationProgressTracker()
    progress_id = "eval-test-1"
    tracker.start(progress_id, listing_id=42)
    started = tracker.get(progress_id)
    assert started["active"] is True
    assert started["status"] == "running"
    assert started["percent"] == 5

    tracker.update(
        progress_id,
        percent=61,
        stage="Аудит LLM",
        detail="Проверяю критерий price_market.",
    )
    updated = tracker.get(progress_id)
    assert updated["percent"] == 61
    assert updated["stage"] == "Аудит LLM"
    assert updated["detail"] == "Проверяю критерий price_market."

    tracker.complete(progress_id, detail="Отчет сформирован.")
    completed = tracker.get(progress_id)
    assert completed["percent"] == 100
    assert completed["status"] == "completed"
    assert completed["stage"] == "Оценка завершена"


def test_bind_eval_progress_sets_context_var() -> None:
    assert get_active_eval_progress_id() is None
    with bind_eval_progress("eval-ctx-1"):
        assert get_active_eval_progress_id() == "eval-ctx-1"
    assert get_active_eval_progress_id() is None