Coverage for tests / test_tutorial / test_many_to_many / test_tutorial001.py: 100%

1import importlib 9 ctx1abcdefghi

2from types import ModuleType 9 ctx1abcdefghi

3 

4import pytest 9 ctx1abcdefghi

5from sqlmodel import create_engine 9 ctx1abcdefghi

6 

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

8 

9 

10@pytest.fixture( 9 ctx1abcdefghi

11 name="mod", 

12 params=[ 

13 pytest.param("tutorial001_py39"), 

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

15 ], 

16) 

17def get_module(request: pytest.FixtureRequest) -> ModuleType: 9 ctx1abcdefghi

18 mod = importlib.import_module(f"docs_src.tutorial.many_to_many.{request.param}") 9 ctx1abcdefghi

19 mod.sqlite_url = "sqlite://" 9 ctx1abcdefghi

20 mod.engine = create_engine(mod.sqlite_url) 9 ctx1abcdefghi

21 return mod 9 ctx1abcdefghi

22 

23 

24expected_calls = [ 9 ctx1abcdefghi

25 [ 

26 "Deadpond:", 

27 {"id": 1, "secret_name": "Dive Wilson", "age": None, "name": "Deadpond"}, 

28 ], 

29 [ 

30 "Deadpond teams:", 

31 [ 

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

33 {"id": 2, "name": "Preventers", "headquarters": "Sharp Tower"}, 

34 ], 

35 ], 

36 [ 

37 "Rusty-Man:", 

38 {"id": 2, "secret_name": "Tommy Sharp", "age": 48, "name": "Rusty-Man"}, 

39 ], 

40 [ 

41 "Rusty-Man Teams:", 

42 [{"id": 2, "name": "Preventers", "headquarters": "Sharp Tower"}], 

43 ], 

44 [ 

45 "Spider-Boy:", 

46 {"id": 3, "secret_name": "Pedro Parqueador", "age": None, "name": "Spider-Boy"}, 

47 ], 

48 [ 

49 "Spider-Boy Teams:", 

50 [{"id": 2, "name": "Preventers", "headquarters": "Sharp Tower"}], 

51 ], 

52] 

53 

54 

55def test_tutorial(print_mock: PrintMock, mod: ModuleType): 9 ctx1abcdefghi

56 mod.main() 9 ctx1jklmnopqr

57 assert print_mock.calls == expected_calls 9 ctx1jklmnopqr