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

1from unittest.mock import patch 10 ctx1klmnopqrst

2 

3import pytest 10 ctx1klmnopqrst

4from sqlalchemy.exc import MultipleResultsFound 10 ctx1klmnopqrst

5from sqlmodel import Session, create_engine, delete 10 ctx1klmnopqrst

6 

7from ...conftest import get_testing_print_function 10 ctx1klmnopqrst

8 

9 

10def test_tutorial(clear_sqlmodel): 10 ctx1klmnopqrst

11 from docs_src.tutorial.one import tutorial004 as mod 10 ctx1abcdefghij

12 

13 mod.sqlite_url = "sqlite://" 10 ctx1abcdefghij

14 mod.engine = create_engine(mod.sqlite_url) 10 ctx1abcdefghij

15 with pytest.raises(MultipleResultsFound): 10 ctx1abcdefghij

16 mod.main() 10 ctx1abcdefghij

17 with Session(mod.engine) as session: 10 ctx1abcdefghij

18 # TODO: create delete() function 

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

20 session.exec(delete(mod.Hero)) 10 ctx1abcdefghij

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

22 session.commit() 10 ctx1abcdefghij

23 

24 calls = [] 10 ctx1abcdefghij

25 

26 new_print = get_testing_print_function(calls) 10 ctx1abcdefghij

27 

28 with patch("builtins.print", new=new_print): 10 ctx1abcdefghij

29 mod.select_heroes() 10 ctx1abcdefghij

30 assert calls == [ 10 ctx1abcdefghij

31 [ 

32 "Hero:", 

33 { 

34 "id": 1, 

35 "name": "Test Hero", 

36 "secret_name": "Secret Test Hero", 

37 "age": 24, 

38 }, 

39 ] 

40 ]