infuanfu/teilchensammler-cli
2
0
Fork 0
Code Issues 3 Pull requests Releases 10 Packages 1 Activity

test: ensure we read expected data from database

Browse source
This commit is contained in:
bronsen 2026-02-22 18:08:39 +01:00
parent d712c7eead
commit d205f4909e
3 changed files with 86 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/teilchensammler_cli/models.py
View file

@ -9,7 +9,6 @@ from sqlmodel import (
select,
)
from .database import engine
logger = logging.getLogger(__name__)
@ -82,8 +81,7 @@ async def make_teilchen_input(text: str) -> TeilchenCreate | None:
return teilchen
async def load_initial_data() -> Sequence[Teilchen]:
from .database import engine
async def load_initial_data(engine) -> Sequence[Teilchen]:
with Session(engine) as session:
statement = select(
@ -93,7 +91,7 @@ async def load_initial_data() -> Sequence[Teilchen]:
return all_teilchen
async def add_to_database(tc: TeilchenCreate) -> Teilchen:
async def add_to_database(tc: TeilchenCreate, engine) -> Teilchen:
with Session(engine) as session:
teilchen = Teilchen.model_validate(tc)
session.add(teilchen)