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

1from unittest.mock import patch 10 ctx1ijklmnopqr

2 

3from sqlmodel import create_engine 10 ctx1ijklmnopqr

4 

5from ...conftest import get_testing_print_function, needs_py310 10 ctx1ijklmnopqr

6 

7expected_calls = [ 10 ctx1ijklmnopqr

8 [ 

9 "Created hero:", 

10 { 

11 "id": 1, 

12 "name": "Deadpond", 

13 "age": None, 

14 "secret_name": "Dive Wilson", 

15 "team_id": 1, 

16 }, 

17 ], 

18 [ 

19 "Hero's team:", 

20 {"name": "Z-Force", "headquarters": "Sister Margaret's Bar", "id": 1}, 

21 ], 

22] 

23 

24 

25@needs_py310 10 ctx1ijklmnopqr

26def test_tutorial(): 10 ctx1ijklmnopqr

27 from docs_src.tutorial.code_structure.tutorial002_py310 import app, database 8 ctx1abcdefgh

28 

29 database.sqlite_url = "sqlite://" 8 ctx1abcdefgh

30 database.engine = create_engine(database.sqlite_url) 8 ctx1abcdefgh

31 app.engine = database.engine 8 ctx1abcdefgh

32 calls = [] 8 ctx1abcdefgh

33 

34 new_print = get_testing_print_function(calls) 8 ctx1abcdefgh

35 

36 with patch("builtins.print", new=new_print): 8 ctx1abcdefgh

37 app.main() 8 ctx1abcdefgh

38 assert calls == expected_calls 8 ctx1abcdefgh