server-core/scripts/smoke_profitsharing.py

# -*- coding: utf-8 -*-
"""微信服务商分账改造 · 冒烟测试（临时 sqlite，不碰主库）。

覆盖：
1. release_escrow：未接微信收单/分账未配置 → 降级台账结算（channel=manual, released）
2. release_escrow：配置齐备 + 接单者 active 绑定 → 发起微信分账（mock create_split）
3. handle_split_notify：回调确认 → escrow released + log shared（幂等）
"""
from __future__ import annotations

import asyncio
import os
import sys
import tempfile
from pathlib import Path

# 临时库：优先 sqlite 内存/临时文件，避免触碰 .env 指定的 MySQL 主库
_tmp = tempfile.NamedTemporaryFile(suffix=".db", delete=False)
_tmp.close()
os.environ["PINEAGENTS_DEMO_DATABASE_URL"] = f"sqlite+aiosqlite:///{_tmp.name}"

ROOT = Path(__file__).resolve().parent
sys.path.insert(0, str(ROOT))

from unittest.mock import patch  # noqa: E402

from app.infrastructure.db import Base  # noqa: E402
from app.infrastructure.repositories import Database  # noqa: E402
import app.pay.models  # noqa: E402,F401  确保新模型注册
import app.infrastructure.models  # noqa: E402,F401
from app.services.settlement_service import SettlementService  # noqa: E402


async def _setup(db: Database):
    async with db._engine.begin() as conn:
        await conn.run_sync(Base.metadata.create_all)


async def _mk_completed_task(db, claimer: str = "") -> dict:
    return await db.tasks.create({
        "task_code": "smk001", "title": "冒烟任务", "status": "completed",
        "budget_min": 1000, "budget_max": 1000, "claimed_by": claimer or None,
    })


async def main() -> None:
    db = Database()
    await _setup(db)

    # ── 1) 未配置分账 / 无收单 → 降级台账结算 ────────────────────────────────
    task = await _mk_completed_task(db)
    svc = SettlementService(db)
    r = await svc.release_escrow(task["id"])
    print("① 降级台账:", r["status"], r["channel"], r["share_status"])
    assert r["status"] == "released" and r["channel"] == "manual"
    # 直接查库验证流水
    import sqlite3
    conn = sqlite3.connect(_tmp.name)
    row = conn.execute("SELECT channel,status,detail FROM settlement_logs ORDER BY created_at DESC LIMIT 1").fetchone()
    conn.close()
    print("① 流水:", row)
    assert row and row[0] == "manual" and row[1] == "shared"

    # ── 2) 配置齐备 + 接单者 active 绑定 → 发起分账 ─────────────────────────
    from app.pay.repository import PaymentBindingRepository
    # 造一个接单者用户 + 商户绑定
    user = await db.users.create("smk_opc", "pw123456", nickname="接单者", role="opc_member")
    bind = await PaymentBindingRepository(db.session).create({
        "user_id": user["id"], "bind_type": "merchant",
        "sub_mchid": "1900000109", "status": "active",
    })
    task2 = await _mk_completed_task(db)
    await db.tasks.claim(task2["id"], user["id"])  # 接单者=用户
    await db.tasks.set_status(task2["id"], "completed")  # 接单后置完成（验收通过）
    esc = await db.escrows.create(task2["id"], task2["title"], 1000, 50)
    await db.escrows.update_fields(esc["id"], {"transaction_id": "4200001234"})
    with patch.object(
        __import__("app.pay.config", fromlist=["profitsharing_enabled"]),
        "profitsharing_enabled", return_value=True,
    ), patch("app.pay.profitsharing.create_split") as mock_split:
        r2 = await svc.release_escrow(task2["id"])
        mock_split.assert_called_once()
        call = mock_split.call_args.kwargs
        print("② 分账发起: channel=%s share_status=%s" % (r2["channel"], r2["share_status"]))
        print("② 接收方:", [(x["type"], x["amount"]) for x in call["receivers"]])
        assert r2["channel"] == "wx_split" and r2["share_status"] == "sharing"
        assert call["receivers"][0]["amount"] == 95000 and call["receivers"][1]["amount"] == 5000
        assert call["receivers"][0]["type"] == "MERCHANT_ID"  # 商户绑定 → MERCHANT_ID
        share_no = call["out_order_no"]

    # ── 3) 回调确认（幂等）───────────────────────────────────────────────────
    ok1 = await svc.handle_split_notify({"out_order_no": share_no})
    conn = sqlite3.connect(_tmp.name)
    esc_state = conn.execute("SELECT status,share_status FROM escrows WHERE id=?", (esc["id"],)).fetchone()
    log_state = conn.execute("SELECT status FROM settlement_logs WHERE share_order_no=?", (share_no,)).fetchone()
    conn.close()
    print("③ 回调后 escrow:", esc_state, "log:", log_state)
    assert ok1 and esc_state == ("released", "shared") and log_state == ("shared",)
    ok2 = await svc.handle_split_notify({"out_order_no": share_no})  # 重放
    assert ok2  # 幂等不报错
    print("③ 幂等重放 OK")

    # ── 4) 特约商户进件闭环（mock 微信侧；用独立用户避开 ② 的 active 绑定）───
    from app.pay import service as pay_service
    user2 = await db.users.create("smk_opc2", "pw123456", nickname="进件方", role="opc_member")
    applyment_body = {
        "contact_info": {"contact_name": "张三", "mobile_phone": "13900000000"},
        "subject_info": {"subject_type": "SUBJECT_TYPE_ENTERPRISE",
                         "business_license_info": {"merchant_name": "测试公司",
                                                   "legal_person": "张三"}},
        "business_info": {"merchant_shortname": "测试商户"},
        "settlement_info": {"settlement_id": "719"},
        "bank_account_info": {"account_name": "测试公司", "account_bank": "工商银行",
                              "bank_address_code": "110000", "account_number": "1234567890"},
    }
    with patch.object(
        __import__("app.pay.config", fromlist=["profitsharing_enabled"]),
        "profitsharing_enabled", return_value=True,
    ), patch("app.pay.profitsharing.submit_applyment",
             return_value={"applyment_id": "2000001234567890",
                           "applyment_state": "AUDITING"}) as mock_sub:
        r4 = await pay_service.create_merchant_applyment(db, user2, **applyment_body)
        print("④ 进件提交: state=%s binding.status=%s applyment_id=%s"
              % (r4["applyment_state"], r4["binding"]["status"], r4["applyment_id"]))
        assert r4["binding"]["status"] == "applying"
        mock_sub.assert_called_once()
        # 幂等：重复提交返回已有，不二次调微信
        r4b = await pay_service.create_merchant_applyment(db, user2, **applyment_body)
        assert mock_sub.call_count == 1 and r4b["applyment_state"] == "EXISTING"
        print("④ 幂等：重复进件被拦截（微信仅调用 1 次）")
        binding_id = r4["binding"]["id"]

    # 4b) 查询进件：审核通过 → 回填 sub_mchid + 添加接收方 → active
    with patch.object(
        __import__("app.pay.config", fromlist=["profitsharing_enabled"]),
        "profitsharing_enabled", return_value=True,
    ), patch("app.pay.profitsharing.query_applyment",
             return_value={"applyment_state": "FINISHED", "sub_mchid": "1900000109"}), \
         patch("app.pay.profitsharing.add_receiver", return_value={}) as mock_add:
        r5 = await pay_service.query_merchant_applyment(db, user2, binding_id)
        print("④ 进件通过: state=%s sub_mchid=%s binding.status=%s"
              % (r5["applyment_state"], r5["sub_mchid"], r5["binding"]["status"]))
        assert r5["sub_mchid"] == "1900000109" and r5["binding"]["status"] == "active"
        mock_add.assert_called_once_with(account_type="MERCHANT_ID", account="1900000109")
        print("④ 自动添加 MERCHANT_ID 接收方 OK")

    # 4c) 驳回场景：新开一个绑定（绕开幂等）→ REJECTED 标记
    with patch.object(
        __import__("app.pay.config", fromlist=["profitsharing_enabled"]),
        "profitsharing_enabled", return_value=True,
    ), patch("app.pay.profitsharing.query_applyment",
             return_value={"applyment_state": "REJECTED",
                           "audit_detail": "营业执照信息不清晰"}) as mock_q:
        # 直接落一条 applying 记录再查
        b_rej = await PaymentBindingRepository(db.session).create({
            "user_id": user2["id"], "bind_type": "merchant",
            "applyment_id": "2000000000000001", "status": "applying"})
        r6 = await pay_service.query_merchant_applyment(db, user2, b_rej["id"])
        print("④ 进件驳回: state=%s status=%s detail=%s"
              % (r6["applyment_state"], r6["binding"]["status"], r6["binding"]["detail"]))
        assert r6["binding"]["status"] == "rejected" and "营业执照" in r6["binding"]["detail"]
        mock_q.assert_called()

    print("\n✅ 冒烟测试全部通过")
    await db.close()
    os.unlink(_tmp.name)


if __name__ == "__main__":
    asyncio.run(main())