30 lines
681 B
Python
30 lines
681 B
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 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"]
|
|
db.query(FinishedGoodsLocation).filter(
|
|
FinishedGoodsLocation.zongpai_no.in_(test_zongpai_nos)
|
|
).delete(synchronize_session=False)
|
|
db.commit()
|