GitHub-Proxy/sqlmodel-fix
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

📝 Add source examples for Python 3.9 and 3.10 (#715)

Browse Source 
* 📝 Add source examples for Python 3.9 and 3.10

* ✅ Add tests for new source examples for Python 3.9 and 3.10, still needs pytest markers

* ✅ Add tests for fastapi examples

* ✅ Update tests for FastAPI app testing, for Python 3.9 and 3.10, fixing multi-app testing conflicts

* ✅ Require Python 3.9 and 3.10 for tests

* ✅ Update tests with missing markers
This commit is contained in:
Sebastián Ramírez
2023-11-29 16:51:55 +01:00
committed by GitHub
parent cce30d7546
commit d8effcbc5c
243 changed files with 20057 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

290
tests/test_tutorial/test_relationship_attributes/test_back_populates/test_tutorial001_py310.py Normal file
View File

@@ -0,0 +1,290 @@
from unittest.mock import patch
import pytest
from sqlalchemy.exc import SAWarning
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 1,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Updated hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Team Wakaland:",
{"headquarters": "Wakaland Capital City", "id": 3, "name": "Wakaland"},
],
[
"Preventers new hero:",
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
],
[
"Preventers new hero:",
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
],
[
"Preventers new hero:",
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
[
"Preventers heroes:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
[
"Hero Spider-Boy:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Preventers Team:",
{"headquarters": "Sharp Tower", "id": 2, "name": "Preventers"},
],
[
"Preventers Team Heroes:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
[
"Spider-Boy without team:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Preventers Team Heroes again:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
["After committing"],
[
"Spider-Boy after commit:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Preventers Team Heroes after commit:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
]
@needs_py310
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.relationship_attributes.back_populates import (
tutorial001_py310 as mod,
)
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
with pytest.warns(SAWarning):
mod.main()
assert calls == expected_calls

290
tests/test_tutorial/test_relationship_attributes/test_back_populates/test_tutorial001_py39.py Normal file
View File

@@ -0,0 +1,290 @@
from unittest.mock import patch
import pytest
from sqlalchemy.exc import SAWarning
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py39
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 1,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Updated hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Team Wakaland:",
{"headquarters": "Wakaland Capital City", "id": 3, "name": "Wakaland"},
],
[
"Preventers new hero:",
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
],
[
"Preventers new hero:",
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
],
[
"Preventers new hero:",
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
[
"Preventers heroes:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
[
"Hero Spider-Boy:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Preventers Team:",
{"headquarters": "Sharp Tower", "id": 2, "name": "Preventers"},
],
[
"Preventers Team Heroes:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
[
"Spider-Boy without team:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Preventers Team Heroes again:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
["After committing"],
[
"Spider-Boy after commit:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Preventers Team Heroes after commit:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
]
@needs_py39
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.relationship_attributes.back_populates import (
tutorial001_py39 as mod,
)
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
with pytest.warns(SAWarning):
mod.main()
assert calls == expected_calls

280
tests/test_tutorial/test_relationship_attributes/test_back_populates/test_tutorial002_py310.py Normal file
View File

@@ -0,0 +1,280 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 1,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Updated hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Team Wakaland:",
{"id": 3, "name": "Wakaland", "headquarters": "Wakaland Capital City"},
],
[
"Preventers new hero:",
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
],
[
"Preventers new hero:",
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
],
[
"Preventers new hero:",
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
[
"Preventers heroes:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
[
"Hero Spider-Boy:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Preventers Team:",
{"id": 2, "name": "Preventers", "headquarters": "Sharp Tower"},
],
[
"Preventers Team Heroes:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
[
"Spider-Boy without team:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Preventers Team Heroes again:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
["After committing"],
[
"Spider-Boy after commit:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Preventers Team Heroes after commit:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
]
@needs_py310
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.relationship_attributes.back_populates import (
tutorial002_py310 as mod,
)
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

280
tests/test_tutorial/test_relationship_attributes/test_back_populates/test_tutorial002_py39.py Normal file
View File

@@ -0,0 +1,280 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py39
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 1,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Updated hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Team Wakaland:",
{"id": 3, "name": "Wakaland", "headquarters": "Wakaland Capital City"},
],
[
"Preventers new hero:",
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
],
[
"Preventers new hero:",
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
],
[
"Preventers new hero:",
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
[
"Preventers heroes:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
[
"Hero Spider-Boy:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Preventers Team:",
{"id": 2, "name": "Preventers", "headquarters": "Sharp Tower"},
],
[
"Preventers Team Heroes:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
[
"Spider-Boy without team:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Preventers Team Heroes again:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
["After committing"],
[
"Spider-Boy after commit:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Preventers Team Heroes after commit:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
]
@needs_py39
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.relationship_attributes.back_populates import (
tutorial002_py39 as mod,
)
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

21
tests/test_tutorial/test_relationship_attributes/test_back_populates/test_tutorial003_py310.py Normal file
View File

@@ -0,0 +1,21 @@
from sqlalchemy import inspect
from sqlalchemy.engine.reflection import Inspector
from sqlmodel import create_engine
from ....conftest import needs_py310
@needs_py310
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.relationship_attributes.back_populates import (
tutorial003_py310 as mod,
)
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
mod.main()
insp: Inspector = inspect(mod.engine)
assert insp.has_table(str(mod.Hero.__tablename__))
assert insp.has_table(str(mod.Weapon.__tablename__))
assert insp.has_table(str(mod.Power.__tablename__))
assert insp.has_table(str(mod.Team.__tablename__))

21
tests/test_tutorial/test_relationship_attributes/test_back_populates/test_tutorial003_py39.py Normal file
View File

@@ -0,0 +1,21 @@
from sqlalchemy import inspect
from sqlalchemy.engine.reflection import Inspector
from sqlmodel import create_engine
from ....conftest import needs_py39
@needs_py39
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.relationship_attributes.back_populates import (
tutorial003_py39 as mod,
)
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
mod.main()
insp: Inspector = inspect(mod.engine)
assert insp.has_table(str(mod.Hero.__tablename__))
assert insp.has_table(str(mod.Weapon.__tablename__))
assert insp.has_table(str(mod.Power.__tablename__))
assert insp.has_table(str(mod.Team.__tablename__))

99
tests/test_tutorial/test_relationship_attributes/test_create_and_update_relationships/test_tutorial001_py310.py Normal file
View File

@@ -0,0 +1,99 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 1,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Updated hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Team Wakaland:",
{"id": 3, "headquarters": "Wakaland Capital City", "name": "Wakaland"},
],
[
"Preventers new hero:",
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
],
[
"Preventers new hero:",
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
],
[
"Preventers new hero:",
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
]
@needs_py310
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.relationship_attributes.create_and_update_relationships import (
tutorial001_py310 as mod,
)
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

99
tests/test_tutorial/test_relationship_attributes/test_create_and_update_relationships/test_tutorial001_py39.py Normal file
View File

@@ -0,0 +1,99 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py39
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 1,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Updated hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Team Wakaland:",
{"id": 3, "headquarters": "Wakaland Capital City", "name": "Wakaland"},
],
[
"Preventers new hero:",
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
],
[
"Preventers new hero:",
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
],
[
"Preventers new hero:",
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
]
@needs_py39
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.relationship_attributes.create_and_update_relationships import (
tutorial001_py39 as mod,
)
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

55
tests/test_tutorial/test_relationship_attributes/test_define_relationship_attributes/test_tutorial001_py310.py Normal file
View File

@@ -0,0 +1,55 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Created hero:",
{
"name": "Deadpond",
"age": None,
"team_id": 1,
"id": 1,
"secret_name": "Dive Wilson",
},
],
[
"Created hero:",
{
"name": "Rusty-Man",
"age": 48,
"team_id": 2,
"id": 2,
"secret_name": "Tommy Sharp",
},
],
[
"Created hero:",
{
"name": "Spider-Boy",
"age": None,
"team_id": None,
"id": 3,
"secret_name": "Pedro Parqueador",
},
],
]
@needs_py310
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.relationship_attributes.define_relationship_attributes import (
tutorial001_py310 as mod,
)
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

55
tests/test_tutorial/test_relationship_attributes/test_define_relationship_attributes/test_tutorial001_py39.py Normal file
View File

@@ -0,0 +1,55 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py39
expected_calls = [
[
"Created hero:",
{
"name": "Deadpond",
"age": None,
"team_id": 1,
"id": 1,
"secret_name": "Dive Wilson",
},
],
[
"Created hero:",
{
"name": "Rusty-Man",
"age": 48,
"team_id": 2,
"id": 2,
"secret_name": "Tommy Sharp",
},
],
[
"Created hero:",
{
"name": "Spider-Boy",
"age": None,
"team_id": None,
"id": 3,
"secret_name": "Pedro Parqueador",
},
],
]
@needs_py39
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.relationship_attributes.define_relationship_attributes import (
tutorial001_py39 as mod,
)
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

107
tests/test_tutorial/test_relationship_attributes/test_read_relationships/test_tutorial001_py310.py Normal file
View File

@@ -0,0 +1,107 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 1,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Updated hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Team Wakaland:",
{"headquarters": "Wakaland Capital City", "id": 3, "name": "Wakaland"},
],
[
"Preventers new hero:",
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
],
[
"Preventers new hero:",
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
],
[
"Preventers new hero:",
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
[
"Spider-Boy's team:",
{"headquarters": "Sharp Tower", "id": 2, "name": "Preventers"},
],
[
"Spider-Boy's team again:",
{"headquarters": "Sharp Tower", "id": 2, "name": "Preventers"},
],
]
@needs_py310
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.relationship_attributes.read_relationships import (
tutorial001_py310 as mod,
)
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

107
tests/test_tutorial/test_relationship_attributes/test_read_relationships/test_tutorial001_py39.py Normal file
View File

@@ -0,0 +1,107 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py39
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 1,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Updated hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Team Wakaland:",
{"headquarters": "Wakaland Capital City", "id": 3, "name": "Wakaland"},
],
[
"Preventers new hero:",
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
],
[
"Preventers new hero:",
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
],
[
"Preventers new hero:",
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
[
"Spider-Boy's team:",
{"headquarters": "Sharp Tower", "id": 2, "name": "Preventers"},
],
[
"Spider-Boy's team again:",
{"headquarters": "Sharp Tower", "id": 2, "name": "Preventers"},
],
]
@needs_py39
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.relationship_attributes.read_relationships import (
tutorial001_py39 as mod,
)
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

149
tests/test_tutorial/test_relationship_attributes/test_read_relationships/test_tutorial002_py310.py Normal file
View File

@@ -0,0 +1,149 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 1,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Updated hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Team Wakaland:",
{"id": 3, "name": "Wakaland", "headquarters": "Wakaland Capital City"},
],
[
"Preventers new hero:",
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
],
[
"Preventers new hero:",
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
],
[
"Preventers new hero:",
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
[
"Preventers heroes:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
[
"Spider-Boy without team:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
]
@needs_py310
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.relationship_attributes.read_relationships import (
tutorial002_py310 as mod,
)
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

149
tests/test_tutorial/test_relationship_attributes/test_read_relationships/test_tutorial002_py39.py Normal file
View File

@@ -0,0 +1,149 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ....conftest import get_testing_print_function, needs_py39
expected_calls = [
[
"Created hero:",
{
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"team_id": 1,
"name": "Deadpond",
},
],
[
"Created hero:",
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
],
[
"Created hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
[
"Updated hero:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
],
[
"Team Wakaland:",
{"id": 3, "name": "Wakaland", "headquarters": "Wakaland Capital City"},
],
[
"Preventers new hero:",
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
],
[
"Preventers new hero:",
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
],
[
"Preventers new hero:",
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
[
"Preventers heroes:",
[
{
"age": 48,
"id": 2,
"secret_name": "Tommy Sharp",
"team_id": 2,
"name": "Rusty-Man",
},
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": 2,
"name": "Spider-Boy",
},
{
"age": 32,
"id": 6,
"secret_name": "Natalia Roman-on",
"team_id": 2,
"name": "Tarantula",
},
{
"age": 36,
"id": 7,
"secret_name": "Steve Weird",
"team_id": 2,
"name": "Dr. Weird",
},
{
"age": 93,
"id": 8,
"secret_name": "Esteban Rogelios",
"team_id": 2,
"name": "Captain North America",
},
],
],
[
"Spider-Boy without team:",
{
"age": None,
"id": 3,
"secret_name": "Pedro Parqueador",
"team_id": None,
"name": "Spider-Boy",
},
],
]
@needs_py39
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.relationship_attributes.read_relationships import (
tutorial002_py39 as mod,
)
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls