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

1from unittest.mock import patch 6 ctx1lghijk

2 

3from sqlmodel import create_engine 6 ctx1lghijk

4 

5from ...conftest import get_testing_print_function 6 ctx1lghijk

6 

7expected_calls = [ 5 ctx1ghijk

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): 6 ctx1lghijk

24 from docs_src.tutorial.offset_and_limit import tutorial002 as mod 6 ctx1abcdef

25 

26 mod.sqlite_url = "sqlite://" 6 ctx1abcdef

27 mod.engine = create_engine(mod.sqlite_url) 6 ctx1abcdef

28 calls = [] 6 ctx1abcdef

29 

30 new_print = get_testing_print_function(calls) 6 ctx1abcdef

31 

32 with patch("builtins.print", new=new_print): 6 ctx1abcdef

33 mod.main() 6 ctx1abcdef

34 assert calls == expected_calls 6 ctx1abcdef