mealie/tests/fixtures/fixture_database.py
2024-08-22 10:14:32 -05:00

22 lines
587 B
Python

from collections.abc import Generator
import pytest
from sqlalchemy.orm import Session, sessionmaker
from mealie.db.db_setup import SessionLocal
from mealie.repos.all_repositories import AllRepositories, get_repositories
@pytest.fixture(scope="module")
def session() -> Generator[sessionmaker[Session], None, None]:
try:
sess = SessionLocal()
yield sess
finally:
sess.close()
@pytest.fixture()
def unfiltered_database(session: Session) -> Generator[AllRepositories, None, None]:
yield get_repositories(session, group_id=None, household_id=None)