lanyuanxiaoyao
9daff73589
- 更新 openspec/config.yaml 中 git 任务相关说明 - 将 scripts.core.* 改为 core.*，scripts.readers.* 改为 readers.* - 优化 lyxy_document_reader.py 中 sys.path 设置方式 - 同步更新所有测试文件的导入路径
95 lines
2.7 KiB
Python
95 lines
2.7 KiB
Python
"""CLI 测试专用 fixtures。"""
|
||
|
||
import pytest
|
||
import sys
|
||
from pathlib import Path
|
||
from io import StringIO
|
||
from contextlib import redirect_stdout, redirect_stderr
|
||
|
||
|
||
@pytest.fixture
|
||
def cli_runner():
|
||
"""CLI 运行器 fixture，用于调用 main() 函数并捕获输出。
|
||
|
||
Returns:
|
||
function: 接受 args 列表，返回 (stdout, stderr, exit_code) 元组
|
||
"""
|
||
def _run_cli(args):
|
||
"""运行 CLI 并捕获输出。
|
||
|
||
Args:
|
||
args: 命令行参数列表（不包含程序名）
|
||
|
||
Returns:
|
||
tuple: (stdout, stderr, exit_code)
|
||
"""
|
||
# 将 scripts/ 目录添加到 sys.path
|
||
project_root = Path(__file__).resolve().parent.parent.parent # tests/test_cli/ 的父目录是 tests/，再父目录是项目根目录
|
||
scripts_dir = project_root / "scripts"
|
||
if str(scripts_dir) not in sys.path:
|
||
sys.path.insert(0, str(scripts_dir))
|
||
|
||
from lyxy_document_reader import main
|
||
|
||
# 保存原始 sys.argv 和 sys.exit
|
||
original_argv = sys.argv
|
||
original_exit = sys.exit
|
||
|
||
stdout_capture = StringIO()
|
||
stderr_capture = StringIO()
|
||
exit_code = 0
|
||
|
||
def mock_exit(code=0):
|
||
nonlocal exit_code
|
||
exit_code = code
|
||
raise SystemExit(code)
|
||
|
||
try:
|
||
# 设置命令行参数
|
||
sys.argv = ['lyxy_document_reader'] + args
|
||
sys.exit = mock_exit
|
||
|
||
# 捕获输出
|
||
with redirect_stdout(stdout_capture), redirect_stderr(stderr_capture):
|
||
try:
|
||
main()
|
||
except SystemExit:
|
||
pass
|
||
|
||
finally:
|
||
# 恢复原始状态
|
||
sys.argv = original_argv
|
||
sys.exit = original_exit
|
||
|
||
return stdout_capture.getvalue(), stderr_capture.getvalue(), exit_code
|
||
|
||
return _run_cli
|
||
|
||
|
||
@pytest.fixture
|
||
def temp_test_file(tmp_path, temp_docx, temp_pdf, temp_html, temp_pptx, temp_xlsx):
|
||
"""根据格式类型创建临时测试文件的 fixture 工厂。
|
||
|
||
Args:
|
||
format_type: 文件格式类型 ('docx', 'pdf', 'html', 'pptx', 'xlsx')
|
||
**kwargs: 传递给对应 fixture 的参数
|
||
|
||
Returns:
|
||
str: 临时文件路径
|
||
"""
|
||
def _create_file(format_type, **kwargs):
|
||
if format_type == 'docx':
|
||
return temp_docx(**kwargs)
|
||
elif format_type == 'pdf':
|
||
return temp_pdf(**kwargs)
|
||
elif format_type == 'html':
|
||
return temp_html(**kwargs)
|
||
elif format_type == 'pptx':
|
||
return temp_pptx(**kwargs)
|
||
elif format_type == 'xlsx':
|
||
return temp_xlsx(**kwargs)
|
||
else:
|
||
raise ValueError(f"不支持的格式类型: {format_type}")
|
||
|
||
return _create_file
|