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

1from unittest.mock import patch 7 ctx1nhijklm

2 

3from sqlmodel import create_engine 7 ctx1nhijklm

4 

5from ...conftest import get_testing_print_function 7 ctx1nhijklm

6 

7expected_calls = [ 6 ctx1hijklm

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 

23def test_tutorial(clear_sqlmodel): 7 ctx1nhijklm

24 from docs_src.tutorial.offset_and_limit import tutorial002 as mod 7 ctx1abcdefg

25 

26 mod.sqlite_url = "sqlite://" 7 ctx1abcdefg

27 mod.engine = create_engine(mod.sqlite_url) 7 ctx1abcdefg

28 calls = [] 7 ctx1abcdefg

29 

30 new_print = get_testing_print_function(calls) 7 ctx1abcdefg

31 

32 with patch("builtins.print", new=new_print): 7 ctx1abcdefg

33 mod.main() 7 ctx1abcdefg

34 assert calls == expected_calls 7 ctx1abcdefg