Coverage for tests / test_tutorial / test_insert / test_tutorial003.py: 100%

1import importlib 9 ctx1jklmnopqr

2from types import ModuleType 9 ctx1jklmnopqr

3 

4import pytest 9 ctx1jklmnopqr

5from sqlmodel import Session, create_engine, select 9 ctx1jklmnopqr

6 

7from ...conftest import needs_py310 9 ctx1jklmnopqr

8 

9 

10@pytest.fixture( 9 ctx1jklmnopqr

11 name="mod", 

12 params=[ 

13 pytest.param("tutorial003_py39"), 

14 pytest.param("tutorial003_py310", marks=needs_py310), 

15 ], 

16) 

17def get_module(request: pytest.FixtureRequest) -> ModuleType: 9 ctx1jklmnopqr

18 mod = importlib.import_module(f"docs_src.tutorial.insert.{request.param}") 9 ctx1jklmnopqr

19 mod.sqlite_url = "sqlite://" 9 ctx1jklmnopqr

20 mod.engine = create_engine(mod.sqlite_url) 9 ctx1jklmnopqr

21 return mod 9 ctx1jklmnopqr

22 

23 

24def test_tutorial(mod: ModuleType): 9 ctx1jklmnopqr

25 mod.main() 9 ctx1abcdefghi

26 with Session(mod.engine) as session: 9 ctx1abcdefghi

27 heroes = session.exec(select(mod.Hero)).all() 9 ctx1abcdefghi

28 heroes_by_name = {hero.name: hero for hero in heroes} 9 ctx1abcdefghi

29 deadpond = heroes_by_name["Deadpond"] 9 ctx1abcdefghi

30 spider_boy = heroes_by_name["Spider-Boy"] 9 ctx1abcdefghi

31 rusty_man = heroes_by_name["Rusty-Man"] 9 ctx1abcdefghi

32 assert deadpond.name == "Deadpond" 9 ctx1abcdefghi

33 assert deadpond.age is None 9 ctx1abcdefghi

34 assert deadpond.id is not None 9 ctx1abcdefghi

35 assert deadpond.secret_name == "Dive Wilson" 9 ctx1abcdefghi

36 assert spider_boy.name == "Spider-Boy" 9 ctx1abcdefghi

37 assert spider_boy.age is None 9 ctx1abcdefghi

38 assert spider_boy.id is not None 9 ctx1abcdefghi

39 assert spider_boy.secret_name == "Pedro Parqueador" 9 ctx1abcdefghi

40 assert rusty_man.name == "Rusty-Man" 9 ctx1abcdefghi

41 assert rusty_man.age == 48 9 ctx1abcdefghi

42 assert rusty_man.id is not None 9 ctx1abcdefghi

43 assert rusty_man.secret_name == "Tommy Sharp" 9 ctx1abcdefghi