Coverage for tests / test_tutorial / test_indexes / test_tutorial001.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("tutorial001_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 [{"secret_name": "Dive Wilson", "age": None, "id": 1, "name": "Deadpond"}] 

29 ] 

30 

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

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

33 expected_indexes = [ 8 ctx1ijklmnop

34 { 

35 "name": "ix_hero_name", 

36 "dialect_options": {}, 

37 "column_names": ["name"], 

38 "unique": 0, 

39 }, 

40 { 

41 "name": "ix_hero_age", 

42 "dialect_options": {}, 

43 "column_names": ["age"], 

44 "unique": 0, 

45 }, 

46 ] 

47 for index in expected_indexes: 8 ctx1ijklmnop

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

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

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

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