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

1from unittest.mock import patch 10 ctx1klmnopqrst

2 

3from sqlalchemy import inspect 10 ctx1klmnopqrst

4from sqlalchemy.engine.reflection import Inspector 10 ctx1klmnopqrst

5from sqlmodel import create_engine 10 ctx1klmnopqrst

6 

7from ...conftest import get_testing_print_function 10 ctx1klmnopqrst

8 

9 

10def test_tutorial(clear_sqlmodel): 10 ctx1klmnopqrst

11 from docs_src.tutorial.indexes import tutorial001 as mod 10 ctx1abcdefghij

12 

13 mod.sqlite_url = "sqlite://" 10 ctx1abcdefghij

14 mod.engine = create_engine(mod.sqlite_url) 10 ctx1abcdefghij

15 calls = [] 10 ctx1abcdefghij

16 

17 new_print = get_testing_print_function(calls) 10 ctx1abcdefghij

18 

19 with patch("builtins.print", new=new_print): 10 ctx1abcdefghij

20 mod.main() 10 ctx1abcdefghij

21 assert calls == [ 10 ctx1abcdefghij

22 [{"secret_name": "Dive Wilson", "age": None, "id": 1, "name": "Deadpond"}] 

23 ] 

24 

25 insp: Inspector = inspect(mod.engine) 10 ctx1abcdefghij

26 indexes = insp.get_indexes(str(mod.Hero.__tablename__)) 10 ctx1abcdefghij

27 expected_indexes = [ 10 ctx1abcdefghij

28 { 

29 "name": "ix_hero_name", 

30 "dialect_options": {}, 

31 "column_names": ["name"], 

32 "unique": 0, 

33 }, 

34 { 

35 "name": "ix_hero_age", 

36 "dialect_options": {}, 

37 "column_names": ["age"], 

38 "unique": 0, 

39 }, 

40 ] 

41 for index in expected_indexes: 10 ctx1abcdefghij

42 assert index in indexes, "This expected index should be in the indexes in DB" 10 ctx1abcdefghij

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

44 indexes.pop(indexes.index(index)) 10 ctx1abcdefghij

45 assert len(indexes) == 0, "The database should only have the expected indexes" 10 ctx1abcdefghij