Coverage for tests/test_tutorial/test_relationship_attributes/test_delete_records_relationship/test_tutorial004_py310.py: 100%

1from unittest.mock import patch 6 ctx1idefgh

2 

3import pytest 6 ctx1idefgh

4from sqlalchemy.exc import IntegrityError 6 ctx1idefgh

5from sqlmodel import Session, create_engine, select 6 ctx1idefgh

6 

7from tests.conftest import get_testing_print_function, needs_py310 6 ctx1idefgh

8 

9 

10@needs_py310 6 ctx1idefgh

11def test_tutorial(clear_sqlmodel): 5 ctx1defgh

12 from docs_src.tutorial.relationship_attributes.cascade_delete_relationships import ( 3 ctx1abc

13 tutorial004_py310 as mod, 

14 ) 

15 

16 mod.sqlite_url = "sqlite://" 3 ctx1abc

17 mod.engine = create_engine(mod.sqlite_url) 3 ctx1abc

18 calls = [] 3 ctx1abc

19 

20 new_print = get_testing_print_function(calls) 3 ctx1abc

21 

22 with patch("builtins.print", new=new_print): 3 ctx1abc

23 mod.create_db_and_tables() 3 ctx1abc

24 mod.create_heroes() 3 ctx1abc

25 mod.select_deleted_heroes() 3 ctx1abc

26 with Session(mod.engine) as session: 3 ctx1abc

27 team = session.exec( 3 ctx1abc

28 select(mod.Team).where(mod.Team.name == "Wakaland") 

29 ).one() 

30 team.heroes.clear() 3 ctx1abc

31 session.add(team) 3 ctx1abc

32 session.commit() 3 ctx1abc

33 mod.delete_team() 3 ctx1abc

34 assert calls == [ 3 ctx1abc

35 [ 

36 "Created hero:", 

37 { 

38 "age": None, 

39 "id": 1, 

40 "name": "Deadpond", 

41 "secret_name": "Dive Wilson", 

42 "team_id": 1, 

43 }, 

44 ], 

45 [ 

46 "Created hero:", 

47 { 

48 "age": 48, 

49 "id": 2, 

50 "name": "Rusty-Man", 

51 "secret_name": "Tommy Sharp", 

52 "team_id": 2, 

53 }, 

54 ], 

55 [ 

56 "Created hero:", 

57 { 

58 "age": None, 

59 "id": 3, 

60 "name": "Spider-Boy", 

61 "secret_name": "Pedro Parqueador", 

62 "team_id": None, 

63 }, 

64 ], 

65 [ 

66 "Updated hero:", 

67 { 

68 "age": None, 

69 "id": 3, 

70 "name": "Spider-Boy", 

71 "secret_name": "Pedro Parqueador", 

72 "team_id": 2, 

73 }, 

74 ], 

75 [ 

76 "Team Wakaland:", 

77 {"headquarters": "Wakaland Capital City", "id": 3, "name": "Wakaland"}, 

78 ], 

79 [ 

80 "Black Lion has no team:", 

81 { 

82 "age": 35, 

83 "id": 4, 

84 "name": "Black Lion", 

85 "secret_name": "Trevor Challa", 

86 "team_id": 3, 

87 }, 

88 ], 

89 [ 

90 "Princess Sure-E has no team:", 

91 { 

92 "age": None, 

93 "id": 5, 

94 "name": "Princess Sure-E", 

95 "secret_name": "Sure-E", 

96 "team_id": 3, 

97 }, 

98 ], 

99 [ 

100 "Deleted team:", 

101 {"headquarters": "Wakaland Capital City", "id": 3, "name": "Wakaland"}, 

102 ], 

103 ] 

104 

105 with pytest.raises(IntegrityError) as exc: 3 ctx1abc

106 mod.main() 3 ctx1abc

107 assert "FOREIGN KEY constraint failed" in str(exc.value) 3 ctx1abc