mirror of
https://github.com/PaiGramTeam/sqlmodel.git
synced 2024-11-29 18:38:58 +00:00
28 lines
1013 B
Python
28 lines
1013 B
Python
|
from sqlmodel import create_engine, Session, select
|
||
|
|
|
||
|
|
|
||
|
|
def test_tutorial(clear_sqlmodel):
|
||
|
|
from docs_src.tutorial.insert import tutorial002 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"
|