Coverage for tests / test_tutorial / test_indexes / test_tutorial002.py: 100%

1import importlib 9 ctx1abcdefghi

2from types import ModuleType 9 ctx1abcdefghi

3 

4import pytest 9 ctx1abcdefghi

5from sqlalchemy import Engine, inspect 9 ctx1abcdefghi

6from sqlalchemy.engine.reflection import Inspector 9 ctx1abcdefghi

7 

8from ...conftest import PrintMock 9 ctx1abcdefghi

9 

10 

11@pytest.fixture( 9 ctx1abcdefghi

12 name="mod", 

13 params=[ 

14 pytest.param("tutorial002_py310"), 

15 ], 

16) 

17def get_module(request: pytest.FixtureRequest, database_engine: Engine) -> ModuleType: 9 ctx1abcdefghi

18 mod = importlib.import_module(f"docs_src.tutorial.indexes.{request.param}") 9 ctx1abcdefghi

19 mod.engine = database_engine 9 ctx1abcdefghi

20 return mod 9 ctx1abcdefghi

21 

22 

23def test_tutorial(print_mock: PrintMock, mod: ModuleType): 9 ctx1abcdefghi

24 mod.main() 9 ctx1jklmnopqr

25 assert print_mock.calls == [ 9 ctx1jklmnopqr

26 [{"name": "Tarantula", "secret_name": "Natalia Roman-on", "age": 32, "id": 4}], 

27 [{"name": "Black Lion", "secret_name": "Trevor Challa", "age": 35, "id": 5}], 

28 ] 

29 

30 insp: Inspector = inspect(mod.engine) 9 ctx1jklmnopqr

31 indexes = insp.get_indexes(str(mod.Hero.__tablename__)) 9 ctx1jklmnopqr

32 assert { 9 ctx1jklmnopqr

33 index["name"]: (index["column_names"], index["unique"]) for index in indexes 

34 } == { 

35 "ix_hero_name": (["name"], False), 

36 "ix_hero_age": (["age"], False), 

37 }