mirror of
https://github.com/fastapi/sqlmodel.git
synced 2026-02-03 02:53:21 +08:00
4ea6a32f6f
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
32 lines
761 B
Python
32 lines
761 B
Python
import importlib
|
|
from types import ModuleType
|
|
|
|
import pytest
|
|
from sqlmodel import create_engine
|
|
|
|
from ...conftest import PrintMock, needs_py310
|
|
|
|
|
|
@pytest.fixture(
|
|
name="mod",
|
|
params=[
|
|
"tutorial008",
|
|
pytest.param("tutorial008_py310", marks=needs_py310),
|
|
],
|
|
)
|
|
def get_module(request: pytest.FixtureRequest) -> ModuleType:
|
|
mod = importlib.import_module(f"docs_src.tutorial.one.{request.param}")
|
|
mod.sqlite_url = "sqlite://"
|
|
mod.engine = create_engine(mod.sqlite_url)
|
|
return mod
|
|
|
|
|
|
def test_tutorial(print_mock: PrintMock, mod: ModuleType):
|
|
mod.main()
|
|
assert print_mock.calls == [
|
|
[
|
|
"Hero:",
|
|
{"name": "Deadpond", "secret_name": "Dive Wilson", "age": None, "id": 1},
|
|
]
|
|
]
|