Coverage for tests/test_tutorial/test_update/test_tutorial001_py310_tutorial002_py310.py: 100%

1from unittest.mock import patch 6 ctx1labcde

2 

3from sqlmodel import create_engine 6 ctx1labcde

4 

5from ...conftest import get_testing_print_function, needs_py310 6 ctx1labcde

6 

7expected_calls = [ 5 ctx1abcde

8 [ 

9 "Hero:", 

10 { 

11 "id": 2, 

12 "name": "Spider-Boy", 

13 "secret_name": "Pedro Parqueador", 

14 "age": None, 

15 }, 

16 ], 

17 [ 

18 "Updated hero:", 

19 { 

20 "id": 2, 

21 "name": "Spider-Boy", 

22 "secret_name": "Pedro Parqueador", 

23 "age": 16, 

24 }, 

25 ], 

26] 

27 

28 

29@needs_py310 6 ctx1labcde

30def test_tutorial001(clear_sqlmodel): 5 ctx1abcde

31 from docs_src.tutorial.update import tutorial001_py310 as mod 3 ctx1fgh

32 

33 mod.sqlite_url = "sqlite://" 3 ctx1fgh

34 mod.engine = create_engine(mod.sqlite_url) 3 ctx1fgh

35 calls = [] 3 ctx1fgh

36 

37 new_print = get_testing_print_function(calls) 3 ctx1fgh

38 

39 with patch("builtins.print", new=new_print): 3 ctx1fgh

40 mod.main() 3 ctx1fgh

41 assert calls == expected_calls 3 ctx1fgh

42 

43 

44@needs_py310 6 ctx1labcde

45def test_tutorial002(clear_sqlmodel): 5 ctx1abcde

46 from docs_src.tutorial.update import tutorial002_py310 as mod 3 ctx1ijk

47 

48 mod.sqlite_url = "sqlite://" 3 ctx1ijk

49 mod.engine = create_engine(mod.sqlite_url) 3 ctx1ijk

50 calls = [] 3 ctx1ijk

51 

52 new_print = get_testing_print_function(calls) 3 ctx1ijk

53 

54 with patch("builtins.print", new=new_print): 3 ctx1ijk

55 mod.main() 3 ctx1ijk

56 assert calls == expected_calls 3 ctx1ijk