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

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

View File

@@ -1,21 +1,28 @@
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
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.where import tutorial005 as mod
@pytest.fixture(
name="mod",
params=[
"tutorial005",
pytest.param("tutorial005_py310", marks=needs_py310),
],
)
def get_module(request: pytest.FixtureRequest) -> ModuleType:
mod = importlib.import_module(f"docs_src.tutorial.where.{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()
assert calls == [
def test_tutorial(print_mock: PrintMock, mod: ModuleType):
mod.main()
assert print_mock.calls == [
[{"name": "Tarantula", "secret_name": "Natalia Roman-on", "age": 32, "id": 4}]
]