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

1import importlib 8 ctx1abcdefgh

2from types import ModuleType 8 ctx1abcdefgh

3 

4import pytest 8 ctx1abcdefgh

5from sqlalchemy import inspect 8 ctx1abcdefgh

6from sqlalchemy.engine.reflection import Inspector 8 ctx1abcdefgh

7from sqlmodel import create_engine 8 ctx1abcdefgh

8 

9from ...conftest import PrintMock 8 ctx1abcdefgh

10 

11 

12@pytest.fixture( 8 ctx1abcdefgh

13 name="mod", 

14 params=[ 

15 pytest.param("tutorial002_py310"), 

16 ], 

17) 

18def get_module(request: pytest.FixtureRequest) -> ModuleType: 8 ctx1abcdefgh

19 mod = importlib.import_module(f"docs_src.tutorial.indexes.{request.param}") 8 ctx1abcdefgh

20 mod.sqlite_url = "sqlite://" 8 ctx1abcdefgh

21 mod.engine = create_engine(mod.sqlite_url) 8 ctx1abcdefgh

22 return mod 8 ctx1abcdefgh

23 

24 

25def test_tutorial(print_mock: PrintMock, mod: ModuleType): 8 ctx1abcdefgh

26 mod.main() 8 ctx1ijklmnop

27 assert print_mock.calls == [ 8 ctx1ijklmnop

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

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

30 ] 

31 

32 insp: Inspector = inspect(mod.engine) 8 ctx1ijklmnop

33 indexes = insp.get_indexes(str(mod.Hero.__tablename__)) 8 ctx1ijklmnop

34 expected_indexes = [ 8 ctx1ijklmnop

35 { 

36 "name": "ix_hero_name", 

37 "dialect_options": {}, 

38 "column_names": ["name"], 

39 "unique": 0, 

40 }, 

41 { 

42 "name": "ix_hero_age", 

43 "dialect_options": {}, 

44 "column_names": ["age"], 

45 "unique": 0, 

46 }, 

47 ] 

48 for index in expected_indexes: 8 ctx1ijklmnop

49 assert index in indexes, "This expected index should be in the indexes in DB" 8 ctx1ijklmnop

50 # Now that this index was checked, remove it from the list of indexes 

51 indexes.pop(indexes.index(index)) 8 ctx1ijklmnop

52 assert len(indexes) == 0, "The database should only have the expected indexes" 8 ctx1ijklmnop