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

20 statements  

« prev     ^ index     » next       coverage.py v7.13.0, created at 2026-01-06 21:09 +0000

1import importlib 1abcdefghi

2from types import ModuleType 1abcdefghi

3 

4import pytest 1abcdefghi

5from sqlmodel import create_engine 1abcdefghi

6 

7from ...conftest import PrintMock, needs_py310 1abcdefghi

8 

9expected_calls = [ 1abcdefghi

10 [ 

11 "Hero 1:", 

12 {"id": 2, "name": "Spider-Boy", "secret_name": "Pedro Parqueador", "age": None}, 

13 ], 

14 [ 

15 "Hero 2:", 

16 { 

17 "id": 7, 

18 "name": "Captain North America", 

19 "secret_name": "Esteban Rogelios", 

20 "age": 93, 

21 }, 

22 ], 

23 [ 

24 "Updated hero 1:", 

25 { 

26 "id": 2, 

27 "name": "Spider-Youngster", 

28 "secret_name": "Pedro Parqueador", 

29 "age": 16, 

30 }, 

31 ], 

32 [ 

33 "Updated hero 2:", 

34 { 

35 "id": 7, 

36 "name": "Captain North America Except Canada", 

37 "secret_name": "Esteban Rogelios", 

38 "age": 110, 

39 }, 

40 ], 

41 [ 

42 "Hero: ", 

43 { 

44 "id": 2, 

45 "name": "Spider-Youngster", 

46 "secret_name": "Pedro Parqueador", 

47 "age": 16, 

48 }, 

49 ], 

50 [ 

51 "Deleted hero:", 

52 { 

53 "id": 2, 

54 "name": "Spider-Youngster", 

55 "secret_name": "Pedro Parqueador", 

56 "age": 16, 

57 }, 

58 ], 

59 ["There's no hero named Spider-Youngster"], 

60] 

61 

62 

63@pytest.fixture(name="module") 1abcdefghi

64def get_module(request: pytest.FixtureRequest) -> ModuleType: 1abcdefghi

65 module = importlib.import_module(f"docs_src.tutorial.delete.{request.param}") 1abcdefghi

66 module.sqlite_url = "sqlite://" 1abcdefghi

67 module.engine = create_engine(module.sqlite_url) 1abcdefghi

68 return module 1abcdefghi

69 

70 

71@pytest.mark.parametrize( 1abcdefghi

72 "module", 

73 [ 

74 "tutorial001_py39", 

75 pytest.param("tutorial001_py310", marks=needs_py310), 

76 ], 

77 indirect=True, 

78) 

79def test_tutorial001(print_mock: PrintMock, module: ModuleType): 1abcdefghi

80 module.main() 1jklmnopqr

81 assert print_mock.calls == expected_calls 1jklmnopqr

82 

83 

84@pytest.mark.parametrize( 1abcdefghi

85 "module", 

86 [ 

87 "tutorial002_py39", 

88 pytest.param("tutorial002_py310", marks=needs_py310), 

89 ], 

90 indirect=True, 

91) 

92def test_tutorial002(print_mock: PrintMock, module: ModuleType): 1abcdefghi

93 module.main() 1stuvwxyzA

94 assert print_mock.calls == expected_calls 1stuvwxyzA