infuanfu/teilchensammler-cli
2
0
Fork 0
Code Issues 3 Pull requests Releases 9 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/main.py
View file

@ -12,7 +12,7 @@ from textual.screen import Screen
from textual.widget import Widget
from textual.widgets import Button, DataTable, Footer, Header, Input, Static
from .database import create_db_and_tables
from .database import create_db_and_tables, engine
from .models import add_to_database, load_initial_data, make_teilchen_input
logging.basicConfig(
@ -68,7 +68,7 @@ class SearchBar(Static):
event.input.value = ""
teilchen = await add_to_database(tc)
teilchen = await add_to_database(tc, engine)
self.screen.query_one(DataTable).add_row(
teilchen.id, teilchen.name, teilchen.description, teilchen.number, teilchen.tags
)
@ -81,7 +81,7 @@ class SearchResults(Widget):
async def on_mount(self) -> None:
table: DataTable = self.query_one(DataTable)
table.add_columns(*TEILCHEN_DATA_HEADER)
table.add_rows(await load_initial_data())
table.add_rows(await load_initial_data(engine))
class AddInventoryScreen(Screen):

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)