zhou
43e1aad1fe
本次提交更新了版本号至0.2.13，同时完成多项改进： 1. 在.gitignore中新增忽略性能分析文件*_profile.html 2. 修复测试用例中echo命令在Windows下无法被正确检测的问题，改用python命令 3. 优化测试用例确保性能统计数据有效，添加耗时模拟函数 4. 为所有CLI任务统一配置项目根目录作为工作目录，解决跨平台执行路径问题 5. 新增测试验证所有任务的cwd配置正确性
176 lines
6.7 KiB
Python
176 lines
6.7 KiB
Python
"""Tests for cli.pymake module."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from pyflowx.cli import pymake
|
|
|
|
|
|
# ---------------------------------------------------------------------- #
|
|
# TaskSpec definitions
|
|
# ---------------------------------------------------------------------- #
|
|
def _find_task(name: str) -> pymake.px.TaskSpec:
|
|
"""从 pymake.tasks 或 aliases 中查找指定名称的 TaskSpec."""
|
|
for spec in pymake.tasks:
|
|
if spec.name == name:
|
|
return spec
|
|
# 单任务别名（doc/lint/tox）内联在 aliases dict 中
|
|
value = pymake.aliases.get(name)
|
|
if isinstance(value, pymake.px.TaskSpec):
|
|
return value
|
|
raise KeyError(f"任务 {name!r} 未找到")
|
|
|
|
|
|
class TestTaskSpecDefinitions:
|
|
"""Test that all TaskSpec definitions are valid."""
|
|
|
|
def test_uv_build_spec(self) -> None:
|
|
"""uv_build spec should be properly defined."""
|
|
spec = _find_task("uv_build")
|
|
assert spec.name == "uv_build"
|
|
assert spec.cmd == ["uv", "build"]
|
|
assert spec.skip_if_missing is False
|
|
|
|
def test_maturin_build_spec(self) -> None:
|
|
"""maturin_build spec should be properly defined."""
|
|
spec = _find_task("maturin_build")
|
|
assert spec.name == "maturin_build"
|
|
assert isinstance(spec.cmd, list)
|
|
assert spec.skip_if_missing is False
|
|
|
|
def test_uv_sync_spec(self) -> None:
|
|
"""uv_sync spec should be properly defined."""
|
|
spec = _find_task("uv_sync")
|
|
assert spec.name == "uv_sync"
|
|
assert spec.cmd == ["uv", "sync"]
|
|
assert spec.skip_if_missing is False
|
|
|
|
def test_git_clean_spec(self) -> None:
|
|
"""git_clean spec should be properly defined."""
|
|
spec = _find_task("git_clean")
|
|
assert spec.name == "git_clean"
|
|
assert spec.cmd == ["gitt", "c"]
|
|
assert spec.skip_if_missing is False
|
|
|
|
def test_test_spec(self) -> None:
|
|
"""test spec should be properly defined."""
|
|
spec = _find_task("test")
|
|
assert spec.name == "test"
|
|
assert isinstance(spec.cmd, list)
|
|
assert "pytest" in spec.cmd
|
|
assert "-m" in spec.cmd
|
|
assert "not slow" in spec.cmd
|
|
assert spec.skip_if_missing is False
|
|
|
|
def test_test_fast_spec(self) -> None:
|
|
"""test_fast spec should be properly defined."""
|
|
spec = _find_task("test_fast")
|
|
assert spec.name == "test_fast"
|
|
assert isinstance(spec.cmd, list)
|
|
assert "pytest" in spec.cmd
|
|
assert "-n" not in spec.cmd # test_fast doesn't use parallel
|
|
assert spec.skip_if_missing is False
|
|
|
|
def test_test_coverage_spec(self) -> None:
|
|
"""test_coverage spec should be properly defined."""
|
|
spec = _find_task("test_coverage")
|
|
assert spec.name == "test_coverage"
|
|
assert isinstance(spec.cmd, list)
|
|
assert "pytest" in spec.cmd
|
|
assert "--cov" in spec.cmd
|
|
assert spec.skip_if_missing is False
|
|
|
|
def test_ruff_lint_spec(self) -> None:
|
|
"""lint spec should be properly defined."""
|
|
spec = _find_task("lint")
|
|
assert spec.name == "lint"
|
|
assert isinstance(spec.cmd, list)
|
|
assert "ruff" in spec.cmd
|
|
assert "check" in spec.cmd
|
|
assert spec.skip_if_missing is False
|
|
|
|
def test_doc_spec(self) -> None:
|
|
"""doc spec should be properly defined."""
|
|
spec = _find_task("doc")
|
|
assert spec.name == "doc"
|
|
assert isinstance(spec.cmd, list)
|
|
assert "sphinx-build" in spec.cmd
|
|
assert spec.skip_if_missing is False
|
|
|
|
def test_hatch_publish_spec(self) -> None:
|
|
"""publish_python spec should be properly defined."""
|
|
spec = _find_task("publish_python")
|
|
assert spec.name == "publish_python"
|
|
assert spec.cmd == ["hatch", "publish"]
|
|
assert spec.skip_if_missing is False
|
|
|
|
def test_twine_publish_spec(self) -> None:
|
|
"""twine_publish spec should be properly defined."""
|
|
spec = _find_task("twine_publish")
|
|
assert spec.name == "twine_publish"
|
|
assert isinstance(spec.cmd, list)
|
|
assert "twine" in spec.cmd
|
|
assert "upload" in spec.cmd
|
|
assert spec.skip_if_missing is False
|
|
|
|
def test_tox_spec(self) -> None:
|
|
"""tox spec should be properly defined."""
|
|
spec = _find_task("tox")
|
|
assert spec.name == "tox"
|
|
assert spec.cmd == ["tox", "-p", "auto"]
|
|
assert spec.skip_if_missing is False
|
|
|
|
def test_all_tasks_have_correct_cwd(self) -> None:
|
|
"""所有任务应该有正确的 cwd 设置（指向项目根目录）."""
|
|
# 验证 ROOT_DIR 定义正确（向上三层到达项目根目录）
|
|
expected_root = Path(__file__).parent.parent.parent
|
|
assert expected_root == pymake.ROOT_DIR
|
|
|
|
# 验证 tasks 中的所有命令任务都有正确的 cwd
|
|
for spec in pymake.tasks:
|
|
if spec.cmd is not None:
|
|
assert spec.cwd == pymake.ROOT_DIR, f"任务 {spec.name} 的 cwd 应为 {pymake.ROOT_DIR}"
|
|
|
|
# 验证 aliases 中的内联任务（doc/lint/tox）也有正确的 cwd
|
|
for name in ("doc", "lint", "tox"):
|
|
spec = _find_task(name)
|
|
assert spec.cwd == pymake.ROOT_DIR, f"任务 {name} 的 cwd 应为 {pymake.ROOT_DIR}"
|
|
|
|
|
|
# ---------------------------------------------------------------------- #
|
|
# main function
|
|
# ---------------------------------------------------------------------- #
|
|
class TestMain:
|
|
"""Test main function."""
|
|
|
|
def test_main_calls_run_cli(self) -> None:
|
|
"""main() should create a CliRunner and call run_cli()."""
|
|
with pytest.raises(SystemExit) as exc_info:
|
|
pymake.main()
|
|
# run_cli() calls sys.exit(), so we should get SystemExit
|
|
# The exit code depends on whether any commands are available
|
|
assert exc_info.value.code in (0, 1, 2)
|
|
|
|
def test_main_with_list_argument(self) -> None:
|
|
"""main() should handle --list argument."""
|
|
with patch("sys.argv", ["pymake", "--list"]), pytest.raises(SystemExit) as exc_info:
|
|
pymake.main()
|
|
assert exc_info.value.code == 0
|
|
|
|
def test_main_creates_runner_with_multiple_commands(self) -> None:
|
|
"""main() should create a CliRunner with multiple commands."""
|
|
# We can't easily test the runner creation without mocking,
|
|
# but we can verify that main() doesn't raise an error for --list
|
|
with patch("sys.argv", ["pymake", "--list"]), pytest.raises(SystemExit):
|
|
pymake.main()
|
|
|
|
def test_main_with_no_args_shows_help(self) -> None:
|
|
"""main() with no args should show help and exit with failure."""
|
|
with patch("sys.argv", ["pymake"]), pytest.raises(SystemExit) as exc_info:
|
|
pymake.main()
|
|
assert exc_info.value.code == 1
|