mirror of
https://github.com/fastapi/sqlmodel.git
synced 2025-11-03 05:41:42 +08:00
✅ Add tests
This commit is contained in:
27
tests/test_tutorial/test_insert/test_tutorial003.py
Normal file
27
tests/test_tutorial/test_insert/test_tutorial003.py
Normal file
@ -0,0 +1,27 @@
|
||||
from sqlmodel import create_engine, Session, select
|
||||
|
||||
|
||||
def test_tutorial(clear_sqlmodel):
|
||||
from docs_src.tutorial.insert import tutorial003 as mod
|
||||
|
||||
mod.sqlite_url = "sqlite://"
|
||||
mod.engine = create_engine(mod.sqlite_url)
|
||||
mod.main()
|
||||
with Session(mod.engine) as session:
|
||||
heroes = session.exec(select(mod.Hero)).all()
|
||||
heroes_by_name = {hero.name: hero for hero in heroes}
|
||||
deadpond = heroes_by_name["Deadpond"]
|
||||
spider_boy = heroes_by_name["Spider-Boy"]
|
||||
rusty_man = heroes_by_name["Rusty-Man"]
|
||||
assert deadpond.name == "Deadpond"
|
||||
assert deadpond.age is None
|
||||
assert deadpond.id is not None
|
||||
assert deadpond.secret_name == "Dive Wilson"
|
||||
assert spider_boy.name == "Spider-Boy"
|
||||
assert spider_boy.age is None
|
||||
assert spider_boy.id is not None
|
||||
assert spider_boy.secret_name == "Pedro Parqueador"
|
||||
assert rusty_man.name == "Rusty-Man"
|
||||
assert rusty_man.age == 48
|
||||
assert rusty_man.id is not None
|
||||
assert rusty_man.secret_name == "Tommy Sharp"
|
||||
Reference in New Issue
Block a user