sqlmodel/tests/test_tutorial/test_one/test_tutorial007.py at main · letsdeepchat/sqlmodel · GitHub

Repository navigation

30 lines (24 loc) · 706 Bytes

import importlib

from types import ModuleType

import pytest

from sqlmodel import create_engine

from ...conftest import PrintMock

@pytest.fixture(

name="mod",

params=[

pytest.param("tutorial007_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},

]