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

1from unittest.mock import patch 7 ctx1hijklmn

2 

3import pytest 7 ctx1hijklmn

4from sqlalchemy.exc import NoResultFound 7 ctx1hijklmn

5from sqlmodel import Session, create_engine, delete 7 ctx1hijklmn

6 

7from ...conftest import get_testing_print_function 7 ctx1hijklmn

8 

9 

10def test_tutorial(clear_sqlmodel): 7 ctx1hijklmn

11 from docs_src.tutorial.one import tutorial005 as mod 7 ctx1abcdefg

12 

13 mod.sqlite_url = "sqlite://" 7 ctx1abcdefg

14 mod.engine = create_engine(mod.sqlite_url) 7 ctx1abcdefg

15 with pytest.raises(NoResultFound): 7 ctx1abcdefg

16 mod.main() 7 ctx1abcdefg

17 with Session(mod.engine) as session: 7 ctx1abcdefg

18 # TODO: create delete() function 

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

20 session.exec(delete(mod.Hero)) 7 ctx1abcdefg

21 session.add(mod.Hero(name="Test Hero", secret_name="Secret Test Hero", age=24)) 7 ctx1abcdefg

22 session.commit() 7 ctx1abcdefg

23 

24 calls = [] 7 ctx1abcdefg

25 

26 new_print = get_testing_print_function(calls) 7 ctx1abcdefg

27 

28 with patch("builtins.print", new=new_print): 7 ctx1abcdefg

29 mod.select_heroes() 7 ctx1abcdefg

30 assert calls == [ 7 ctx1abcdefg

31 [ 

32 "Hero:", 

33 { 

34 "id": 1, 

35 "name": "Test Hero", 

36 "secret_name": "Secret Test Hero", 

37 "age": 24, 

38 }, 

39 ] 

40 ]