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

1from unittest.mock import patch 6 ctx1ghijkl

2 

3import pytest 6 ctx1ghijkl

4from sqlalchemy.exc import MultipleResultsFound 6 ctx1ghijkl

5from sqlmodel import Session, create_engine, delete 6 ctx1ghijkl

6 

7from ...conftest import get_testing_print_function 6 ctx1ghijkl

8 

9 

10def test_tutorial(clear_sqlmodel): 6 ctx1ghijkl

11 from docs_src.tutorial.one import tutorial004 as mod 6 ctx1abcdef

12 

13 mod.sqlite_url = "sqlite://" 6 ctx1abcdef

14 mod.engine = create_engine(mod.sqlite_url) 6 ctx1abcdef

15 with pytest.raises(MultipleResultsFound): 6 ctx1abcdef

16 mod.main() 6 ctx1abcdef

17 with Session(mod.engine) as session: 6 ctx1abcdef

18 # TODO: create delete() function 

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

20 session.exec(delete(mod.Hero)) 6 ctx1abcdef

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

22 session.commit() 6 ctx1abcdef

23 

24 calls = [] 6 ctx1abcdef

25 

26 new_print = get_testing_print_function(calls) 6 ctx1abcdef

27 

28 with patch("builtins.print", new=new_print): 6 ctx1abcdef

29 mod.select_heroes() 6 ctx1abcdef

30 assert calls == [ 6 ctx1abcdef

31 [ 

32 "Hero:", 

33 { 

34 "id": 1, 

35 "name": "Test Hero", 

36 "secret_name": "Secret Test Hero", 

37 "age": 24, 

38 }, 

39 ] 

40 ]