feat(box): include item total quantity
This commit is contained in:
@@ -4,7 +4,11 @@ from sqlalchemy.orm import Session
|
||||
|
||||
from app.core.database import SessionLocal
|
||||
from app.main import app
|
||||
from app.models.finished_goods import FinishedGoodsLocation
|
||||
from app.models.finished_goods import (
|
||||
FinishedGoodsBox,
|
||||
FinishedGoodsBoxItem,
|
||||
FinishedGoodsLocation,
|
||||
)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
@@ -34,4 +38,20 @@ def cleanup_test_data(db: Session):
|
||||
db.query(FinishedGoodsLocation).filter(
|
||||
FinishedGoodsLocation.zongpai_no.in_(test_zongpai_nos)
|
||||
).delete(synchronize_session=False)
|
||||
test_boxes = (
|
||||
db.query(FinishedGoodsBox)
|
||||
.filter(
|
||||
FinishedGoodsBox.paichan_no == "W00009",
|
||||
FinishedGoodsBox.box_no >= 900,
|
||||
)
|
||||
.all()
|
||||
)
|
||||
test_box_ids = [box.id for box in test_boxes]
|
||||
if test_box_ids:
|
||||
db.query(FinishedGoodsBoxItem).filter(
|
||||
FinishedGoodsBoxItem.box_id.in_(test_box_ids)
|
||||
).delete(synchronize_session=False)
|
||||
db.query(FinishedGoodsBox).filter(
|
||||
FinishedGoodsBox.id.in_(test_box_ids)
|
||||
).delete(synchronize_session=False)
|
||||
db.commit()
|
||||
|
||||
Reference in New Issue
Block a user