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

1from typing import Any, Dict, List, Union 10 ctx1qrstuvwxyz

2from unittest.mock import patch 10 ctx1qrstuvwxyz

3 

4from sqlmodel import create_engine 10 ctx1qrstuvwxyz

5 

6from ...conftest import get_testing_print_function, needs_py310 10 ctx1qrstuvwxyz

7 

8 

9def check_calls(calls: List[List[Union[str, Dict[str, Any]]]]): 10 ctx1qrstuvwxyz

10 assert calls[0][0] == { 16 ctx1abcdefghijklmnop

11 "name": "Deadpond", 

12 "secret_name": "Dive Wilson", 

13 "age": None, 

14 "id": 1, 

15 } 

16 assert calls[1][0] == { 16 ctx1abcdefghijklmnop

17 "name": "Spider-Boy", 

18 "secret_name": "Pedro Parqueador", 

19 "age": None, 

20 "id": 2, 

21 } 

22 assert calls[2][0] == { 16 ctx1abcdefghijklmnop

23 "name": "Rusty-Man", 

24 "secret_name": "Tommy Sharp", 

25 "age": 48, 

26 "id": 3, 

27 } 

28 

29 

30@needs_py310 10 ctx1qrstuvwxyz

31def test_tutorial_001(clear_sqlmodel): 10 ctx1qrstuvwxyz

32 from docs_src.tutorial.select import tutorial001_py310 as mod 8 ctx1acegikmo

33 

34 mod.sqlite_url = "sqlite://" 8 ctx1acegikmo

35 mod.engine = create_engine(mod.sqlite_url) 8 ctx1acegikmo

36 calls = [] 8 ctx1acegikmo

37 

38 new_print = get_testing_print_function(calls) 8 ctx1acegikmo

39 

40 with patch("builtins.print", new=new_print): 8 ctx1acegikmo

41 mod.main() 8 ctx1acegikmo

42 check_calls(calls) 8 ctx1acegikmo

43 

44 

45@needs_py310 10 ctx1qrstuvwxyz

46def test_tutorial_002(clear_sqlmodel): 10 ctx1qrstuvwxyz

47 from docs_src.tutorial.select import tutorial002_py310 as mod 8 ctx1bdfhjlnp

48 

49 mod.sqlite_url = "sqlite://" 8 ctx1bdfhjlnp

50 mod.engine = create_engine(mod.sqlite_url) 8 ctx1bdfhjlnp

51 calls = [] 8 ctx1bdfhjlnp

52 

53 new_print = get_testing_print_function(calls) 8 ctx1bdfhjlnp

54 

55 with patch("builtins.print", new=new_print): 8 ctx1bdfhjlnp

56 mod.main() 8 ctx1bdfhjlnp

57 check_calls(calls) 8 ctx1bdfhjlnp