You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
# backend/tests/test_seed_research_trip.py
|
|
from sqlmodel import select
|
|
|
|
from app.models import (
|
|
RagChunk,
|
|
ResearchChart,
|
|
ResearchCollection,
|
|
ResearchQA,
|
|
ResearchReport,
|
|
ResearchSource,
|
|
SavedTrip,
|
|
Trip,
|
|
TripChecklist,
|
|
TripDay,
|
|
TripPlan,
|
|
TripPrep,
|
|
TripRoute,
|
|
)
|
|
|
|
|
|
def test_research_seed_counts(client, session):
|
|
s, _ = session
|
|
assert len(s.exec(select(ResearchCollection)).all()) == 3
|
|
srcs = s.exec(select(ResearchSource)).all()
|
|
assert len(srcs) == 5 and sum(1 for x in srcs if x.learned) == 4
|
|
assert len(s.exec(select(ResearchReport)).all()) == 1
|
|
assert len(s.exec(select(ResearchQA)).all()) == 1
|
|
assert len(s.exec(select(ResearchChart)).all()) == 1
|
|
assert len(s.exec(select(RagChunk)).all()) >= 5 # 소스마다 1청크 이상
|
|
|
|
|
|
def test_trip_seed_counts(client, session):
|
|
s, _ = session
|
|
assert len(s.exec(select(Trip)).all()) == 1
|
|
assert len(s.exec(select(TripRoute)).all()) == 2
|
|
preps = s.exec(select(TripPrep)).all()
|
|
assert len(preps) == 5 and sum(1 for p in preps if p.state == "doing") == 1
|
|
assert len(s.exec(select(TripDay)).all()) == 2
|
|
assert len(s.exec(select(TripChecklist)).all()) == 7
|
|
assert len(s.exec(select(SavedTrip)).all()) == 3
|
|
assert len(s.exec(select(TripPlan)).all()) == 2
|