Coverage for tests / test_tutorial / test_relationship_attributes / test_delete_records_relationship / test_tutorial004.py: 100%
26 statements
« prev ^ index » next coverage.py v7.13.0, created at 2026-01-06 21:09 +0000
1import importlib 1abcdefghi
2from types import ModuleType 1abcdefghi
4import pytest 1abcdefghi
5from sqlalchemy.exc import IntegrityError 1abcdefghi
6from sqlmodel import Session, create_engine, select 1abcdefghi
8from ....conftest import PrintMock, needs_py310 1abcdefghi
11@pytest.fixture( 1abcdefghi
12 name="mod",
13 params=[
14 pytest.param("tutorial004_py39"),
15 pytest.param("tutorial004_py310", marks=needs_py310),
16 ],
17)
18def get_module(request: pytest.FixtureRequest) -> ModuleType: 1abcdefghi
19 mod = importlib.import_module( 1abcdefghi
20 f"docs_src.tutorial.relationship_attributes.cascade_delete_relationships.{request.param}"
21 )
22 mod.sqlite_url = "sqlite://" 1abcdefghi
23 mod.engine = create_engine(mod.sqlite_url) 1abcdefghi
24 return mod 1abcdefghi
27def test_tutorial(print_mock: PrintMock, mod: ModuleType): 1abcdefghi
28 mod.create_db_and_tables() 1jklmnopqr
29 mod.create_heroes() 1jklmnopqr
30 mod.select_deleted_heroes() 1jklmnopqr
31 with Session(mod.engine) as session: 1jklmnopqr
32 team = session.exec(select(mod.Team).where(mod.Team.name == "Wakaland")).one() 1jklmnopqr
33 team.heroes.clear() 1jklmnopqr
34 session.add(team) 1jklmnopqr
35 session.commit() 1jklmnopqr
36 mod.delete_team() 1jklmnopqr
37 assert print_mock.calls == [ 1jklmnopqr
38 [
39 "Created hero:",
40 {
41 "age": None,
42 "id": 1,
43 "name": "Deadpond",
44 "secret_name": "Dive Wilson",
45 "team_id": 1,
46 },
47 ],
48 [
49 "Created hero:",
50 {
51 "age": 48,
52 "id": 2,
53 "name": "Rusty-Man",
54 "secret_name": "Tommy Sharp",
55 "team_id": 2,
56 },
57 ],
58 [
59 "Created hero:",
60 {
61 "age": None,
62 "id": 3,
63 "name": "Spider-Boy",
64 "secret_name": "Pedro Parqueador",
65 "team_id": None,
66 },
67 ],
68 [
69 "Updated hero:",
70 {
71 "age": None,
72 "id": 3,
73 "name": "Spider-Boy",
74 "secret_name": "Pedro Parqueador",
75 "team_id": 2,
76 },
77 ],
78 [
79 "Team Wakaland:",
80 {"headquarters": "Wakaland Capital City", "id": 3, "name": "Wakaland"},
81 ],
82 [
83 "Black Lion has no team:",
84 {
85 "age": 35,
86 "id": 4,
87 "name": "Black Lion",
88 "secret_name": "Trevor Challa",
89 "team_id": 3,
90 },
91 ],
92 [
93 "Princess Sure-E has no team:",
94 {
95 "age": None,
96 "id": 5,
97 "name": "Princess Sure-E",
98 "secret_name": "Sure-E",
99 "team_id": 3,
100 },
101 ],
102 [
103 "Deleted team:",
104 {"headquarters": "Wakaland Capital City", "id": 3, "name": "Wakaland"},
105 ],
106 ]
108 with pytest.raises(IntegrityError) as exc: 1jklmnopqr
109 mod.main() 1jklmnopqr
110 assert "FOREIGN KEY constraint failed" in str(exc.value) 1jklmnopqr