git.ipfire.org Git - thirdparty/fastapi/sqlmodel.git/blob

   1 from unittest.mock import patch
   2
   3 from sqlmodel import create_engine
   4
   5 from ....conftest import get_testing_print_function, needs_py310
   6
   7 expected_calls = [
   8     [
   9         "Created hero:",
  10         {
  11             "name": "Deadpond",
  12             "age": None,
  13             "team_id": 1,
  14             "id": 1,
  15             "secret_name": "Dive Wilson",
  16         },
  17     ],
  18     [
  19         "Created hero:",
  20         {
  21             "name": "Rusty-Man",
  22             "age": 48,
  23             "team_id": 2,
  24             "id": 2,
  25             "secret_name": "Tommy Sharp",
  26         },
  27     ],
  28     [
  29         "Created hero:",
  30         {
  31             "name": "Spider-Boy",
  32             "age": None,
  33             "team_id": None,
  34             "id": 3,
  35             "secret_name": "Pedro Parqueador",
  36         },
  37     ],
  38 ]
  39
  40
  41 @needs_py310
  42 def test_tutorial(clear_sqlmodel):
  43     from docs_src.tutorial.relationship_attributes.define_relationship_attributes import (
  44         tutorial001_py310 as mod,
  45     )
  46
  47     mod.sqlite_url = "sqlite://"
  48     mod.engine = create_engine(mod.sqlite_url)
  49     calls = []
  50
  51     new_print = get_testing_print_function(calls)
  52
  53     with patch("builtins.print", new=new_print):
  54         mod.main()
  55     assert calls == expected_calls