AI原生开发:ClaudeCode与国产大模型协同实战 第7章:AI驱动的契约式TDD——做团队最严格的守门人

# voucher_spec.yaml (精简版)contracts:  RedeemVoucherAPI:    endpoint: /voucher/redeem    method: POST    request:      voucher_id: { type: string, required: true }      order_id: { type: string, required: true }      order_amount: { type: float, required: true, minimum: 0.01 }    response:      success:        code: 200        data: { deducted_amount: float }      errors:        - code: 40401          message: "Voucher not found"        - code: 40402          message: "Voucher already redeemed"        - code: 40403          message: "Order amount does not meet minimum requirement"

# Role你是一个极其严苛的测试工程师，信奉契约精神。# Task请根据提供的 OpenSpec 规约，使用 Python + Pytest 生成单元测试代码。# OpenSpec 规约（粘贴上面的 voucher_spec.yaml）# Directive1. 必须覆盖所有 contracts.errors 中的异常码（40401, 40402, 40403）。2. 必须覆盖成功的正常路径（200）。3. 业务逻辑类 VoucherService 暂时不存在，请使用 unittest.mock 的 MagicMock 来模拟依赖（如数据库查询）。4. 测试类名和方法名必须能清晰表达意图，不能使用 test_1, test_2 这种无意义命名。请直接输出 Pytest 测试代码。

import pytestfrom unittest.mock import MagicMock# 模拟的业务异常类（根据OpenSpec生成）class VoucherError(Exception):    def __init__(self, code, message):        self.code = code        self.message = message# 模拟的业务服务类（待后续实现）class VoucherService:    def redeem(self, voucher_id: str, order_id: str, order_amount: float) -> dict:        pass # 待实现@pytest.fixturedef mock_db():    return MagicMock()@pytest.fixturedef service(mock_db):    return VoucherService(db=mock_db)class TestVoucherRedeem:    """代金券核销契约测试"""    def test_redeem_success(self, service, mock_db):        """正常核销成功"""        mock_voucher = MagicMock(id="V1", is_redeemed=False, min_requirement=50.0, face_value=10.0)        mock_db.get_voucher.return_value = mock_voucher        result = service.redeem("V1", "O1", 100.0)        assert result["code"] == 200        assert result["data"]["deducted_amount"] == 10.0    def test_redeem_voucher_not_found(self, service, mock_db):        """异常40401: 代金券不存在"""        mock_db.get_voucher.return_value = None        with pytest.raises(VoucherError) as exc_info:            service.redeem("V_NONEXIST", "O1", 100.0)        assert exc_info.value.code == 40401    def test_redeem_already_redeemed(self, service, mock_db):        """异常40402: 代金券已核销"""        mock_voucher = MagicMock(id="V1", is_redeemed=True, min_requirement=50.0, face_value=10.0)        mock_db.get_voucher.return_value = mock_voucher        with pytest.raises(VoucherError) as exc_info:            service.redeem("V1", "O1", 100.0)        assert exc_info.value.code == 40402    def test_redeem_order_amount_insufficient(self, service, mock_db):        """异常40403: 订单金额不满足最低要求"""        mock_voucher = MagicMock(id="V1", is_redeemed=False, min_requirement=50.0, face_value=10.0)        mock_db.get_voucher.return_value = mock_voucher        with pytest.raises(VoucherError) as exc_info:            service.redeem("V1", "O1", 30.0) # 30.0 < 50.0        assert exc_info.value.code == 40403

import pytestfrom unittest.mock import MagicMockfrom connection_timeout import ConnectionTimeoutError # 假设的DB异常# ... (复用前面的 VoucherService 和 VoucherError 定义)class TestVoucherRedeemEvil:    """代金券核销：腹黑边界与故障注入测试"""    def test_redeem_sql_injection_attempt(self, service, mock_db):        """防SQL注入：voucher_id包含恶意片段"""        mock_db.get_voucher.return_value = None        # 传入恶意ID，不应抛出未捕获异常，应视为找不到        with pytest.raises(VoucherError) as exc_info:            service.redeem("V1'; DROP TABLE vouchers;", "O1", 100.0)        assert exc_info.value.code == 40401    def test_redeem_negative_order_amount(self, service, mock_db):        """参数防御：订单金额为负数"""        with pytest.raises(VoucherError) as exc_info:            service.redeem("V1", "O1", -10.0)        # 假设我们在Service层加了参数校验，返回400        assert exc_info.value.code == 400     def test_redeem_db_connection_timeout(self, service, mock_db):        """依赖容灾：数据库查询超时"""        # 模拟数据库突然抽风        mock_db.get_voucher.side_effect = ConnectionTimeoutError("DB timed out")        with pytest.raises(VoucherError) as exc_info:            service.redeem("V1", "O1", 100.0)        # 业务层必须兜底，不能把底层数据库超时暴露给前端        assert exc_info.value.code == 50001