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

1import importlib 8 ctx1abcdefgh

2from types import ModuleType 8 ctx1abcdefgh

3 

4import pytest 8 ctx1abcdefgh

5from sqlalchemy.exc import MultipleResultsFound 8 ctx1abcdefgh

6from sqlmodel import Session, create_engine, delete 8 ctx1abcdefgh

7 

8from ...conftest import PrintMock 8 ctx1abcdefgh

9 

10 

11@pytest.fixture( 8 ctx1abcdefgh

12 name="mod", 

13 params=[ 

14 pytest.param("tutorial004_py310"), 

15 ], 

16) 

17def get_module(request: pytest.FixtureRequest) -> ModuleType: 8 ctx1abcdefgh

18 mod = importlib.import_module(f"docs_src.tutorial.one.{request.param}") 8 ctx1abcdefgh

19 mod.sqlite_url = "sqlite://" 8 ctx1abcdefgh

20 mod.engine = create_engine(mod.sqlite_url) 8 ctx1abcdefgh

21 return mod 8 ctx1abcdefgh

22 

23 

24def test_tutorial(print_mock: PrintMock, mod: ModuleType): 8 ctx1abcdefgh

25 with pytest.raises(MultipleResultsFound): 8 ctx1ijklmnop

26 mod.main() 8 ctx1ijklmnop

27 with Session(mod.engine) as session: 8 ctx1ijklmnop

28 # TODO: create delete() function 

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

30 session.exec(delete(mod.Hero)) 8 ctx1ijklmnop

31 session.add(mod.Hero(name="Test Hero", secret_name="Secret Test Hero", age=24)) 8 ctx1ijklmnop

32 session.commit() 8 ctx1ijklmnop

33 

34 mod.select_heroes() 8 ctx1ijklmnop

35 assert print_mock.calls == [ 8 ctx1ijklmnop

36 [ 

37 "Hero:", 

38 { 

39 "id": 1, 

40 "name": "Test Hero", 

41 "secret_name": "Secret Test Hero", 

42 "age": 24, 

43 }, 

44 ] 

45 ]