"""RunReport 测试."""

from __future__ import annotations

from datetime import datetime, timedelta

import pyflowx as px
from pyflowx.task import TaskResult, TaskSpec, TaskStatus


def _fn() -> int:
    return 1


def _make_result(
    name: str = "a",
    status: TaskStatus = TaskStatus.SUCCESS,
    value: object = 42,
    error: BaseException | None = None,
    duration: float = 0.5,
    attempts: int = 1,
) -> TaskResult[object]:
    """构造测试用 TaskResult 实例."""
    spec: TaskSpec[object] = TaskSpec[object](name, _fn)
    start = datetime(2024, 1, 1, 0, 0, 0)
    # 用 timedelta 精确表达秒数，避免 int() 截断小数
    end = start + timedelta(seconds=duration) if duration else None
    return TaskResult[object](
        spec=spec,
        status=status,
        value=value,
        error=error,
        attempts=attempts,
        started_at=start,
        finished_at=end,
    )


class TestRunReportAccess:
    """测试 RunReport 的访问接口."""

    def test_getitem_returns_value(self) -> None:
        """report[name] 应返回任务结果值."""
        report = px.RunReport()
        report.results["a"] = _make_result("a", value=7)
        assert report["a"] == 7

    def test_result_of_returns_full_result(self) -> None:
        """result_of 应返回完整的 TaskResult 对象."""
        report = px.RunReport()
        r = _make_result("a")
        report.results["a"] = r
        assert report.result_of("a") is r

    def test_contains(self) -> None:
        """in 运算符应正确判断任务是否存在."""
        report = px.RunReport()
        report.results["a"] = _make_result("a")
        assert "a" in report
        assert "b" not in report

    def test_iter_and_len(self) -> None:
        """应支持迭代任务名并返回任务数量."""
        report = px.RunReport()
        report.results["a"] = _make_result("a")
        report.results["b"] = _make_result("b")
        assert list(report) == ["a", "b"]
        assert len(report) == 2


class TestRunReportSummary:
    """测试 RunReport 的 summary 方法."""

    def test_summary_success(self) -> None:
        """应正确汇总成功和跳过的任务."""
        report = px.RunReport()
        report.results["a"] = _make_result("a", status=TaskStatus.SUCCESS, duration=1.0)
        report.results["b"] = _make_result("b", status=TaskStatus.SKIPPED, duration=0.0)
        s = report.summary()
        assert s["success"] is True
        assert s["total_tasks"] == 2
        assert s["by_status"] == {"success": 1, "skipped": 1}
        assert s["total_duration_seconds"] == 1.0

    def test_summary_with_none_duration(self) -> None:
        """未开始/未结束的任务 duration 为 None，不应计入总时长."""
        report = px.RunReport()
        spec: TaskSpec[object] = TaskSpec("a", _fn)  # type: ignore[arg-type]
        report.results["a"] = TaskResult(spec=spec, status=TaskStatus.FAILED)
        s = report.summary()
        assert s["total_duration_seconds"] == 0.0

    def test_failed_tasks(self) -> None:
        """failed_tasks 应返回所有失败任务名."""
        report = px.RunReport()
        report.results["a"] = _make_result("a", status=TaskStatus.SUCCESS)
        report.results["b"] = _make_result(
            "b", status=TaskStatus.FAILED, error=ValueError("x")
        )
        assert report.failed_tasks() == ["b"]


class TestRunReportDescribe:
    """测试 RunReport 的 describe 方法."""

    def test_describe_success(self) -> None:
        """应正确描述成功状态和耗时."""
        report = px.RunReport()
        report.results["a"] = _make_result("a", status=TaskStatus.SUCCESS, duration=0.5)
        desc = report.describe()
        assert "RunReport(success=True)" in desc
        assert "a: success" in desc
        assert "0.500s" in desc

    def test_describe_with_error(self) -> None:
        """应正确描述失败状态和错误信息."""
        report = px.RunReport(success=False)
        report.results["a"] = _make_result(
            "a", status=TaskStatus.FAILED, error=ValueError("boom"), duration=0.1
        )
        desc = report.describe()
        assert "success=False" in desc
        assert "error=ValueError" in desc

    def test_describe_no_duration(self) -> None:
        """无耗时的任务应显示为 '-'."""
        report = px.RunReport()
        spec: TaskSpec[object] = TaskSpec("a", _fn)  # type: ignore[arg-type]
        report.results["a"] = TaskResult(spec=spec, status=TaskStatus.PENDING)
        desc = report.describe()
        assert "-" in desc  # duration 显示为 "-"