zhou
40f641611b
1. 将CliRunner默认执行策略从sequential改为dependency 2. 新增RunReport的任务状态查询和时长统计方法 3. 实现task装饰器并补充executor参数文档 4. 新增进程池执行器支持CPU密集型任务 5. 新增Graph.chain链式构建和add_subgraph子图合并功能 6. 新增流式任务传递、进程池执行、命名空间等多类测试用例 7. 补充tests目录路径导入配置
176 lines
6.5 KiB
Python
176 lines
6.5 KiB
Python
"""RunReport 测试."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime, timedelta
|
|
from typing import Any
|
|
|
|
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: Any = 42,
|
|
error: BaseException | None = None,
|
|
duration: float = 0.5,
|
|
attempts: int = 1,
|
|
) -> TaskResult[Any]:
|
|
"""构造测试用 TaskResult 实例."""
|
|
spec: TaskSpec[Any] = TaskSpec[Any](name, _fn)
|
|
start = datetime(2024, 1, 1, 0, 0, 0)
|
|
# 用 timedelta 精确表达秒数，避免 int() 截断小数
|
|
end = start + timedelta(seconds=duration) if duration else None
|
|
return TaskResult[Any](
|
|
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[Any] = TaskSpec[Any]("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[Any] = TaskSpec[Any]("a", _fn) # type: ignore[arg-type]
|
|
report.results["a"] = TaskResult[Any](spec=spec, status=TaskStatus.PENDING)
|
|
desc = report.describe()
|
|
assert "-" in desc # duration 显示为 "-"
|
|
|
|
|
|
class TestRunReportQueries:
|
|
"""测试 RunReport 的新查询 API."""
|
|
|
|
def test_succeeded_tasks(self) -> None:
|
|
"""succeeded_tasks 返回 SUCCESS 状态的任务名."""
|
|
report = px.RunReport()
|
|
report.results["a"] = _make_result("a", status=TaskStatus.SUCCESS)
|
|
report.results["b"] = _make_result("b", status=TaskStatus.FAILED)
|
|
report.results["c"] = _make_result("c", status=TaskStatus.SUCCESS)
|
|
assert report.succeeded_tasks() == ["a", "c"]
|
|
|
|
def test_skipped_tasks(self) -> None:
|
|
"""skipped_tasks 返回 SKIPPED 状态的任务名."""
|
|
report = px.RunReport()
|
|
report.results["a"] = _make_result("a", status=TaskStatus.SKIPPED)
|
|
report.results["b"] = _make_result("b", status=TaskStatus.SUCCESS)
|
|
assert report.skipped_tasks() == ["a"]
|
|
|
|
def test_tasks_by_status(self) -> None:
|
|
"""tasks_by_status 按指定状态过滤."""
|
|
report = px.RunReport()
|
|
report.results["a"] = _make_result("a", status=TaskStatus.FAILED)
|
|
report.results["b"] = _make_result("b", status=TaskStatus.FAILED)
|
|
report.results["c"] = _make_result("c", status=TaskStatus.SUCCESS)
|
|
assert report.tasks_by_status(TaskStatus.FAILED) == ["a", "b"]
|
|
assert report.tasks_by_status(TaskStatus.SUCCESS) == ["c"]
|
|
assert report.tasks_by_status(TaskStatus.SKIPPED) == []
|
|
|
|
def test_durations(self) -> None:
|
|
"""durations 返回任务名 -> 时长映射."""
|
|
report = px.RunReport()
|
|
report.results["a"] = _make_result("a", duration=1.5)
|
|
report.results["b"] = _make_result("b", duration=2.0)
|
|
durs = report.durations()
|
|
assert durs["a"] == 1.5
|
|
assert durs["b"] == 2.0
|
|
|
|
def test_durations_no_duration(self) -> None:
|
|
"""无时长的任务应返回 0.0."""
|
|
report = px.RunReport()
|
|
spec: TaskSpec[Any] = TaskSpec[Any]("a", _fn) # type: ignore[arg-type]
|
|
report.results["a"] = TaskResult[Any](spec=spec, status=TaskStatus.PENDING)
|
|
durs = report.durations()
|
|
assert durs["a"] == 0.0
|
|
|