python/sqlmodel
1
0
Fork 0
mirror of https://github.com/fastapi/sqlmodel.git synced 2026-03-13 09:29:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

✅ Simplify tests for code examples, one test file for multiple variants (#1664)

Browse Source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Motov Yurii
2025-12-24 16:47:05 +01:00
committed by GitHub
parent 4c4d5b44ff
commit 4ea6a32f6f
186 changed files with 1769 additions and 14628 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
tests/test_advanced/test_decimal/test_tutorial001.py
View File

@@ -1,9 +1,26 @@
import importlib
from decimal import Decimal
from unittest.mock import patch
from types import ModuleType
import pytest
from sqlmodel import create_engine
from ...conftest import get_testing_print_function
from ...conftest import PrintMock, needs_py310
@pytest.fixture(
name="mod",
params=[
"tutorial001",
pytest.param("tutorial001_py310", marks=needs_py310),
],
)
def get_module(request: pytest.FixtureRequest) -> ModuleType:
mod = importlib.import_module(f"docs_src.advanced.decimal.{request.param}")
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
return mod
expected_calls = [
[
@@ -30,15 +47,6 @@ expected_calls = [
]
def test_tutorial():
from docs_src.advanced.decimal import tutorial001 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
def test_tutorial(print_mock: PrintMock, mod: ModuleType):
mod.main()
assert print_mock.calls == expected_calls

45
tests/test_advanced/test_decimal/test_tutorial001_py310.py
View File

@@ -1,45 +0,0 @@
from decimal import Decimal
from unittest.mock import patch
from sqlmodel import create_engine
from ...conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Hero 1:",
{
"name": "Deadpond",
"age": None,
"id": 1,
"secret_name": "Dive Wilson",
"money": Decimal("1.100"),
},
],
[
"Hero 2:",
{
"name": "Rusty-Man",
"age": 48,
"id": 3,
"secret_name": "Tommy Sharp",
"money": Decimal("2.200"),
},
],
["Total money: 3.300"],
]
@needs_py310
def test_tutorial():
from docs_src.advanced.decimal 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

31
tests/test_advanced/test_uuid/test_tutorial001.py
View File

@@ -1,31 +1,38 @@
from unittest.mock import patch
import importlib
from types import ModuleType
import pytest
from dirty_equals import IsUUID
from sqlmodel import create_engine
from ...conftest import get_testing_print_function
from ...conftest import PrintMock, needs_py310
def test_tutorial() -> None:
from docs_src.advanced.uuid import tutorial001 as mod
@pytest.fixture(
name="mod",
params=[
"tutorial001",
pytest.param("tutorial001_py310", marks=needs_py310),
],
)
def get_module(request: pytest.FixtureRequest) -> ModuleType:
mod = importlib.import_module(f"docs_src.advanced.uuid.{request.param}")
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
return mod
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
first_uuid = calls[1][0]["id"]
def test_tutorial(print_mock: PrintMock, mod: ModuleType) -> None:
mod.main()
first_uuid = print_mock.calls[1][0]["id"]
assert first_uuid == IsUUID(4)
second_uuid = calls[7][0]["id"]
second_uuid = print_mock.calls[7][0]["id"]
assert second_uuid == IsUUID(4)
assert first_uuid != second_uuid
assert calls == [
assert print_mock.calls == [
["The hero before saving in the DB"],
[
{

72
tests/test_advanced/test_uuid/test_tutorial001_py310.py
View File

@@ -1,72 +0,0 @@
from unittest.mock import patch
from dirty_equals import IsUUID
from sqlmodel import create_engine
from ...conftest import get_testing_print_function, needs_py310
@needs_py310
def test_tutorial() -> None:
from docs_src.advanced.uuid 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()
first_uuid = calls[1][0]["id"]
assert first_uuid == IsUUID(4)
second_uuid = calls[7][0]["id"]
assert second_uuid == IsUUID(4)
assert first_uuid != second_uuid
assert calls == [
["The hero before saving in the DB"],
[
{
"name": "Deadpond",
"secret_name": "Dive Wilson",
"id": first_uuid,
"age": None,
}
],
["The hero ID was already set"],
[first_uuid],
["After saving in the DB"],
[
{
"name": "Deadpond",
"secret_name": "Dive Wilson",
"age": None,
"id": first_uuid,
}
],
["Created hero:"],
[
{
"name": "Spider-Boy",
"secret_name": "Pedro Parqueador",
"age": None,
"id": second_uuid,
}
],
["Created hero ID:"],
[second_uuid],
["Selected hero:"],
[
{
"name": "Spider-Boy",
"secret_name": "Pedro Parqueador",
"age": None,
"id": second_uuid,
}
],
["Selected hero ID:"],
[second_uuid],
]

31
tests/test_advanced/test_uuid/test_tutorial002.py
View File

@@ -1,31 +1,38 @@
from unittest.mock import patch
import importlib
from types import ModuleType
import pytest
from dirty_equals import IsUUID
from sqlmodel import create_engine
from ...conftest import get_testing_print_function
from ...conftest import PrintMock, needs_py310
def test_tutorial() -> None:
from docs_src.advanced.uuid import tutorial002 as mod
@pytest.fixture(
name="mod",
params=[
"tutorial002",
pytest.param("tutorial002_py310", marks=needs_py310),
],
)
def get_module(request: pytest.FixtureRequest) -> ModuleType:
mod = importlib.import_module(f"docs_src.advanced.uuid.{request.param}")
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
return mod
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
first_uuid = calls[1][0]["id"]
def test_tutorial(print_mock: PrintMock, mod: ModuleType) -> None:
mod.main()
first_uuid = print_mock.calls[1][0]["id"]
assert first_uuid == IsUUID(4)
second_uuid = calls[7][0]["id"]
second_uuid = print_mock.calls[7][0]["id"]
assert second_uuid == IsUUID(4)
assert first_uuid != second_uuid
assert calls == [
assert print_mock.calls == [
["The hero before saving in the DB"],
[
{

72
tests/test_advanced/test_uuid/test_tutorial002_py310.py
View File

@@ -1,72 +0,0 @@
from unittest.mock import patch
from dirty_equals import IsUUID
from sqlmodel import create_engine
from ...conftest import get_testing_print_function, needs_py310
@needs_py310
def test_tutorial() -> None:
from docs_src.advanced.uuid 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()
first_uuid = calls[1][0]["id"]
assert first_uuid == IsUUID(4)
second_uuid = calls[7][0]["id"]
assert second_uuid == IsUUID(4)
assert first_uuid != second_uuid
assert calls == [
["The hero before saving in the DB"],
[
{
"name": "Deadpond",
"secret_name": "Dive Wilson",
"id": first_uuid,
"age": None,
}
],
["The hero ID was already set"],
[first_uuid],
["After saving in the DB"],
[
{
"name": "Deadpond",
"secret_name": "Dive Wilson",
"age": None,
"id": first_uuid,
}
],
["Created hero:"],
[
{
"name": "Spider-Boy",
"secret_name": "Pedro Parqueador",
"age": None,
"id": second_uuid,
}
],
["Created hero ID:"],
[second_uuid],
["Selected hero:"],
[
{
"name": "Spider-Boy",
"secret_name": "Pedro Parqueador",
"age": None,
"id": second_uuid,
}
],
["Selected hero ID:"],
[second_uuid],
]