Coverage for tests / test_tutorial / test_limit_and_offset / test_tutorial002.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("tutorial002_py39"), 

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

15 ], 

16) 

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

18 mod = importlib.import_module(f"docs_src.tutorial.offset_and_limit.{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 [ 

27 { 

28 "id": 4, 

29 "name": "Tarantula", 

30 "secret_name": "Natalia Roman-on", 

31 "age": 32, 

32 }, 

33 {"id": 5, "name": "Black Lion", "secret_name": "Trevor Challa", "age": 35}, 

34 {"id": 6, "name": "Dr. Weird", "secret_name": "Steve Weird", "age": 36}, 

35 ] 

36 ] 

37] 

38 

39 

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

41 mod.main() 9 ctx1jklmnopqr

42 assert print_mock.calls == expected_calls 9 ctx1jklmnopqr