Coverage for tests / test_tutorial / test_relationship_attributes / test_read_relationships / test_tutorial002.py: 100%
15 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 sqlmodel import create_engine 1abcdefghi
7from ....conftest import PrintMock, needs_py310 1abcdefghi
10@pytest.fixture( 1abcdefghi
11 name="mod",
12 params=[
13 pytest.param("tutorial002_py39"),
14 pytest.param("tutorial002_py310", marks=needs_py310),
15 ],
16)
17def get_module(request: pytest.FixtureRequest) -> ModuleType: 1abcdefghi
18 mod = importlib.import_module( 1abcdefghi
19 f"docs_src.tutorial.relationship_attributes.read_relationships.{request.param}"
20 )
21 mod.sqlite_url = "sqlite://" 1abcdefghi
22 mod.engine = create_engine(mod.sqlite_url) 1abcdefghi
23 return mod 1abcdefghi
26expected_calls = [ 1abcdefghi
27 [
28 "Created hero:",
29 {
30 "age": None,
31 "id": 1,
32 "secret_name": "Dive Wilson",
33 "team_id": 1,
34 "name": "Deadpond",
35 },
36 ],
37 [
38 "Created hero:",
39 {
40 "age": 48,
41 "id": 2,
42 "secret_name": "Tommy Sharp",
43 "team_id": 2,
44 "name": "Rusty-Man",
45 },
46 ],
47 [
48 "Created hero:",
49 {
50 "age": None,
51 "id": 3,
52 "secret_name": "Pedro Parqueador",
53 "team_id": None,
54 "name": "Spider-Boy",
55 },
56 ],
57 [
58 "Updated hero:",
59 {
60 "age": None,
61 "id": 3,
62 "secret_name": "Pedro Parqueador",
63 "team_id": 2,
64 "name": "Spider-Boy",
65 },
66 ],
67 [
68 "Team Wakaland:",
69 {"id": 3, "name": "Wakaland", "headquarters": "Wakaland Capital City"},
70 ],
71 [
72 "Preventers new hero:",
73 {
74 "age": 32,
75 "id": 6,
76 "secret_name": "Natalia Roman-on",
77 "team_id": 2,
78 "name": "Tarantula",
79 },
80 ],
81 [
82 "Preventers new hero:",
83 {
84 "age": 36,
85 "id": 7,
86 "secret_name": "Steve Weird",
87 "team_id": 2,
88 "name": "Dr. Weird",
89 },
90 ],
91 [
92 "Preventers new hero:",
93 {
94 "age": 93,
95 "id": 8,
96 "secret_name": "Esteban Rogelios",
97 "team_id": 2,
98 "name": "Captain North America",
99 },
100 ],
101 [
102 "Preventers heroes:",
103 [
104 {
105 "age": 48,
106 "id": 2,
107 "secret_name": "Tommy Sharp",
108 "team_id": 2,
109 "name": "Rusty-Man",
110 },
111 {
112 "age": None,
113 "id": 3,
114 "secret_name": "Pedro Parqueador",
115 "team_id": 2,
116 "name": "Spider-Boy",
117 },
118 {
119 "age": 32,
120 "id": 6,
121 "secret_name": "Natalia Roman-on",
122 "team_id": 2,
123 "name": "Tarantula",
124 },
125 {
126 "age": 36,
127 "id": 7,
128 "secret_name": "Steve Weird",
129 "team_id": 2,
130 "name": "Dr. Weird",
131 },
132 {
133 "age": 93,
134 "id": 8,
135 "secret_name": "Esteban Rogelios",
136 "team_id": 2,
137 "name": "Captain North America",
138 },
139 ],
140 ],
141 [
142 "Spider-Boy without team:",
143 {
144 "age": None,
145 "id": 3,
146 "secret_name": "Pedro Parqueador",
147 "team_id": None,
148 "name": "Spider-Boy",
149 },
150 ],
151]
154def test_tutorial(print_mock: PrintMock, mod: ModuleType): 1abcdefghi
155 mod.main() 1jklmnopqr
156 assert print_mock.calls == expected_calls 1jklmnopqr