Coverage for tests / test_tutorial / test_code_structure / test_tutorial002.py: 100%

1import importlib 9 ctx1abcdefghi

2from dataclasses import dataclass 9 ctx1abcdefghi

3from types import ModuleType 9 ctx1abcdefghi

4 

5import pytest 9 ctx1abcdefghi

6from sqlmodel import create_engine 9 ctx1abcdefghi

7 

8from ...conftest import PrintMock, needs_py310 9 ctx1abcdefghi

9 

10expected_calls = [ 9 ctx1abcdefghi

11 [ 

12 "Created hero:", 

13 { 

14 "id": 1, 

15 "name": "Deadpond", 

16 "age": None, 

17 "secret_name": "Dive Wilson", 

18 "team_id": 1, 

19 }, 

20 ], 

21 [ 

22 "Hero's team:", 

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

24 ], 

25] 

26 

27 

28@dataclass 9 ctx1abcdefghi

29class Modules: 9 ctx1abcdefghi

30 app: ModuleType 9 ctx1abcdefghi

31 database: ModuleType 9 ctx1abcdefghi

32 

33 

34@pytest.fixture( 9 ctx1abcdefghi

35 name="modules", 

36 params=[ 

37 pytest.param("tutorial002_py39"), 

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

39 ], 

40) 

41def get_modules(request: pytest.FixtureRequest) -> Modules: 9 ctx1abcdefghi

42 app_module = importlib.import_module( 9 ctx1abcdefghi

43 f"docs_src.tutorial.code_structure.{request.param}.app" 

44 ) 

45 database_module = importlib.import_module( 9 ctx1abcdefghi

46 f"docs_src.tutorial.code_structure.{request.param}.database" 

47 ) 

48 database_module.sqlite_url = "sqlite://" 9 ctx1abcdefghi

49 database_module.engine = create_engine(database_module.sqlite_url) 9 ctx1abcdefghi

50 app_module.engine = database_module.engine 9 ctx1abcdefghi

51 

52 return Modules(app=app_module, database=database_module) 9 ctx1abcdefghi

53 

54 

55def test_tutorial(print_mock: PrintMock, modules: Modules): 9 ctx1abcdefghi

56 modules.app.main() 9 ctx1jklmnopqr

57 assert print_mock.calls == expected_calls 9 ctx1jklmnopqr