mirror of
https://github.com/EChachati/SQLModel-CRUD-manager.git
synced 2025-08-14 19:10:18 +08:00
93 lines
2.3 KiB
Python
93 lines
2.3 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_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_list(last_id):
|
|
heroes = crud.list()
|
|
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_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(last_id)
|