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

88
tests/test_tutorial/test_delete/test_tutorial001_py310_tutorial002_py310.py
View File

@@ -1,88 +0,0 @@
from unittest.mock import patch
from sqlmodel import create_engine
from ...conftest import get_testing_print_function, needs_py310
expected_calls = [
[
"Hero 1:",
{"id": 2, "name": "Spider-Boy", "secret_name": "Pedro Parqueador", "age": None},
],
[
"Hero 2:",
{
"id": 7,
"name": "Captain North America",
"secret_name": "Esteban Rogelios",
"age": 93,
},
],
[
"Updated hero 1:",
{
"id": 2,
"name": "Spider-Youngster",
"secret_name": "Pedro Parqueador",
"age": 16,
},
],
[
"Updated hero 2:",
{
"id": 7,
"name": "Captain North America Except Canada",
"secret_name": "Esteban Rogelios",
"age": 110,
},
],
[
"Hero: ",
{
"id": 2,
"name": "Spider-Youngster",
"secret_name": "Pedro Parqueador",
"age": 16,
},
],
[
"Deleted hero:",
{
"id": 2,
"name": "Spider-Youngster",
"secret_name": "Pedro Parqueador",
"age": 16,
},
],
["There's no hero named Spider-Youngster"],
]
@needs_py310
def test_tutorial001(clear_sqlmodel):
from docs_src.tutorial.delete 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
@needs_py310
def test_tutorial002(clear_sqlmodel):
from docs_src.tutorial.delete 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

56
tests/test_tutorial/test_delete/test_tutorial001_tutorial002.py
View File

@@ -1,8 +1,10 @@
from unittest.mock import patch
import importlib
from types import ModuleType
import pytest
from sqlmodel import create_engine
from ...conftest import get_testing_print_function
from ...conftest import PrintMock, needs_py310
expected_calls = [
[
@@ -58,29 +60,35 @@ expected_calls = [
]
def test_tutorial001(clear_sqlmodel):
from docs_src.tutorial.delete 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
@pytest.fixture(name="module")
def get_module(request: pytest.FixtureRequest) -> ModuleType:
module = importlib.import_module(f"docs_src.tutorial.delete.{request.param}")
module.sqlite_url = "sqlite://"
module.engine = create_engine(module.sqlite_url)
return module
def test_tutorial002(clear_sqlmodel):
from docs_src.tutorial.delete import tutorial002 as mod
@pytest.mark.parametrize(
"module",
[
"tutorial001",
pytest.param("tutorial001_py310", marks=needs_py310),
],
indirect=True,
)
def test_tutorial001(print_mock: PrintMock, module: ModuleType):
module.main()
assert print_mock.calls == expected_calls
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
@pytest.mark.parametrize(
"module",
[
"tutorial002",
pytest.param("tutorial002_py310", marks=needs_py310),
],
indirect=True,
)
def test_tutorial002(print_mock: PrintMock, module: ModuleType):
module.main()
assert print_mock.calls == expected_calls