152 lines
4.3 KiB
Python
152 lines
4.3 KiB
Python
from fastapi.testclient import TestClient
|
||
from sqlalchemy.orm import Session
|
||
|
||
from app.models.attachment import AttachmentCategory
|
||
|
||
|
||
def _seed(db: Session) -> list[int]:
|
||
cats = [
|
||
AttachmentCategory(name="TEST_CERT", sort_order=1, is_active=True),
|
||
AttachmentCategory(name="TEST_EXTRA", sort_order=2, is_active=True),
|
||
]
|
||
for c in cats:
|
||
db.add(c)
|
||
db.commit()
|
||
for c in cats:
|
||
db.refresh(c)
|
||
return [c.id for c in cats]
|
||
|
||
|
||
def test_attachment_box_basic_and_flip(client: TestClient, db: Session):
|
||
"""扫附件码即到货：未判定 → 自动 has，item 不存在则按 ERP 数量建。"""
|
||
cert_id, _ = _seed(db)
|
||
resp = client.post(
|
||
"/CargoTrace/attachment/box",
|
||
json={
|
||
"zongpai_no": "26BW0011",
|
||
"category_id": cert_id,
|
||
"box_no": 931,
|
||
"quantity": 30,
|
||
},
|
||
)
|
||
assert resp.status_code == 200
|
||
body = resp.json()
|
||
assert body["paichan_no"] == "W00009"
|
||
assert body["box_no"] == 931
|
||
assert body["quantity"] == 30
|
||
|
||
status = client.get(
|
||
"/CargoTrace/attachment/status", params={"zongpai_no": "26BW0011"}
|
||
).json()
|
||
assert status["determination"] == "has"
|
||
item = status["items"][0]
|
||
assert item["expected_qty"] == 80 # ERP 数量兜底
|
||
assert item["boxed_qty"] == 30
|
||
|
||
|
||
def test_attachment_box_rejects_over_expected(client: TestClient, db: Session):
|
||
cert_id, _ = _seed(db)
|
||
client.put(
|
||
"/CargoTrace/attachment/config",
|
||
json={
|
||
"zongpai_no": "26BW0011",
|
||
"determination": "has",
|
||
"items": [{"category_id": cert_id, "expected_qty": 80}],
|
||
},
|
||
)
|
||
first = client.post(
|
||
"/CargoTrace/attachment/box",
|
||
json={
|
||
"zongpai_no": "26BW0011",
|
||
"category_id": cert_id,
|
||
"box_no": 932,
|
||
"quantity": 70,
|
||
},
|
||
)
|
||
assert first.status_code == 200
|
||
over = client.post(
|
||
"/CargoTrace/attachment/box",
|
||
json={
|
||
"zongpai_no": "26BW0011",
|
||
"category_id": cert_id,
|
||
"box_no": 933,
|
||
"quantity": 11,
|
||
},
|
||
)
|
||
assert over.status_code == 400
|
||
assert over.json()["error_code"] == "INVALID_QUANTITY"
|
||
|
||
|
||
def test_attachment_box_duplicate_in_same_box(client: TestClient, db: Session):
|
||
cert_id, _ = _seed(db)
|
||
client.post(
|
||
"/CargoTrace/attachment/box",
|
||
json={
|
||
"zongpai_no": "26BW0011",
|
||
"category_id": cert_id,
|
||
"box_no": 934,
|
||
"quantity": 10,
|
||
},
|
||
)
|
||
dup = client.post(
|
||
"/CargoTrace/attachment/box",
|
||
json={
|
||
"zongpai_no": "26BW0011",
|
||
"category_id": cert_id,
|
||
"box_no": 934,
|
||
"quantity": 5,
|
||
},
|
||
)
|
||
assert dup.status_code == 409
|
||
assert dup.json()["error_code"] == "DUPLICATE_BOX_NO"
|
||
|
||
|
||
def test_attachment_box_mixed_with_product(client: TestClient, db: Session):
|
||
"""附件与产品同箱：先装产品，再装附件，同 box_no。"""
|
||
cert_id, _ = _seed(db)
|
||
client.post(
|
||
"/CargoTrace/box",
|
||
json={"zongpai_no": "26BW0011", "box_no": 935, "quantity": 10},
|
||
)
|
||
resp = client.post(
|
||
"/CargoTrace/attachment/box",
|
||
json={
|
||
"zongpai_no": "26BW0011",
|
||
"category_id": cert_id,
|
||
"box_no": 935,
|
||
"quantity": 5,
|
||
},
|
||
)
|
||
assert resp.status_code == 200
|
||
info = client.get(
|
||
"/CargoTrace/box/info", params={"zongpai_no": "26BW0011"}
|
||
).json()
|
||
box = next(b for b in info["existing_boxes"] if b["box_no"] == 935)
|
||
kinds = {i["kind"] for i in box["items"]}
|
||
assert kinds == {"product", "attachment"}
|
||
|
||
|
||
def test_attachment_box_completion(client: TestClient, db: Session):
|
||
cert_id, _ = _seed(db)
|
||
client.put(
|
||
"/CargoTrace/attachment/config",
|
||
json={
|
||
"zongpai_no": "26BW0011",
|
||
"determination": "has",
|
||
"items": [{"category_id": cert_id, "expected_qty": 80}],
|
||
},
|
||
)
|
||
client.post(
|
||
"/CargoTrace/attachment/box",
|
||
json={
|
||
"zongpai_no": "26BW0011",
|
||
"category_id": cert_id,
|
||
"box_no": 936,
|
||
"quantity": 80,
|
||
},
|
||
)
|
||
status = client.get(
|
||
"/CargoTrace/attachment/status", params={"zongpai_no": "26BW0011"}
|
||
).json()
|
||
assert status["all_complete"] is True
|