56 lines
2.4 KiB
Python
56 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""2026-09-07 预算版本管理UI修复 - 生产API验证"""
|
|
import json, urllib.request, ssl, sys
|
|
|
|
BASE = "https://cma.sxbh.ltd/api/cma"
|
|
|
|
def http(url, data=None, headers=None, method=None):
|
|
req = urllib.request.Request(url, data=data, headers=headers or {}, method=method)
|
|
ctx = ssl.create_default_context()
|
|
ctx.check_hostname = False
|
|
ctx.verify_mode = ssl.CERT_NONE
|
|
with urllib.request.urlopen(req, timeout=30, context=ctx) as r:
|
|
return json.loads(r.read().decode())
|
|
|
|
def login(entity_id):
|
|
body = json.dumps({"username": "admin", "password": "admin123", "entity_id": entity_id}).encode()
|
|
r = http(BASE + "/auth/login", data=body, headers={"Content-Type": "application/json"}, method="POST")
|
|
return r["token"]
|
|
|
|
def versions(token, year=None):
|
|
url = BASE + "/budget/versions" + (f"?year={year}" if year else "")
|
|
r = http(url, headers={"Authorization": f"Bearer {token}"})
|
|
return r if isinstance(r, list) else r.get("data", r)
|
|
|
|
ok = True
|
|
for eid, ename in [(1, "酣客"), (2, "博海")]:
|
|
try:
|
|
tok = login(eid)
|
|
except Exception as e:
|
|
print(f"[{ename} entity{eid}] 登录失败: {e}")
|
|
ok = False
|
|
continue
|
|
try:
|
|
vs = versions(tok)
|
|
print(f"\n=== {ename} (entity={eid}) 共{len(vs)}个版本 ===")
|
|
for v in vs:
|
|
keys = sorted(v.keys())
|
|
print(f" {v['version']:>10} | status={v['status']:<10} | approval_status={v['approval_status']:<10} | plan_count={v['plan_count']} | kpi_count={v['kpi_count']} | total={v['total_budget']} | last_updated={v['last_updated']!r} | period_count={v['period_count']}")
|
|
# 断言新字段存在
|
|
for f in ["approval_status", "plan_count", "last_updated", "status", "kpi_count", "updated_at"]:
|
|
assert f in v, f"缺字段 {f}"
|
|
# 状态机映射断言
|
|
if v["status"] == "active":
|
|
assert v["approval_status"] == "draft", f"{v['version']} active 应映射 draft"
|
|
else:
|
|
assert v["approval_status"] == v["status"]
|
|
assert v["plan_count"] == v["kpi_count"]
|
|
assert v["last_updated"] == v["updated_at"]
|
|
except Exception as e:
|
|
print(f"[{ename}] 版本校验异常: {type(e).__name__} {e}")
|
|
ok = False
|
|
|
|
print("\n" + ("ALL_PASS" if ok else "HAS_FAIL"))
|
|
sys.exit(0 if ok else 1)
|