baa841ae42
Adds 5 tables in the CargoTrace schema: attachment_category, finished_goods_attachment, finished_goods_attachment_item, finished_goods_attachment_location, finished_goods_box_attachment_item. Includes a one-off create_all script and extends test cleanup. Co-Authored-By: Claude <noreply@anthropic.com>
90 lines
2.7 KiB
Python
90 lines
2.7 KiB
Python
import pytest
|
|
from fastapi.testclient import TestClient
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.core.database import SessionLocal
|
|
from app.main import app
|
|
from app.models.finished_goods import (
|
|
FinishedGoodsBox,
|
|
FinishedGoodsBoxItem,
|
|
FinishedGoodsLocation,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def client():
|
|
return TestClient(app)
|
|
|
|
|
|
@pytest.fixture
|
|
def db():
|
|
session = SessionLocal()
|
|
yield session
|
|
session.close()
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def cleanup_test_data(db: Session):
|
|
yield
|
|
test_zongpai_nos = [
|
|
"26B999",
|
|
"26C999",
|
|
"26BW0999",
|
|
"26T999",
|
|
"26T998",
|
|
"26B998",
|
|
"26C998",
|
|
"26BW0011",
|
|
"26BW0012",
|
|
]
|
|
|
|
# Attachment tables (order: children before parents / before shared box)
|
|
from app.models.attachment import (
|
|
AttachmentCategory,
|
|
FinishedGoodsAttachment,
|
|
FinishedGoodsAttachmentItem,
|
|
FinishedGoodsAttachmentLocation,
|
|
FinishedGoodsBoxAttachmentItem,
|
|
)
|
|
|
|
db.query(FinishedGoodsBoxAttachmentItem).filter(
|
|
FinishedGoodsBoxAttachmentItem.zongpai_no.in_(test_zongpai_nos)
|
|
).delete(synchronize_session=False)
|
|
db.query(FinishedGoodsAttachmentLocation).filter(
|
|
FinishedGoodsAttachmentLocation.zongpai_no.in_(test_zongpai_nos)
|
|
).delete(synchronize_session=False)
|
|
db.query(FinishedGoodsAttachmentItem).filter(
|
|
FinishedGoodsAttachmentItem.zongpai_no.in_(test_zongpai_nos)
|
|
).delete(synchronize_session=False)
|
|
db.query(FinishedGoodsAttachment).filter(
|
|
FinishedGoodsAttachment.zongpai_no.in_(test_zongpai_nos)
|
|
).delete(synchronize_session=False)
|
|
db.query(AttachmentCategory).filter(
|
|
AttachmentCategory.name.like("TEST_%")
|
|
).delete(synchronize_session=False)
|
|
|
|
# Existing product cleanup (unchanged) — attachment box items for these boxes:
|
|
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(FinishedGoodsBoxAttachmentItem).filter(
|
|
FinishedGoodsBoxAttachmentItem.box_id.in_(test_box_ids)
|
|
).delete(synchronize_session=False)
|
|
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()
|