Coverage for tests/test_tutorial/test_connect/test_select/test_tutorial001_tutorial002.py: 100%

1from unittest.mock import patch 6 ctx1rmnopq

2 

3from sqlmodel import create_engine 6 ctx1rmnopq

4 

5from ....conftest import get_testing_print_function 6 ctx1rmnopq

6 

7expected_calls = [ 5 ctx1mnopq

8 [ 

9 "Created hero:", 

10 { 

11 "age": None, 

12 "id": 1, 

13 "secret_name": "Dive Wilson", 

14 "team_id": 2, 

15 "name": "Deadpond", 

16 }, 

17 ], 

18 [ 

19 "Created hero:", 

20 { 

21 "age": 48, 

22 "id": 2, 

23 "secret_name": "Tommy Sharp", 

24 "team_id": 1, 

25 "name": "Rusty-Man", 

26 }, 

27 ], 

28 [ 

29 "Created hero:", 

30 { 

31 "age": None, 

32 "id": 3, 

33 "secret_name": "Pedro Parqueador", 

34 "team_id": None, 

35 "name": "Spider-Boy", 

36 }, 

37 ], 

38 [ 

39 "Hero:", 

40 { 

41 "age": None, 

42 "id": 1, 

43 "secret_name": "Dive Wilson", 

44 "team_id": 2, 

45 "name": "Deadpond", 

46 }, 

47 "Team:", 

48 {"id": 2, "name": "Z-Force", "headquarters": "Sister Margaret's Bar"}, 

49 ], 

50 [ 

51 "Hero:", 

52 { 

53 "age": 48, 

54 "id": 2, 

55 "secret_name": "Tommy Sharp", 

56 "team_id": 1, 

57 "name": "Rusty-Man", 

58 }, 

59 "Team:", 

60 {"id": 1, "name": "Preventers", "headquarters": "Sharp Tower"}, 

61 ], 

62] 

63 

64 

65def test_tutorial001(clear_sqlmodel): 6 ctx1rmnopq

66 from docs_src.tutorial.connect.select import tutorial001 as mod 6 ctx1abcdef

67 

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

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

70 calls = [] 6 ctx1abcdef

71 

72 new_print = get_testing_print_function(calls) 6 ctx1abcdef

73 

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

75 mod.main() 6 ctx1abcdef

76 assert calls == expected_calls 6 ctx1abcdef

77 

78 

79def test_tutorial002(clear_sqlmodel): 6 ctx1rmnopq

80 from docs_src.tutorial.connect.select import tutorial002 as mod 6 ctx1ghijkl

81 

82 mod.sqlite_url = "sqlite://" 6 ctx1ghijkl

83 mod.engine = create_engine(mod.sqlite_url) 6 ctx1ghijkl

84 calls = [] 6 ctx1ghijkl

85 

86 new_print = get_testing_print_function(calls) 6 ctx1ghijkl

87 

88 with patch("builtins.print", new=new_print): 6 ctx1ghijkl

89 mod.main() 6 ctx1ghijkl

90 assert calls == expected_calls 6 ctx1ghijkl