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

1from unittest.mock import patch 10 ctx1uvwxyzABCD

2 

3from sqlmodel import create_engine 10 ctx1uvwxyzABCD

4 

5from ...conftest import get_testing_print_function 10 ctx1uvwxyzABCD

6 

7expected_calls = [ 10 ctx1uvwxyzABCD

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 

29def test_tutorial001(clear_sqlmodel): 10 ctx1uvwxyzABCD

30 from docs_src.tutorial.update import tutorial001 as mod 10 ctx1abcdefghij

31 

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

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

34 calls = [] 10 ctx1abcdefghij

35 

36 new_print = get_testing_print_function(calls) 10 ctx1abcdefghij

37 

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

39 mod.main() 10 ctx1abcdefghij

40 assert calls == expected_calls 10 ctx1abcdefghij

41 

42 

43def test_tutorial002(clear_sqlmodel): 10 ctx1uvwxyzABCD

44 from docs_src.tutorial.update import tutorial002 as mod 10 ctx1klmnopqrst

45 

46 mod.sqlite_url = "sqlite://" 10 ctx1klmnopqrst

47 mod.engine = create_engine(mod.sqlite_url) 10 ctx1klmnopqrst

48 calls = [] 10 ctx1klmnopqrst

49 

50 new_print = get_testing_print_function(calls) 10 ctx1klmnopqrst

51 

52 with patch("builtins.print", new=new_print): 10 ctx1klmnopqrst

53 mod.main() 10 ctx1klmnopqrst

54 assert calls == expected_calls 10 ctx1klmnopqrst