test: add integration tests for location API

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Misaka_Company
2026-05-11 11:16:33 +08:00
parent d68d602411
commit 6a4ea34e24
3 changed files with 86 additions and 0 deletions

29
tests/conftest.py Normal file
View File

@@ -0,0 +1,29 @@
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()