Coverage for tests/test_tutorial/test_limit_and_offset/test_tutorial002_py310.py: 100%

1from unittest.mock import patch 6 ctx1idefgh

2 

3from sqlmodel import create_engine 6 ctx1idefgh

4 

5from ...conftest import get_testing_print_function, needs_py310 6 ctx1idefgh

6 

7expected_calls = [ 5 ctx1defgh

8 [ 

9 [ 

10 { 

11 "id": 4, 

12 "name": "Tarantula", 

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

14 "age": 32, 

15 }, 

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

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

18 ] 

19 ] 

20] 

21 

22 

23@needs_py310 6 ctx1idefgh

24def test_tutorial(clear_sqlmodel): 5 ctx1defgh

25 from docs_src.tutorial.offset_and_limit import tutorial002_py310 as mod 3 ctx1abc

26 

27 mod.sqlite_url = "sqlite://" 3 ctx1abc

28 mod.engine = create_engine(mod.sqlite_url) 3 ctx1abc

29 calls = [] 3 ctx1abc

30 

31 new_print = get_testing_print_function(calls) 3 ctx1abc

32 

33 with patch("builtins.print", new=new_print): 3 ctx1abc

34 mod.main() 3 ctx1abc

35 assert calls == expected_calls 3 ctx1abc