zhou
413ab40044
1. 在pyproject.toml中为测试文件添加ARG001和ARG002规则忽略 2. 重构多个CLI测试文件，移除冗余的mock断言、导入顺序调整 3. 统一测试用例的帮助信息输出逻辑，移除SystemExit捕获，简化测试流程 4. 拆分合并冗余的测试类，按功能细化测试用例 5. 移除测试代码中多余的注释和pytest导入
111 lines
4.3 KiB
Python
111 lines
4.3 KiB
Python
"""Tests for cli.envpy module."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
import pyflowx as px
|
|
from pyflowx.cli import envpy
|
|
|
|
|
|
# ---------------------------------------------------------------------- #
|
|
# set_pip_mirror
|
|
# ---------------------------------------------------------------------- #
|
|
class TestSetPipMirror:
|
|
"""Test set_pip_mirror function."""
|
|
|
|
def test_set_pip_mirror_tsinghua(self, tmp_path: Path) -> None:
|
|
"""Should set tsinghua mirror."""
|
|
with patch.object(Path, "home", return_value=tmp_path):
|
|
envpy.set_pip_mirror("tsinghua")
|
|
# Check pip config
|
|
pip_config = tmp_path / "pip" / "pip.ini"
|
|
if envpy.Constants.IS_WINDOWS:
|
|
assert pip_config.exists() or (tmp_path / "pip" / "pip.conf").exists()
|
|
|
|
def test_set_pip_mirror_aliyun(self, tmp_path: Path) -> None:
|
|
"""Should set aliyun mirror."""
|
|
with patch.object(Path, "home", return_value=tmp_path):
|
|
envpy.set_pip_mirror("aliyun")
|
|
# Check pip config
|
|
pip_dir = tmp_path / "pip"
|
|
assert pip_dir.exists()
|
|
|
|
def test_set_pip_mirror_with_token(self, tmp_path: Path) -> None:
|
|
"""Should set mirror with token."""
|
|
with patch.object(Path, "home", return_value=tmp_path):
|
|
envpy.set_pip_mirror("tsinghua", token="test_token")
|
|
# Check that token is set
|
|
|
|
def test_set_pip_mirror_creates_pip_dir(self, tmp_path: Path) -> None:
|
|
"""Should create pip directory if it doesn't exist."""
|
|
pip_dir = tmp_path / "pip"
|
|
with patch.object(Path, "home", return_value=tmp_path):
|
|
envpy.set_pip_mirror("tsinghua")
|
|
assert pip_dir.exists()
|
|
assert pip_dir.is_dir()
|
|
|
|
|
|
# ---------------------------------------------------------------------- #
|
|
# main function
|
|
# ---------------------------------------------------------------------- #
|
|
class TestMain:
|
|
"""Test main function."""
|
|
|
|
def test_main_mirror_tsinghua(self) -> None:
|
|
"""main() should handle mirror tsinghua command."""
|
|
with patch("sys.argv", ["envpy", "mirror", "tsinghua"]), patch.object(px, "run") as mock_run, patch.object(
|
|
envpy, "set_pip_mirror"
|
|
):
|
|
envpy.main()
|
|
assert mock_run.called
|
|
|
|
def test_main_mirror_aliyun(self) -> None:
|
|
"""main() should handle mirror aliyun command."""
|
|
with patch("sys.argv", ["envpy", "mirror", "aliyun"]), patch.object(px, "run") as mock_run, patch.object(
|
|
envpy, "set_pip_mirror"
|
|
):
|
|
envpy.main()
|
|
assert mock_run.called
|
|
|
|
def test_main_mirror_with_token(self) -> None:
|
|
"""main() should handle mirror with token."""
|
|
with patch("sys.argv", ["envpy", "mirror", "tsinghua", "--token", "test_token"]), patch.object(
|
|
px, "run"
|
|
) as mock_run, patch.object(envpy, "set_pip_mirror"):
|
|
envpy.main()
|
|
assert mock_run.called
|
|
|
|
def test_main_with_no_args_shows_help(self) -> None:
|
|
"""main() with no args should show help and return."""
|
|
with patch("sys.argv", ["envpy"]):
|
|
envpy.main()
|
|
# Should print help and return
|
|
|
|
def test_main_invalid_mirror_shows_error(self) -> None:
|
|
"""main() with invalid mirror should show error."""
|
|
with patch("sys.argv", ["envpy", "mirror", "invalid"]), pytest.raises(SystemExit) as exc_info:
|
|
envpy.main()
|
|
assert exc_info.value.code == 2
|
|
|
|
def test_main_creates_task_spec_with_correct_name(self) -> None:
|
|
"""main() should create TaskSpec with correct name."""
|
|
with patch("sys.argv", ["envpy", "mirror", "tsinghua"]), patch.object(px, "run") as mock_run, patch.object(
|
|
envpy, "set_pip_mirror"
|
|
):
|
|
envpy.main()
|
|
graph = mock_run.call_args[0][0]
|
|
task_names = list(graph.all_specs().keys())
|
|
assert "set_pip_mirror" in task_names
|
|
|
|
def test_main_uses_thread_strategy(self) -> None:
|
|
"""main() should use thread strategy."""
|
|
with patch("sys.argv", ["envpy", "mirror", "tsinghua"]), patch.object(px, "run") as mock_run, patch.object(
|
|
envpy, "set_pip_mirror"
|
|
):
|
|
envpy.main()
|
|
assert mock_run.call_args[1]["strategy"] == "thread"
|