mirror of
https://github.com/fastapi/sqlmodel.git
synced 2026-02-04 03:33:28 +08:00
4ea6a32f6f
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
import importlib
|
|
from types import ModuleType
|
|
|
|
import pytest
|
|
from sqlalchemy import inspect
|
|
from sqlalchemy.engine.reflection import Inspector
|
|
from sqlmodel import create_engine
|
|
|
|
from ...conftest import PrintMock, needs_py310
|
|
|
|
|
|
@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.tutorial.indexes.{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 == [
|
|
[{"name": "Tarantula", "secret_name": "Natalia Roman-on", "age": 32, "id": 4}],
|
|
[{"name": "Black Lion", "secret_name": "Trevor Challa", "age": 35, "id": 5}],
|
|
]
|
|
|
|
insp: Inspector = inspect(mod.engine)
|
|
indexes = insp.get_indexes(str(mod.Hero.__tablename__))
|
|
expected_indexes = [
|
|
{
|
|
"name": "ix_hero_name",
|
|
"dialect_options": {},
|
|
"column_names": ["name"],
|
|
"unique": 0,
|
|
},
|
|
{
|
|
"name": "ix_hero_age",
|
|
"dialect_options": {},
|
|
"column_names": ["age"],
|
|
"unique": 0,
|
|
},
|
|
]
|
|
for index in expected_indexes:
|
|
assert index in indexes, "This expected index should be in the indexes in DB"
|
|
# Now that this index was checked, remove it from the list of indexes
|
|
indexes.pop(indexes.index(index))
|
|
assert len(indexes) == 0, "The database should only have the expected indexes"
|