PGLite + FastAPI

1.txt

# ติดตั้ง
$pip install fastapi
$pip install py-pglite
$pip install sqlmodel

$pip install pytest
$pip install coverage


# Run test and coverage
$coverage run -m pytest
$coverage report -m

test.py

from fastapi.testclient import TestClient
from sqlmodel import SQLModel, Field, Session
from api import app

class User(SQLModel, table=True):
    __table_args__ = {"extend_existing": True}
    id: int | None = Field(default=None, primary_key=True)
    name: str

client = TestClient(app)

def test_api_success_to_get_user(pglite_session: Session):
    # Create a user
    user = User(name="Alice")
    pglite_session.add(user)
    pglite_session.commit()
    pglite_session.refresh(user)

    response = client.get(f"/users/{user.id}")
    assert response.status_code == 200
    assert response.json() == {"id": user.id, "name": "Alice"}