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

1from unittest.mock import patch 7 ctx1hijklmn

2 

3from sqlalchemy import inspect 7 ctx1hijklmn

4from sqlalchemy.engine.reflection import Inspector 7 ctx1hijklmn

5from sqlmodel import create_engine 7 ctx1hijklmn

6 

7from ...conftest import get_testing_print_function 7 ctx1hijklmn

8 

9 

10def test_tutorial(clear_sqlmodel): 7 ctx1hijklmn

11 from docs_src.tutorial.indexes import tutorial002 as mod 7 ctx1gabcdef

12 

13 mod.sqlite_url = "sqlite://" 7 ctx1gabcdef

14 mod.engine = create_engine(mod.sqlite_url) 7 ctx1gabcdef

15 calls = [] 7 ctx1gabcdef

16 

17 new_print = get_testing_print_function(calls) 7 ctx1gabcdef

18 

19 with patch("builtins.print", new=new_print): 7 ctx1gabcdef

20 mod.main() 7 ctx1gabcdef

21 assert calls == [ 7 ctx1gabcdef

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

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

24 ] 

25 

26 insp: Inspector = inspect(mod.engine) 7 ctx1gabcdef

27 indexes = insp.get_indexes(str(mod.Hero.__tablename__)) 7 ctx1gabcdef

28 expected_indexes = [ 6 ctx1abcdef

29 { 

30 "name": "ix_hero_name", 

31 "dialect_options": {}, 

32 "column_names": ["name"], 

33 "unique": 0, 

34 }, 

35 { 

36 "name": "ix_hero_age", 

37 "dialect_options": {}, 

38 "column_names": ["age"], 

39 "unique": 0, 

40 }, 

41 ] 

42 for index in expected_indexes: 7 ctx1gabcdef

43 assert index in indexes, "This expected index should be in the indexes in DB" 7 ctx1gabcdef

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

45 indexes.pop(indexes.index(index)) 7 ctx1gabcdef

46 assert len(indexes) == 0, "The database should only have the expected indexes" 7 ctx1gabcdef