Coverage for tests / test_tutorial / test_one / test_tutorial004.py: 100%

1import importlib 9 ctx1abcdefghi

2from types import ModuleType 9 ctx1abcdefghi

3 

4import pytest 9 ctx1abcdefghi

5from sqlalchemy import Engine 9 ctx1abcdefghi

6from sqlalchemy.exc import MultipleResultsFound 9 ctx1abcdefghi

7from sqlmodel import Session, delete 9 ctx1abcdefghi

8 

9from ...conftest import PrintMock 9 ctx1abcdefghi

10 

11 

12@pytest.fixture( 9 ctx1abcdefghi

13 name="mod", 

14 params=[ 

15 pytest.param("tutorial004_py310"), 

16 ], 

17) 

18def get_module(request: pytest.FixtureRequest, database_engine: Engine) -> ModuleType: 9 ctx1abcdefghi

19 mod = importlib.import_module(f"docs_src.tutorial.one.{request.param}") 9 ctx1abcdefghi

20 mod.engine = database_engine 9 ctx1abcdefghi

21 return mod 9 ctx1abcdefghi

22 

23 

24def test_tutorial(print_mock: PrintMock, mod: ModuleType): 9 ctx1abcdefghi

25 with pytest.raises(MultipleResultsFound): 9 ctx1jklmnopqr

26 mod.main() 9 ctx1jklmnopqr

27 with Session(mod.engine) as session: 9 ctx1jklmnopqr

28 # TODO: create delete() function 

29 # TODO: add overloads for .exec() with delete object 

30 session.exec(delete(mod.Hero)) 9 ctx1jklmnopqr

31 hero = mod.Hero(name="Test Hero", secret_name="Secret Test Hero", age=24) 9 ctx1jklmnopqr

32 session.add(hero) 9 ctx1jklmnopqr

33 session.commit() 9 ctx1jklmnopqr

34 session.refresh(hero) 9 ctx1jklmnopqr

35 hero_id = hero.id 9 ctx1jklmnopqr

36 assert hero_id is not None 9 ctx1jklmnopqr

37 

38 mod.select_heroes() 9 ctx1jklmnopqr

39 assert print_mock.calls == [ 9 ctx1jklmnopqr

40 [ 

41 "Hero:", 

42 { 

43 "id": hero_id, 

44 "name": "Test Hero", 

45 "secret_name": "Secret Test Hero", 

46 "age": 24, 

47 }, 

48 ] 

49 ]