Files
bj_power/bj_power_wms/e2e_scope_dist.py
T

83 lines
3.4 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
"""端到端验证:问题四(备料台账无单选全部) + 问题六(盘点范围 + 物料分布)"""
import json, urllib.request, urllib.parse, urllib.error
BASE = "http://127.0.0.1:8902"
TOKEN = None
def req(method, path, payload=None, params=None):
url = BASE + path
if params:
qs = urllib.parse.urlencode({k: v for k, v in params.items() if v not in (None, "")})
if qs:
url += "?" + qs
data = json.dumps(payload).encode() if payload is not None else None
r = urllib.request.Request(url, data=data, method=method)
r.add_header("Content-Type", "application/json")
if TOKEN:
r.add_header("Authorization", "Bearer " + TOKEN)
try:
with urllib.request.urlopen(r, timeout=20) as resp:
return json.loads(resp.read().decode())
except urllib.error.HTTPError as e:
return {"_http": e.code, "_body": e.read().decode()[:300]}
def login():
global TOKEN
d = req("POST", "/api/auth/login", {"username": "admin", "password": "123456"})
TOKEN = d.get("data", {}).get("token")
assert TOKEN, f"登录失败: {d}"
print("✓ 登录成功 admin")
passed = []
def check(name, cond, extra=""):
passed.append(bool(cond))
print(("✓ " if cond else "✗ ") + name + (f" {extra}" if extra and not cond else ""))
def main():
login()
# ---------- 问题四:备料台账无单号查全部 ----------
d = req("GET", "/api/ledger/query", params={"page": 1, "pageSize": 20})
t4 = d.get("data", {}) or d
check("ledger/query 无 orderNo 返回 total+list", "list" in t4 and "total" in t4,
f"got {list(t4.keys())}")
# ---------- 问题六:盘点范围按物料过滤 ----------
# 先看库存有哪些物料可盘点,选一个真实物料
st = req("GET", "/api/stock/query", params={"page": 1, "pageSize": 10})
stock_data = st.get("data", {}) or st
rows = stock_data.get("list") or []
if rows:
mat = rows[0]["materialCode"]
# 按该物料发起盘点
d = req("POST", "/api/stocktake/start", {"operator": "e2e", "materialCodes": [mat]})
data = d.get("data", {}) or d
check("startStocktake 按物料过滤成功", bool(data.get("stocktakeNo")),
f"got {data}")
check("按物料盘点 totalTargets>0", data.get("totalTargets", 0) > 0,
f"totalTargets={data.get('totalTargets')}")
# 按类型过滤发起
d2 = req("POST", "/api/stocktake/start", {"operator": "e2e", "manageMode": 1})
data2 = d2.get("data", {}) or d2
check("startStocktake 按类型(结构件)成功", bool(data2.get("stocktakeNo")),
f"got {data2}")
else:
check("stock/query 有数据(需先造数才能验盘点范围)", False, "库存为空")
# ---------- 问题六:物料分布数据源(聚合含 zoneCode) ----------
d = req("GET", "/api/stock/query", params={"page": 1, "pageSize": 50})
dist = d.get("data", {}) or d
dl = dist.get("list") or []
check("stock/query 返回 list(分布图数据源)", len(dl) > 0, f"len={len(dl)}")
if dl:
r0 = dl[0]
check("聚合行含 zoneCode/materialCode/totalQty",
all(k in r0 for k in ("zoneCode", "materialCode", "totalQty")),
f"keys={list(r0.keys())}")
print("\n==== 结果:%d/%d 通过 ====" % (sum(passed), len(passed)))
return all(passed)
if __name__ == "__main__":
ok = main()
raise SystemExit(0 if ok else 1)