Coverage for tests/test_tutorial/test_limit_and_offset/test_tutorial001.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 {"id": 1, "name": "Deadpond", "secret_name": "Dive Wilson", "age": None}, 

11 { 

12 "id": 2, 

13 "name": "Spider-Boy", 

14 "secret_name": "Pedro Parqueador", 

15 "age": None, 

16 }, 

17 {"id": 3, "name": "Rusty-Man", "secret_name": "Tommy Sharp", "age": 48}, 

18 ] 

19 ] 

20] 

21 

22 

23def test_tutorial(clear_sqlmodel): 6 ctx1lghijk

24 from docs_src.tutorial.offset_and_limit import tutorial001 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