Coverage for tests/test_tutorial/test_select/test_tutorial003_py310_tutorial004_py310.py: 100%

1from typing import Any, Dict, List, Union 6 ctx1abcdef

2from unittest.mock import patch 6 ctx1abcdef

3 

4from sqlmodel import create_engine 6 ctx1abcdef

5 

6from ...conftest import get_testing_print_function, needs_py310 6 ctx1abcdef

7 

8 

9def check_calls(calls: List[List[Union[str, Dict[str, Any]]]]): 6 ctx1abcdef

10 assert calls[0][0] == [ 8 ctx1ghijklmn

11 { 

12 "name": "Deadpond", 

13 "secret_name": "Dive Wilson", 

14 "age": None, 

15 "id": 1, 

16 }, 

17 { 

18 "name": "Spider-Boy", 

19 "secret_name": "Pedro Parqueador", 

20 "age": None, 

21 "id": 2, 

22 }, 

23 { 

24 "name": "Rusty-Man", 

25 "secret_name": "Tommy Sharp", 

26 "age": 48, 

27 "id": 3, 

28 }, 

29 ] 

30 

31 

32@needs_py310 6 ctx1abcdef

33def test_tutorial_003(clear_sqlmodel): 6 ctx1abcdef

34 from docs_src.tutorial.select import tutorial003_py310 as mod 4 ctx1hjln

35 

36 mod.sqlite_url = "sqlite://" 4 ctx1hjln

37 mod.engine = create_engine(mod.sqlite_url) 4 ctx1hjln

38 calls = [] 4 ctx1hjln

39 

40 new_print = get_testing_print_function(calls) 4 ctx1hjln

41 

42 with patch("builtins.print", new=new_print): 4 ctx1hjln

43 mod.main() 4 ctx1hjln

44 check_calls(calls) 4 ctx1hjln

45 

46 

47@needs_py310 6 ctx1abcdef

48def test_tutorial_002(clear_sqlmodel): 6 ctx1abcdef

49 from docs_src.tutorial.select import tutorial004_py310 as mod 4 ctx1gikm

50 

51 mod.sqlite_url = "sqlite://" 4 ctx1gikm

52 mod.engine = create_engine(mod.sqlite_url) 4 ctx1gikm

53 calls = [] 4 ctx1gikm

54 

55 new_print = get_testing_print_function(calls) 4 ctx1gikm

56 

57 with patch("builtins.print", new=new_print): 4 ctx1gikm

58 mod.main() 4 ctx1gikm

59 check_calls(calls) 4 ctx1gikm