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

1import importlib 9 ctx1abcdefghi

2from types import ModuleType 9 ctx1abcdefghi

3from typing import Any 9 ctx1abcdefghi

4 

5import pytest 9 ctx1abcdefghi

6from sqlalchemy import Engine 9 ctx1abcdefghi

7 

8from ...conftest import PrintMock 9 ctx1abcdefghi

9 

10 

11def check_calls(calls: list[list[str | dict[str, Any]]]): 9 ctx1abcdefghi

12 assert calls[0][0] == { 18 ctx1jklmnopqrstuvwxyzA

13 "name": "Deadpond", 

14 "secret_name": "Dive Wilson", 

15 "age": None, 

16 "id": 1, 

17 } 

18 assert calls[1][0] == { 18 ctx1jklmnopqrstuvwxyzA

19 "name": "Spider-Boy", 

20 "secret_name": "Pedro Parqueador", 

21 "age": None, 

22 "id": 2, 

23 } 

24 assert calls[2][0] == { 18 ctx1jklmnopqrstuvwxyzA

25 "name": "Rusty-Man", 

26 "secret_name": "Tommy Sharp", 

27 "age": 48, 

28 "id": 3, 

29 } 

30 

31 

32@pytest.fixture(name="module") 9 ctx1abcdefghi

33def get_module(request: pytest.FixtureRequest, database_engine: Engine) -> ModuleType: 9 ctx1abcdefghi

34 module = importlib.import_module(f"docs_src.tutorial.select.{request.param}") 9 ctx1abcdefghi

35 module.engine = database_engine 9 ctx1abcdefghi

36 return module 9 ctx1abcdefghi

37 

38 

39@pytest.mark.parametrize( 9 ctx1abcdefghi

40 "module", 

41 [ 

42 pytest.param("tutorial001_py310"), 

43 ], 

44 indirect=True, 

45) 

46def test_tutorial_001(print_mock: PrintMock, module: ModuleType): 9 ctx1abcdefghi

47 module.main() 9 ctx1jlnprtvxz

48 check_calls(print_mock.calls) 9 ctx1jlnprtvxz

49 

50 

51@pytest.mark.parametrize( 9 ctx1abcdefghi

52 "module", 

53 [ 

54 pytest.param("tutorial002_py310"), 

55 ], 

56 indirect=True, 

57) 

58def test_tutorial_002(print_mock: PrintMock, module: ModuleType): 9 ctx1abcdefghi

59 module.main() 9 ctx1kmoqsuwyA

60 check_calls(print_mock.calls) 9 ctx1kmoqsuwyA