mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
"""Shared pytest fixtures for the ``tests/`` tree.
|
|
|
|
Provides an ephemeral PostgreSQL engine for tests that exercise SQLModel
|
|
repositories. PostgreSQL has no true in-memory mode; ``pytest-postgresql``
|
|
starts a real, throwaway server in a temp directory (the process is started
|
|
once per session and a fresh database is created/dropped per test). That is
|
|
the closest equivalent to "in-memory" and matches production behaviour far
|
|
better than SQLite (enums, JSONB, constraint semantics, etc.).
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import glob
|
|
from collections.abc import Iterator
|
|
from typing import Any
|
|
|
|
import pytest
|
|
from psycopg import Connection
|
|
from pytest_postgresql import factories
|
|
from sqlalchemy import Engine
|
|
from sqlmodel import SQLModel, create_engine
|
|
|
|
# Importing the SQLModel row modules registers their tables on
|
|
# SQLModel.metadata so ``create_all`` builds the full schema. Imports look
|
|
# unused; they aren't.
|
|
|
|
|
|
# pg_ctl ships under a versioned path and is not on PATH in the dev container.
|
|
_PG_CTL = next(iter(sorted(glob.glob("/usr/lib/postgresql/*/bin/pg_ctl"))), "pg_ctl")
|
|
|
|
postgresql_proc = factories.postgresql_proc(
|
|
executable=_PG_CTL
|
|
) # pyright: ignore[reportUnknownMemberType]
|
|
postgresql = factories.postgresql("postgresql_proc")
|
|
|
|
|
|
@pytest.fixture
|
|
def db_engine(postgresql: Connection[Any]) -> Iterator[Engine]:
|
|
"""A SQLModel engine bound to a fresh, ephemeral PostgreSQL database."""
|
|
info = postgresql.info
|
|
url = f"postgresql+psycopg://{info.user}:@{info.host}:{info.port}/{info.dbname}"
|
|
engine = create_engine(url)
|
|
SQLModel.metadata.create_all(engine)
|
|
try:
|
|
yield engine
|
|
finally:
|
|
SQLModel.metadata.drop_all(engine)
|
|
engine.dispose()
|