mirror of
https://github.com/EChachati/SQLModel-CRUD-manager.git
synced 2025-08-14 19:10:18 +08:00
143 lines
3.9 KiB
Python
143 lines
3.9 KiB
Python
import pytest
|
|
from fastapi import HTTPException
|
|
from sqlmodel import Field, Session, SQLModel, create_engine, select
|
|
|
|
from sqlmodel_crud_manager.crud import CRUDManager
|
|
|
|
|
|
class HeroCreate(SQLModel):
|
|
name: str
|
|
secret_name: str
|
|
age: int | None = None
|
|
is_alive: bool = True
|
|
|
|
|
|
class Hero(HeroCreate, table=True):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
|
|
|
|
SECRET_NAME = "Dive Wilson"
|
|
HERO_NAME = "Deadpond"
|
|
|
|
|
|
engine = create_engine("sqlite:///tests/testing.db")
|
|
SQLModel.metadata.create_all(engine)
|
|
crud = CRUDManager(Hero, engine)
|
|
|
|
|
|
@pytest.fixture(name="last_id")
|
|
def get_last_id_fixture():
|
|
return Session(engine).exec(select(Hero).order_by(Hero.id.desc())).first().id
|
|
|
|
|
|
def test_create():
|
|
hero = HeroCreate(name=HERO_NAME, secret_name=SECRET_NAME)
|
|
hero = crud.create(hero)
|
|
assert hero.id is not None
|
|
assert hero.name == HERO_NAME
|
|
assert hero.secret_name == SECRET_NAME
|
|
assert hero.age is None
|
|
assert hero.is_alive is True
|
|
|
|
|
|
def test_get(last_id):
|
|
hero = crud.get(last_id)
|
|
assert hero.id == last_id
|
|
assert hero.name == HERO_NAME
|
|
assert hero.secret_name == SECRET_NAME
|
|
assert hero.age is None
|
|
assert hero.is_alive is True
|
|
|
|
|
|
def test_get_or_404(last_id):
|
|
hero = crud.get_or_404(last_id)
|
|
assert hero.id == last_id
|
|
assert hero.name == HERO_NAME
|
|
assert hero.secret_name == SECRET_NAME
|
|
assert hero.age is None
|
|
assert hero.is_alive is True
|
|
|
|
with pytest.raises(HTTPException):
|
|
crud.get_or_404(last_id + 1)
|
|
|
|
|
|
def test_get_by_ids(last_id):
|
|
heroes = crud.get_by_ids([last_id])
|
|
assert len(heroes) == 1
|
|
assert heroes[0].id == last_id
|
|
assert heroes[0].name == HERO_NAME
|
|
assert heroes[0].secret_name == SECRET_NAME
|
|
assert heroes[0].age is None
|
|
assert heroes[0].is_alive is True
|
|
|
|
|
|
def test_get_by_field(last_id):
|
|
heroes = crud.get_by_field("name", HERO_NAME)
|
|
assert heroes.id == last_id
|
|
assert heroes.name == HERO_NAME
|
|
assert heroes.secret_name == SECRET_NAME
|
|
assert heroes.age is None
|
|
assert heroes.is_alive is True
|
|
|
|
|
|
def test_get_by_fields(last_id):
|
|
heroes = crud.get_by_fields({"name": HERO_NAME, "secret_name": SECRET_NAME})
|
|
assert heroes.id == last_id
|
|
assert heroes.name == HERO_NAME
|
|
assert heroes.secret_name == SECRET_NAME
|
|
assert heroes.age is None
|
|
assert heroes.is_alive is True
|
|
|
|
|
|
def test_get_or_create(last_id):
|
|
hero = HeroCreate(name=HERO_NAME, secret_name=SECRET_NAME)
|
|
hero = crud.get_or_create(hero, search_field="name")
|
|
assert hero.id is not None
|
|
assert hero.id == last_id
|
|
assert hero.name == HERO_NAME
|
|
|
|
hero = HeroCreate(name="NotExistent", secret_name=SECRET_NAME)
|
|
hero = crud.get_or_create(hero, search_field="name")
|
|
assert hero.id is not None
|
|
assert hero.id != last_id
|
|
assert hero.name == "NotExistent"
|
|
|
|
hero = HeroCreate(name="NotExistentV2", secret_name=SECRET_NAME)
|
|
with pytest.raises(HTTPException):
|
|
crud.get_or_create(hero, search_field="id")
|
|
with pytest.raises(HTTPException):
|
|
crud.get_or_create(hero, search_field="NotExistent")
|
|
|
|
|
|
def test_list(last_id):
|
|
heroes = crud.list()
|
|
assert len(heroes) == 2
|
|
assert heroes[-1].id == last_id
|
|
assert heroes[0].name == HERO_NAME
|
|
assert heroes[0].secret_name == SECRET_NAME
|
|
assert heroes[0].age is None
|
|
assert heroes[0].is_alive is True
|
|
|
|
|
|
def test_update(last_id):
|
|
hero = crud.get(last_id)
|
|
hero.name = HERO_NAME
|
|
hero.age = 30
|
|
crud.update(hero)
|
|
assert hero.id == last_id
|
|
assert hero.name == HERO_NAME
|
|
assert hero.secret_name == SECRET_NAME
|
|
assert hero.age == 30
|
|
assert hero.is_alive is True
|
|
|
|
|
|
def test_delete(last_id):
|
|
hero = crud.delete(last_id)
|
|
assert hero.id == last_id
|
|
assert hero.name == HERO_NAME
|
|
assert hero.secret_name == SECRET_NAME
|
|
assert hero.age == 30
|
|
assert hero.is_alive is True
|
|
with pytest.raises(HTTPException):
|
|
crud.get_or_404(last_id)
|