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

1from unittest.mock import patch 10 ctx1ijklmnopqr

2 

3import pytest 10 ctx1ijklmnopqr

4from sqlalchemy.exc import NoResultFound 10 ctx1ijklmnopqr

5from sqlmodel import Session, create_engine, delete 10 ctx1ijklmnopqr

6 

7from ...conftest import get_testing_print_function, needs_py310 10 ctx1ijklmnopqr

8 

9 

10@needs_py310 10 ctx1ijklmnopqr

11def test_tutorial(clear_sqlmodel): 10 ctx1ijklmnopqr

12 from docs_src.tutorial.one import tutorial005_py310 as mod 8 ctx1abcdefgh

13 

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

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

16 with pytest.raises(NoResultFound): 8 ctx1abcdefgh

17 mod.main() 8 ctx1abcdefgh

18 with Session(mod.engine) as session: 8 ctx1abcdefgh

19 # TODO: create delete() function 

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

21 session.exec(delete(mod.Hero)) 8 ctx1abcdefgh

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

23 session.commit() 8 ctx1abcdefgh

24 

25 calls = [] 8 ctx1abcdefgh

26 

27 new_print = get_testing_print_function(calls) 8 ctx1abcdefgh

28 

29 with patch("builtins.print", new=new_print): 8 ctx1abcdefgh

30 mod.select_heroes() 8 ctx1abcdefgh

31 assert calls == [ 8 ctx1abcdefgh

32 [ 

33 "Hero:", 

34 { 

35 "id": 1, 

36 "name": "Test Hero", 

37 "secret_name": "Secret Test Hero", 

38 "age": 24, 

39 }, 

40 ] 

41 ]