108bf87225
补齐源快照、资料预演、正式导入及工资核算的依赖链路。 新增版本化输入、逐段 FIFO、生成锁和结构化当月调整， 并提供默认关闭的周期采集、有限重试及 Windows 托管。 原始客户资料、私有配置及首页改动不纳入本次提交。 BREAKING CHANGE: 升级前需应用新增迁移，并通过 PAYROLL_MAKER_USER_ID 绑定实际工资制作人，否则业务只读。 旧单批次核算默认关闭；历史兼容需显式启用 PAYROLL_ALLOW_LEGACY_CALCULATION。生产启用和真实金额待验收。
257 lines
10 KiB
Python
257 lines
10 KiB
Python
"""单据编号唯一键冲突重试的行为锁定（audit 上下文内循环重试）。
|
||
|
||
守：1062 兜底重试若以「吞异常后 return 重入整个 create_*」实现，会产生
|
||
三重缺陷 —— 外层审计把已回滚的首次尝试记成 success=True（假成功，与
|
||
audit_service.py「上下文内不得吞掉异常后正常退出」的约定相反）、重试再
|
||
开一层审计（违反「不支持嵌套」）、持续冲突时递归至 RecursionError。
|
||
修复口径：审计上下文内 for 循环重试（上限 _NO_RETRY），成功/最终失败
|
||
各恰好一条操作日志（以 POST 前的日志条数为基线断言增量）。
|
||
|
||
制造冲突的手段：monkeypatch ``business_service._generate_no`` 首次返回
|
||
库内已存在编号 → flush 撞唯一键 → 走重试分支。审计按 username 冗余溯
|
||
源（账号删除后 user_id 被 SET NULL），断言以基线增量免受既有数据干扰。
|
||
"""
|
||
|
||
import uuid
|
||
from datetime import date
|
||
|
||
import pytest
|
||
from fastapi.testclient import TestClient
|
||
from sqlalchemy import func, select
|
||
|
||
from app.core.constants import ACTION_CREATE, MODULE_BUSINESS, MODULE_RECEIPT
|
||
from app.core.db import get_session_factory
|
||
from app.main import app
|
||
from app.models import OperationLog
|
||
from app.services import business_service, receipt_service
|
||
from tests.integration.factories import create_business
|
||
|
||
pytestmark = pytest.mark.integration
|
||
|
||
_BUSINESSES = "/api/v1/business/businesses"
|
||
_CUSTOMERS = "/api/v1/base/customers"
|
||
_RECEIPTS = "/api/v1/business/receipts"
|
||
|
||
|
||
@pytest.fixture
|
||
def quiet_client():
|
||
"""持续冲突用例依赖全局异常处理器产出 500 信封。
|
||
|
||
TestClient 默认对 ServerErrorMiddleware 的响应直接 re-raise
|
||
（raise_server_exceptions=True），会掩盖「原子处理器已正确收敛 500」
|
||
的行为本身 —— 审计断言反而无从谈起。
|
||
"""
|
||
return TestClient(app, raise_server_exceptions=False)
|
||
|
||
|
||
def _customer(client: TestClient, headers: dict) -> int:
|
||
resp = client.post(
|
||
_CUSTOMERS,
|
||
json={
|
||
"company_name": f"t_retry公司_{uuid.uuid4().hex[:8]}",
|
||
"owner_name": "张老板",
|
||
"customer_type": "personal",
|
||
},
|
||
headers=headers,
|
||
)
|
||
assert resp.status_code == 200, resp.text
|
||
return int(resp.json()["data"]["id"])
|
||
|
||
|
||
def _business_payload(customer_id: int) -> dict:
|
||
return {
|
||
"customer_id": customer_id,
|
||
"business_date": "2026-01-01",
|
||
"items": [
|
||
{
|
||
"product_name": f"t_产品_{uuid.uuid4().hex[:8]}",
|
||
"product_category": "municipal_water",
|
||
"quantity": "1.00",
|
||
"sale_price": "10.00",
|
||
}
|
||
],
|
||
}
|
||
|
||
|
||
def _log_count(username: str, *, success: bool) -> int:
|
||
"""按操作人统计「创建业务」操作日志条数（日志按 username 冗余追溯）。"""
|
||
factory = get_session_factory()
|
||
with factory() as session:
|
||
stmt = (
|
||
select(func.count())
|
||
.select_from(OperationLog)
|
||
.where(
|
||
OperationLog.username == username,
|
||
OperationLog.module == MODULE_BUSINESS,
|
||
OperationLog.action == ACTION_CREATE,
|
||
OperationLog.summary.like("创建业务%"),
|
||
OperationLog.success == int(success),
|
||
)
|
||
)
|
||
return int(session.execute(stmt).scalar_one())
|
||
|
||
|
||
def _baseline(writer: dict) -> tuple[int, int]:
|
||
"""POST 前的「创建业务」成功/失败日志条数基线（免受既有数据干扰）。"""
|
||
return (
|
||
_log_count(writer["username"], success=True),
|
||
_log_count(writer["username"], success=False),
|
||
)
|
||
|
||
|
||
class TestAuditRetry:
|
||
"""重试成功与持续失败两条路径的审计记录数。"""
|
||
|
||
def test_首次撞号重试成功_审计恰一条真成功(
|
||
self, quiet_client: TestClient, writer: dict, monkeypatch
|
||
) -> None:
|
||
client = quiet_client
|
||
customer_id = _customer(client, writer["headers"])
|
||
first = client.post(
|
||
_BUSINESSES, json=_business_payload(customer_id), headers=writer["headers"]
|
||
)
|
||
assert first.status_code == 200, first.text
|
||
existing_no = first.json()["data"]["business_no"]
|
||
|
||
real_generate = business_service._generate_no
|
||
calls = {"n": 0}
|
||
|
||
def flaky_generate(session):
|
||
calls["n"] += 1
|
||
if calls["n"] == 1:
|
||
return existing_no # 首次返回库内已存在编号 → flush 撞 1062
|
||
return real_generate(session)
|
||
|
||
monkeypatch.setattr(business_service, "_generate_no", flaky_generate)
|
||
base_ok, base_fail = _baseline(writer)
|
||
|
||
second = client.post(
|
||
_BUSINESSES, json=_business_payload(customer_id), headers=writer["headers"]
|
||
)
|
||
assert second.status_code == 200, second.text
|
||
assert calls["n"] == 2, "重试应恰好再取一次号"
|
||
# 修复前的症状：外层审计正常退出记 success=True（假成功），重试成功
|
||
# 又记一条 —— 成功日志增量为 2；修复后成功恰 +1、失败 +0
|
||
assert _log_count(writer["username"], success=True) == base_ok + 1
|
||
assert _log_count(writer["username"], success=False) == base_fail
|
||
|
||
def test_持续冲突_最终失败且审计恰一条失败(
|
||
self, quiet_client: TestClient, writer: dict, monkeypatch
|
||
) -> None:
|
||
client = quiet_client
|
||
customer_id = _customer(client, writer["headers"])
|
||
first = client.post(
|
||
_BUSINESSES, json=_business_payload(customer_id), headers=writer["headers"]
|
||
)
|
||
assert first.status_code == 200, first.text
|
||
existing_no = first.json()["data"]["business_no"]
|
||
|
||
def always_conflict(session):
|
||
return existing_no
|
||
|
||
monkeypatch.setattr(business_service, "_generate_no", always_conflict)
|
||
base_ok, base_fail = _baseline(writer)
|
||
|
||
second = client.post(
|
||
_BUSINESSES, json=_business_payload(customer_id), headers=writer["headers"]
|
||
)
|
||
# 未捕获 IntegrityError 由全局 handler 收敛 400（数据完整性冲突）——
|
||
# 完整性冲突对用户是「操作被拒绝」而非「服务端异常」，400 更贴近事实
|
||
assert second.status_code == 400, second.text
|
||
assert "完整性冲突" in second.json()["message"]
|
||
# 修复前的症状:return _retry_create 无界递归直至 RecursionError,
|
||
# 每层各写一条记录；修复后重试有上限,最终失败仅一条失败日志
|
||
assert _log_count(writer["username"], success=False) == base_fail + 1
|
||
assert _log_count(writer["username"], success=True) == base_ok
|
||
|
||
|
||
def _receipt_log_count(username: str, *, success: bool) -> int:
|
||
"""按操作人统计「新增回款」操作日志条数。"""
|
||
factory = get_session_factory()
|
||
with factory() as session:
|
||
stmt = (
|
||
select(func.count())
|
||
.select_from(OperationLog)
|
||
.where(
|
||
OperationLog.username == username,
|
||
OperationLog.module == MODULE_RECEIPT,
|
||
OperationLog.action == ACTION_CREATE,
|
||
OperationLog.summary.like("新增回款%"),
|
||
OperationLog.success == int(success),
|
||
)
|
||
)
|
||
return int(session.execute(stmt).scalar_one())
|
||
|
||
|
||
def _receipt_payload(business_id: int) -> dict:
|
||
return {
|
||
"business_id": business_id,
|
||
"receipt_date": str(date.today()),
|
||
"amount": "1200.00",
|
||
}
|
||
|
||
|
||
class TestReceiptAuditRetry:
|
||
"""回款新增的 1062 重试路径（回款与业务同构，但走独立实现）。
|
||
|
||
回款侧的重试循环本次从 create 函数体内移入 helper 调用，重试边界、
|
||
审计条数、批次号重取都必须保持原样 —— 用例会盯住这三条。
|
||
"""
|
||
|
||
def test_首次撞号重试成功_审计恰一条真成功(
|
||
self, quiet_client: TestClient, writer: dict, monkeypatch
|
||
) -> None:
|
||
client = quiet_client
|
||
business = create_business(client, writer["headers"])
|
||
first = client.post(
|
||
_RECEIPTS, json=_receipt_payload(business["id"]), headers=writer["headers"]
|
||
)
|
||
assert first.status_code == 200, first.text
|
||
existing_no = first.json()["data"]["receipt_batch_no"]
|
||
|
||
real_generate = receipt_service._generate_no
|
||
calls = {"n": 0}
|
||
|
||
def flaky_generate(session):
|
||
calls["n"] += 1
|
||
if calls["n"] == 1:
|
||
return existing_no # 首次返回库内已存在批次号 → flush 撞 1062
|
||
return real_generate(session)
|
||
|
||
monkeypatch.setattr(receipt_service, "_generate_no", flaky_generate)
|
||
base_ok = _receipt_log_count(writer["username"], success=True)
|
||
base_fail = _receipt_log_count(writer["username"], success=False)
|
||
|
||
second = client.post(
|
||
_RECEIPTS, json=_receipt_payload(business["id"]), headers=writer["headers"]
|
||
)
|
||
assert second.status_code == 200, second.text
|
||
assert calls["n"] == 2, "重试应恰好再取一次号"
|
||
assert _receipt_log_count(writer["username"], success=True) == base_ok + 1
|
||
assert _receipt_log_count(writer["username"], success=False) == base_fail
|
||
|
||
def test_持续冲突_最终失败且审计恰一条失败(
|
||
self, quiet_client: TestClient, writer: dict, monkeypatch
|
||
) -> None:
|
||
client = quiet_client
|
||
business = create_business(client, writer["headers"])
|
||
first = client.post(
|
||
_RECEIPTS, json=_receipt_payload(business["id"]), headers=writer["headers"]
|
||
)
|
||
assert first.status_code == 200, first.text
|
||
existing_no = first.json()["data"]["receipt_batch_no"]
|
||
|
||
def always_conflict(session):
|
||
return existing_no
|
||
|
||
monkeypatch.setattr(receipt_service, "_generate_no", always_conflict)
|
||
base_ok = _receipt_log_count(writer["username"], success=True)
|
||
base_fail = _receipt_log_count(writer["username"], success=False)
|
||
|
||
second = client.post(
|
||
_RECEIPTS, json=_receipt_payload(business["id"]), headers=writer["headers"]
|
||
)
|
||
assert second.status_code == 400, second.text
|
||
assert _receipt_log_count(writer["username"], success=False) == base_fail + 1
|
||
assert _receipt_log_count(writer["username"], success=True) == base_ok
|