153 lines
5.0 KiB
Python
153 lines
5.0 KiB
Python
import pytest
|
|
from typer.testing import CliRunner
|
|
from unittest.mock import Mock, patch
|
|
import sys
|
|
from pathlib import Path
|
|
from loguru import logger
|
|
|
|
# 测试 CLI 命令
|
|
runner = CliRunner()
|
|
|
|
|
|
def test_cli_init_success():
|
|
"""测试 init 命令成功执行"""
|
|
from src.llm_codegen.cli import app # 假设从项目根目录运行测试
|
|
|
|
# 模拟 CodeGenerator 和其方法，避免实际调用 API
|
|
with patch('src.llm_codegen.cli.CodeGenerator') as mock_generator:
|
|
mock_instance = Mock()
|
|
mock_instance.run = Mock()
|
|
mock_generator.return_value = mock_instance
|
|
|
|
# 创建一个虚拟的 README 文件用于测试
|
|
test_readme = Path("test_readme.md")
|
|
test_readme.write_text("# Test Project\n\nA test project for CLI.")
|
|
|
|
result = runner.invoke(app, ["init", str(test_readme), "--output", "./test_output"])
|
|
|
|
# 清理
|
|
test_readme.unlink()
|
|
|
|
assert result.exit_code == 0
|
|
assert "初始化失败" not in result.stdout
|
|
mock_generator.assert_called_once()
|
|
mock_instance.run.assert_called_once_with(test_readme)
|
|
|
|
|
|
def test_cli_init_failure_no_readme():
|
|
"""测试 init 命令当 README 不存在时失败"""
|
|
from src.llm_codegen.cli import app
|
|
|
|
result = runner.invoke(app, ["init", "nonexistent.md"])
|
|
|
|
assert result.exit_code != 0 # 应该退出码非零
|
|
|
|
|
|
def test_cli_enhance_success():
|
|
"""测试 enhance 命令成功执行（简化版，基于工单）"""
|
|
from src.llm_codegen.cli import app
|
|
|
|
# 模拟依赖文件和环境
|
|
with patch('src.llm_codegen.cli.CodeGenerator') as mock_generator, \
|
|
patch('src.llm_codegen.cli.Checker') as mock_checker, \
|
|
patch('pathlib.Path.exists') as mock_exists:
|
|
|
|
mock_exists.return_value = True # 模拟 design.json 存在
|
|
mock_instance = Mock()
|
|
mock_instance.run_full_check_and_fix = Mock(return_value=True)
|
|
mock_checker.return_value = mock_instance
|
|
mock_generator.return_value = Mock()
|
|
|
|
# 创建一个虚拟的工单文件
|
|
test_issue = Path("test_feature.issue")
|
|
test_issue.write_text("name: Add feature\ndescription: Test feature")
|
|
|
|
result = runner.invoke(app, ["enhance", str(test_issue), "--output", "./test_output"])
|
|
|
|
# 清理
|
|
test_issue.unlink()
|
|
|
|
assert result.exit_code == 0
|
|
assert "增强失败" not in result.stdout
|
|
mock_checker.assert_called_once()
|
|
mock_instance.run_full_check_and_fix.assert_called_once()
|
|
|
|
|
|
def test_cli_fix_success():
|
|
"""测试 fix 命令成功执行（简化版，基于工单）"""
|
|
from src.llm_codegen.cli import app
|
|
|
|
with patch('src.llm_codegen.cli.CodeGenerator') as mock_generator, \
|
|
patch('src.llm_codegen.cli.Checker') as mock_checker, \
|
|
patch('pathlib.Path.exists') as mock_exists:
|
|
|
|
mock_exists.return_value = True
|
|
mock_instance = Mock()
|
|
mock_instance.run_full_check_and_fix = Mock(return_value=True)
|
|
mock_checker.return_value = mock_instance
|
|
mock_generator.return_value = Mock()
|
|
|
|
test_issue = Path("test_bug.issue")
|
|
test_issue.write_text("name: Fix bug\ndescription: Test bug")
|
|
|
|
result = runner.invoke(app, ["fix", str(test_issue), "--output", "./test_output"])
|
|
|
|
test_issue.unlink()
|
|
|
|
assert result.exit_code == 0
|
|
assert "修复失败" not in result.stdout
|
|
mock_checker.assert_called_once()
|
|
mock_instance.run_full_check_and_fix.assert_called_once()
|
|
|
|
|
|
def test_cli_help():
|
|
"""测试 CLI 帮助命令"""
|
|
from src.llm_codegen.cli import app
|
|
|
|
result = runner.invoke(app, ["--help"])
|
|
assert result.exit_code == 0
|
|
assert "基于LLM的自动化代码生成与维护工具" in result.stdout
|
|
|
|
# 测试子命令帮助
|
|
result = runner.invoke(app, ["init", "--help"])
|
|
assert result.exit_code == 0
|
|
assert "README.md 文件路径" in result.stdout
|
|
|
|
|
|
def test_cli_enhance_no_design():
|
|
"""测试 enhance 命令当 design.json 不存在时失败"""
|
|
from src.llm_codegen.cli import app
|
|
|
|
with patch('pathlib.Path.exists') as mock_exists:
|
|
mock_exists.return_value = False # 模拟 design.json 不存在
|
|
|
|
test_issue = Path("test_feature.issue")
|
|
test_issue.write_text("name: Test")
|
|
|
|
result = runner.invoke(app, ["enhance", str(test_issue)])
|
|
|
|
test_issue.unlink()
|
|
|
|
assert result.exit_code != 0
|
|
|
|
|
|
def test_cli_fix_no_design():
|
|
"""测试 fix 命令当 design.json 不存在时失败"""
|
|
from src.llm_codegen.cli import app
|
|
|
|
with patch('pathlib.Path.exists') as mock_exists:
|
|
mock_exists.return_value = False
|
|
|
|
test_issue = Path("test_bug.issue")
|
|
test_issue.write_text("name: Test")
|
|
|
|
result = runner.invoke(app, ["fix", str(test_issue)])
|
|
|
|
test_issue.unlink()
|
|
|
|
assert result.exit_code != 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
pytest.main([__file__])
|