zhou
3f966a230e
CI / Lint, Typecheck & Test (push) Successful in 2m5s
将 13 个工具入口文件重构为通过 px.run_yaml 调用 YAML 配置, 辅助函数移至 _ops 模块。新增 run_yaml 便捷函数支持 job 选择 和传递依赖收集,修复 _build_cmd 列表变量展开,新增 bump_project_version 高层函数封装版本号更新+git 提交流程。
138 lines
4.9 KiB
Python
138 lines
4.9 KiB
Python
"""Tests for cli.filedate module."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
import pyflowx as px
|
|
from pyflowx.cli import filedate
|
|
from pyflowx.cli._ops import files
|
|
|
|
|
|
# ---------------------------------------------------------------------- #
|
|
# get_file_timestamp
|
|
# ---------------------------------------------------------------------- #
|
|
class TestGetFileTimestamp:
|
|
"""Test get_file_timestamp function."""
|
|
|
|
def test_get_file_timestamp(self, tmp_path: Path) -> None:
|
|
"""Should get file timestamp."""
|
|
test_file = tmp_path / "test.txt"
|
|
test_file.write_text("test content")
|
|
|
|
timestamp = files.get_file_timestamp(test_file)
|
|
assert len(timestamp) == 8 # YYYYMMDD format
|
|
assert timestamp.isdigit()
|
|
|
|
|
|
# ---------------------------------------------------------------------- #
|
|
# remove_date_prefix
|
|
# ---------------------------------------------------------------------- #
|
|
class TestRemoveDatePrefix:
|
|
"""Test remove_date_prefix function."""
|
|
|
|
def test_remove_date_prefix_with_date(self, tmp_path: Path) -> None:
|
|
"""Should remove date prefix from filename."""
|
|
test_file = tmp_path / "20240101_test.txt"
|
|
test_file.write_text("test content")
|
|
|
|
new_path = files.remove_date_prefix(test_file)
|
|
assert new_path.name == "test.txt"
|
|
|
|
def test_remove_date_prefix_without_date(self, tmp_path: Path) -> None:
|
|
"""Should not change filename without date prefix."""
|
|
test_file = tmp_path / "test.txt"
|
|
test_file.write_text("test content")
|
|
|
|
new_path = files.remove_date_prefix(test_file)
|
|
assert new_path == test_file
|
|
|
|
|
|
# ---------------------------------------------------------------------- #
|
|
# add_date_prefix
|
|
# ---------------------------------------------------------------------- #
|
|
class TestAddDatePrefix:
|
|
"""Test add_date_prefix function."""
|
|
|
|
def test_add_date_prefix(self, tmp_path: Path) -> None:
|
|
"""Should add date prefix to filename."""
|
|
test_file = tmp_path / "test.txt"
|
|
test_file.write_text("test content")
|
|
|
|
new_path = files.add_date_prefix(test_file)
|
|
assert new_path.name.startswith("20") # Starts with year
|
|
assert "_test.txt" in new_path.name
|
|
|
|
|
|
# ---------------------------------------------------------------------- #
|
|
# process_file_date
|
|
# ---------------------------------------------------------------------- #
|
|
class TestProcessFileDate:
|
|
"""Test process_file_date function."""
|
|
|
|
def test_process_file_date_add(self, tmp_path: Path) -> None:
|
|
"""Should add date prefix."""
|
|
test_file = tmp_path / "test.txt"
|
|
test_file.write_text("test content")
|
|
|
|
files.process_file_date(test_file, clear=False)
|
|
# File should be renamed with date prefix
|
|
|
|
def test_process_file_date_clear(self, tmp_path: Path) -> None:
|
|
"""Should clear date prefix."""
|
|
test_file = tmp_path / "20240101_test.txt"
|
|
test_file.write_text("test content")
|
|
|
|
files.process_file_date(test_file, clear=True)
|
|
# File should be renamed without date prefix
|
|
|
|
|
|
# ---------------------------------------------------------------------- #
|
|
# process_files_date
|
|
# ---------------------------------------------------------------------- #
|
|
class TestProcessFilesDate:
|
|
"""Test process_files_date function."""
|
|
|
|
def test_process_files_date_batch(self, tmp_path: Path) -> None:
|
|
"""Should process multiple files."""
|
|
file_list = []
|
|
for i in range(3):
|
|
test_file = tmp_path / f"test{i}.txt"
|
|
test_file.write_text(f"content{i}")
|
|
file_list.append(test_file)
|
|
|
|
files.process_files_date(file_list, clear=False)
|
|
# All files should be processed
|
|
|
|
|
|
# ---------------------------------------------------------------------- #
|
|
# main function
|
|
# ---------------------------------------------------------------------- #
|
|
class TestMain:
|
|
"""Test main function."""
|
|
|
|
def test_main_add_command(self, tmp_path: Path) -> None:
|
|
"""main() should handle add command."""
|
|
test_file = tmp_path / "test.txt"
|
|
test_file.write_text("test content")
|
|
|
|
with patch("sys.argv", ["filedate", "add", str(test_file)]), patch.object(px, "run") as mock_run:
|
|
filedate.main()
|
|
assert mock_run.called
|
|
|
|
def test_main_clear_command(self, tmp_path: Path) -> None:
|
|
"""main() should handle clear command."""
|
|
test_file = tmp_path / "20240101_test.txt"
|
|
test_file.write_text("test content")
|
|
|
|
with patch("sys.argv", ["filedate", "clear", str(test_file)]), patch.object(px, "run") as mock_run:
|
|
filedate.main()
|
|
assert mock_run.called
|
|
|
|
def test_main_with_no_args_shows_help(self) -> None:
|
|
"""main() with no args should show help."""
|
|
with patch("sys.argv", ["filedate"]):
|
|
filedate.main()
|
|
# Should print help and return
|