75 lines
2.7 KiB
Python
75 lines
2.7 KiB
Python
|
# ============================================================
|
||
|
|
# MD-04 Excel/CSV 导入管线黄金测试
|
||
|
|
# ============================================================
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from server.aps_domain.importers import apply_import_commit, preview_file, validate_batch
|
||
|
|
from server.state.seed import seed_world
|
||
|
|
|
||
|
|
|
||
|
|
def _next_id_factory(world):
|
||
|
|
tables = {
|
||
|
|
"audit": "auditEvents",
|
||
|
|
"salesOrder": "salesOrders",
|
||
|
|
"material": "materials",
|
||
|
|
}
|
||
|
|
counters: dict[str, int] = {}
|
||
|
|
|
||
|
|
def next_id(kind: str) -> int:
|
||
|
|
if kind not in counters:
|
||
|
|
rows = world.get(tables.get(kind, kind + "s"), [])
|
||
|
|
counters[kind] = max((r.get("id", 0) for r in rows if isinstance(r, dict)), default=0)
|
||
|
|
counters[kind] += 1
|
||
|
|
return counters[kind]
|
||
|
|
|
||
|
|
return next_id
|
||
|
|
|
||
|
|
|
||
|
|
def test_validate_equipment_batch():
|
||
|
|
world = seed_world()
|
||
|
|
rows = [
|
||
|
|
{"code": "EQ-X1", "name": "试验压接机", "capabilities": "OP-PRESS,OP-WELD",
|
||
|
|
"opStdTime": "OP-PRESS:2,OP-WELD:5", "movable": "是", "zone": "ZONE-A",
|
||
|
|
"availabilityRate": "0.95", "status": "RUNNING"},
|
||
|
|
]
|
||
|
|
batch = validate_batch("equipment", rows, world)
|
||
|
|
assert batch["okCount"] == 1
|
||
|
|
assert batch["errorCount"] == 0
|
||
|
|
assert "OP-PRESS" in batch["okRows"][0]["capabilities"]
|
||
|
|
|
||
|
|
|
||
|
|
def test_preview_csv_orders_and_commit():
|
||
|
|
world = seed_world()
|
||
|
|
product = next(m for m in world["materials"] if m.get("type") == "FINISHED_PRODUCT")
|
||
|
|
before = len(world["salesOrders"])
|
||
|
|
csv_text = (
|
||
|
|
"订单号,客户,产品编码,数量,交期,优先级\n"
|
||
|
|
f"SO-IMP-1,导入客户,{product['code']},50,2026-08-01,3\n"
|
||
|
|
).encode("utf-8")
|
||
|
|
preview = preview_file("orders.csv", csv_text, world)
|
||
|
|
assert preview["canCommit"] is True
|
||
|
|
assert preview["totalOk"] >= 1
|
||
|
|
assert preview["batches"][0]["kind"] == "orders"
|
||
|
|
|
||
|
|
applied = apply_import_commit(world, _next_id_factory(world), preview["batches"])
|
||
|
|
assert applied["total"] >= 1
|
||
|
|
assert len(world["salesOrders"]) == before + 1
|
||
|
|
assert any(o.get("customerName") == "导入客户" for o in world["salesOrders"])
|
||
|
|
|
||
|
|
|
||
|
|
def test_commit_flex_equipment():
|
||
|
|
world = seed_world()
|
||
|
|
batches = [{
|
||
|
|
"kind": "equipment",
|
||
|
|
"sheet": "设备",
|
||
|
|
"okRows": [{
|
||
|
|
"code": "EQ-NEW-99", "name": "新压接", "capabilities": ["OP-PRESS"],
|
||
|
|
"opStdTime": {"OP-PRESS": 1.5}, "movable": True, "moveTimeMin": 15,
|
||
|
|
"zone": "ZONE-A", "availabilityRate": 1.0, "status": "RUNNING",
|
||
|
|
"adaptableMolds": [],
|
||
|
|
}],
|
||
|
|
}]
|
||
|
|
applied = apply_import_commit(world, _next_id_factory(world), batches)
|
||
|
|
assert applied["summary"].get("flexEquipment", 0) == 1
|
||
|
|
assert any(e["code"] == "EQ-NEW-99" for e in world["flexEquipment"])
|