225 lines
7.9 KiB
Python
225 lines
7.9 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""2026-09-07 预算版本管理UI修复 - 浏览器页面实测
|
||
流程: 登录博海(entity2) → /budget 版本管理 → 提取行DOM断言 → switch-entity 到酣客(entity1) → 再提取断言
|
||
验收: 博海 v2026H2 行: 状态草稿/条目数25/总额462/最后更新非空/提交审批按钮可见
|
||
酣客 active版本有提交审批; 2026H2 approved 显示已批准 无提交按钮
|
||
"""
|
||
import json
|
||
import sys
|
||
import time
|
||
import urllib.request
|
||
import websocket
|
||
|
||
CDP_HTTP = "http://127.0.0.1:9224"
|
||
BASE = "https://cma.sxbh.ltd/api/cma"
|
||
|
||
|
||
def http_get(path):
|
||
with urllib.request.urlopen(CDP_HTTP + path, timeout=5) as r:
|
||
return json.loads(r.read().decode())
|
||
|
||
|
||
class CDP:
|
||
def __init__(self, ws_url):
|
||
self.ws = websocket.create_connection(ws_url, timeout=40)
|
||
self.msg_id = 0
|
||
|
||
def call(self, method, params=None):
|
||
self.msg_id += 1
|
||
mid = self.msg_id
|
||
self.ws.send(json.dumps({"id": mid, "method": method, "params": params or {}}))
|
||
while True:
|
||
resp = json.loads(self.ws.recv())
|
||
if resp.get("id") == mid:
|
||
if "error" in resp:
|
||
raise RuntimeError(f"{method}: {resp['error']}")
|
||
return resp.get("result", {})
|
||
|
||
def eval(self, expr):
|
||
r = self.call("Runtime.evaluate", {"expression": expr, "returnByValue": True, "awaitPromise": True})
|
||
if r.get("exceptionDetails"):
|
||
return "EXC: " + json.dumps(r["exceptionDetails"], ensure_ascii=False)[:300]
|
||
return r.get("result", {}).get("value")
|
||
|
||
def close(self):
|
||
try:
|
||
self.ws.close()
|
||
except Exception:
|
||
pass
|
||
|
||
|
||
def open_page(url):
|
||
ver = http_get("/json/version")
|
||
bws = CDP(ver["webSocketDebuggerUrl"])
|
||
t = bws.call("Target.createTarget", {"url": url})
|
||
tid = t["targetId"]
|
||
bws.close()
|
||
time.sleep(2.5)
|
||
for p in http_get("/json"):
|
||
if p["id"] == tid:
|
||
return tid, p["webSocketDebuggerUrl"]
|
||
return tid, None
|
||
|
||
|
||
def wait_for(c, expr, timeout=25, interval=1.0):
|
||
"""轮询直到 expr 为 truthy，返回其值"""
|
||
for _ in range(int(timeout / interval)):
|
||
v = c.eval(expr)
|
||
if v:
|
||
return v
|
||
time.sleep(interval)
|
||
return None
|
||
|
||
|
||
def login_via_fetch(c, entity_id):
|
||
"""同源 fetch 登录并写 localStorage，返回状态文本"""
|
||
expr = f"""
|
||
(async () => {{
|
||
const r = await fetch('{BASE}/auth/login', {{
|
||
method: 'POST', headers: {{'Content-Type': 'application/json'}},
|
||
body: JSON.stringify({{username:'admin', password:'admin123', entity_id:{entity_id}}})
|
||
}});
|
||
const j = await r.json();
|
||
if (!r.ok) return 'LOGIN_FAIL:' + JSON.stringify(j);
|
||
localStorage.setItem('cma_token', j.token);
|
||
localStorage.setItem('cma_entity_id', String(j.entity_id));
|
||
localStorage.setItem('cma_entity_name', j.entity_name || '');
|
||
localStorage.setItem('cma_user', JSON.stringify(j.user || {{}}));
|
||
return 'LOGIN_OK:' + j.entity_name + ':' + j.token.slice(0, 8);
|
||
}})()
|
||
"""
|
||
return c.eval(expr)
|
||
|
||
|
||
JS_GO_BUDGET = """
|
||
(() => {
|
||
location.href = '/budget';
|
||
return 'NAV';
|
||
})()
|
||
"""
|
||
|
||
JS_EXTRACT = r"""
|
||
(() => {
|
||
// 等表格渲染完
|
||
const rows = Array.from(document.querySelectorAll('.el-table__body-wrapper tbody tr'));
|
||
if (!rows.length) return 'NO_ROWS';
|
||
const out = [];
|
||
for (const tr of rows) {
|
||
const tds = Array.from(tr.querySelectorAll('td'));
|
||
if (tds.length < 6) continue;
|
||
const getTxt = (td) => (td.innerText || '').trim().replace(/\s+/g, ' ');
|
||
const version = getTxt(tds[0]);
|
||
const status = getTxt(tds[1]);
|
||
const planCount = getTxt(tds[2]);
|
||
const total = getTxt(tds[3]);
|
||
const updated = getTxt(tds[4]);
|
||
const opTxt = (tds[5] ? tds[5].innerText || '' : '').trim().replace(/\s+/g, ' ');
|
||
const btns = Array.from(tds[5].querySelectorAll('button')).map(b => (b.innerText || '').trim().replace(/\s+/g, ''));
|
||
const checkboxCount = tds[5].querySelectorAll('.el-checkbox').length;
|
||
out.push({ version, status, planCount, total, updated, opTxt, btns, checkboxCount });
|
||
}
|
||
return JSON.stringify(out);
|
||
})()
|
||
"""
|
||
|
||
JS_CLICK_VERSION_TAB = """
|
||
(() => {
|
||
const tabs = Array.from(document.querySelectorAll('[role="tab"]'));
|
||
const t = tabs.find(e => (e.innerText || '').replace(/\\s/g, '') === '版本管理');
|
||
if (!t) return 'NO_TAB:' + tabs.map(x => x.innerText).join('|');
|
||
t.click();
|
||
return 'TAB_CLICKED';
|
||
})()
|
||
"""
|
||
|
||
|
||
def main():
|
||
tid, ws = open_page("https://cma.sxbh.ltd/login")
|
||
if not ws:
|
||
print("FAIL: no ws for new page")
|
||
sys.exit(1)
|
||
c = CDP(ws)
|
||
c.call("Page.enable")
|
||
c.call("Runtime.enable")
|
||
# 等待页面导航到 cma 域完成（约 blank → login 竞态修复，2026-09-07 复跑实测：
|
||
# Target.createTarget 后偶发停留在 about:blank → 访问 localStorage 抛 SecurityError）
|
||
nav = wait_for(c, "location.hostname === 'cma.sxbh.ltd'", timeout=25)
|
||
if not nav:
|
||
print("FAIL: 页面未导航到 cma.sxbh.ltd")
|
||
sys.exit(1)
|
||
time.sleep(3)
|
||
|
||
# 1) 登录博海 entity=2
|
||
r = login_via_fetch(c, 2)
|
||
print("LOGIN_E2:", r)
|
||
if not (isinstance(r, str) and r.startswith("LOGIN_OK")):
|
||
print("FAIL login entity2")
|
||
sys.exit(1)
|
||
c.eval(JS_GO_BUDGET)
|
||
time.sleep(5)
|
||
print("URL1:", c.eval("location.href"))
|
||
# 确认当前账套
|
||
print("ENTITY1:", c.eval("localStorage.getItem('cma_entity_id')"))
|
||
tab = wait_for(c, JS_CLICK_VERSION_TAB, timeout=15)
|
||
print("TAB1:", tab)
|
||
time.sleep(4)
|
||
rows_json = wait_for(c, JS_EXTRACT, timeout=20)
|
||
print("ROWS_E2_RAW:", rows_json)
|
||
rows_e2 = json.loads(rows_json) if isinstance(rows_json, str) and rows_json.startswith("[") else []
|
||
for row in rows_e2:
|
||
print(" E2", json.dumps(row, ensure_ascii=False))
|
||
|
||
# 断言博海
|
||
ok = True
|
||
target = [r for r in rows_e2 if r.get("version", "").strip().startswith("v2026H2")]
|
||
if not target:
|
||
print("FAIL: 博海未见 v2026H2")
|
||
ok = False
|
||
else:
|
||
r = target[0]
|
||
assert "草稿" in r["status"], f"状态应含草稿, got {r['status']}"
|
||
assert r["planCount"].replace(" ", "") == "25", f"条目数应25, got {r['planCount']}"
|
||
assert "462" in r["total"].replace(",", ""), f"总额应462, got {r['total']}"
|
||
assert r["updated"], "最后更新应非空"
|
||
assert "提交审批" in r["opTxt"], f"应有提交审批按钮, got {r['btns']}"
|
||
print("E2_ASSERT_OK:", json.dumps(r, ensure_ascii=False))
|
||
|
||
# 2) 切换酣客 entity=1（同页面 fetch switch-entity 或直接重新登录）
|
||
r = login_via_fetch(c, 1)
|
||
print("LOGIN_E1:", r)
|
||
c.eval(JS_GO_BUDGET)
|
||
time.sleep(5)
|
||
print("ENTITY2:", c.eval("localStorage.getItem('cma_entity_id')"))
|
||
tab = wait_for(c, JS_CLICK_VERSION_TAB, timeout=15)
|
||
print("TAB2:", tab)
|
||
time.sleep(4)
|
||
rows_json = wait_for(c, JS_EXTRACT, timeout=20)
|
||
print("ROWS_E1_RAW:", rows_json)
|
||
rows_e1 = json.loads(rows_json) if isinstance(rows_json, str) and rows_json.startswith("[") else []
|
||
for row in rows_e1:
|
||
print(" E1", json.dumps(row, ensure_ascii=False))
|
||
|
||
ok2 = True
|
||
for row in rows_e1:
|
||
ver = row.get("version", "").strip()
|
||
if ver == "2026H2":
|
||
if "已批准" not in row["status"] or "提交审批" in row["opTxt"]:
|
||
print(f"FAIL: 酣客2026H2 approved 应只读已批准无提交, got {json.dumps(row, ensure_ascii=False)}")
|
||
ok2 = False
|
||
elif row["status"].replace(" ", "").startswith("草稿"):
|
||
if "提交审批" not in row["opTxt"]:
|
||
print(f"FAIL: 酣客{ver} active/草稿 应有提交审批, got {json.dumps(row, ensure_ascii=False)}")
|
||
ok2 = False
|
||
if not rows_e1:
|
||
ok2 = False
|
||
print("FAIL: 酣客无行")
|
||
|
||
print("\n" + ("ALL_PASS" if (ok and ok2) else "HAS_FAIL"))
|
||
c.close()
|
||
sys.exit(0 if (ok and ok2) else 1)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|