Coverage for tests / test_tutorial / test_relationship_attributes / test_back_populates / 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

3 

4import pytest 1abcdefghi

5from sqlmodel import create_engine 1abcdefghi

6 

7from ....conftest import PrintMock, needs_py310 1abcdefghi

8 

9 

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.back_populates.{request.param}" 

20 ) 

21 mod.sqlite_url = "sqlite://" 1abcdefghi

22 mod.engine = create_engine(mod.sqlite_url) 1abcdefghi

23 return mod 1abcdefghi

24 

25 

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 "Hero Spider-Boy:", 

143 { 

144 "age": None, 

145 "id": 3, 

146 "secret_name": "Pedro Parqueador", 

147 "team_id": 2, 

148 "name": "Spider-Boy", 

149 }, 

150 ], 

151 [ 

152 "Preventers Team:", 

153 {"id": 2, "name": "Preventers", "headquarters": "Sharp Tower"}, 

154 ], 

155 [ 

156 "Preventers Team Heroes:", 

157 [ 

158 { 

159 "age": 48, 

160 "id": 2, 

161 "secret_name": "Tommy Sharp", 

162 "team_id": 2, 

163 "name": "Rusty-Man", 

164 }, 

165 { 

166 "age": None, 

167 "id": 3, 

168 "secret_name": "Pedro Parqueador", 

169 "team_id": 2, 

170 "name": "Spider-Boy", 

171 }, 

172 { 

173 "age": 32, 

174 "id": 6, 

175 "secret_name": "Natalia Roman-on", 

176 "team_id": 2, 

177 "name": "Tarantula", 

178 }, 

179 { 

180 "age": 36, 

181 "id": 7, 

182 "secret_name": "Steve Weird", 

183 "team_id": 2, 

184 "name": "Dr. Weird", 

185 }, 

186 { 

187 "age": 93, 

188 "id": 8, 

189 "secret_name": "Esteban Rogelios", 

190 "team_id": 2, 

191 "name": "Captain North America", 

192 }, 

193 ], 

194 ], 

195 [ 

196 "Spider-Boy without team:", 

197 { 

198 "age": None, 

199 "id": 3, 

200 "secret_name": "Pedro Parqueador", 

201 "team_id": 2, 

202 "name": "Spider-Boy", 

203 }, 

204 ], 

205 [ 

206 "Preventers Team Heroes again:", 

207 [ 

208 { 

209 "age": 48, 

210 "id": 2, 

211 "secret_name": "Tommy Sharp", 

212 "team_id": 2, 

213 "name": "Rusty-Man", 

214 }, 

215 { 

216 "age": 32, 

217 "id": 6, 

218 "secret_name": "Natalia Roman-on", 

219 "team_id": 2, 

220 "name": "Tarantula", 

221 }, 

222 { 

223 "age": 36, 

224 "id": 7, 

225 "secret_name": "Steve Weird", 

226 "team_id": 2, 

227 "name": "Dr. Weird", 

228 }, 

229 { 

230 "age": 93, 

231 "id": 8, 

232 "secret_name": "Esteban Rogelios", 

233 "team_id": 2, 

234 "name": "Captain North America", 

235 }, 

236 ], 

237 ], 

238 ["After committing"], 

239 [ 

240 "Spider-Boy after commit:", 

241 { 

242 "age": None, 

243 "id": 3, 

244 "secret_name": "Pedro Parqueador", 

245 "team_id": None, 

246 "name": "Spider-Boy", 

247 }, 

248 ], 

249 [ 

250 "Preventers Team Heroes after commit:", 

251 [ 

252 { 

253 "age": 48, 

254 "id": 2, 

255 "secret_name": "Tommy Sharp", 

256 "team_id": 2, 

257 "name": "Rusty-Man", 

258 }, 

259 { 

260 "age": 32, 

261 "id": 6, 

262 "secret_name": "Natalia Roman-on", 

263 "team_id": 2, 

264 "name": "Tarantula", 

265 }, 

266 { 

267 "age": 36, 

268 "id": 7, 

269 "secret_name": "Steve Weird", 

270 "team_id": 2, 

271 "name": "Dr. Weird", 

272 }, 

273 { 

274 "age": 93, 

275 "id": 8, 

276 "secret_name": "Esteban Rogelios", 

277 "team_id": 2, 

278 "name": "Captain North America", 

279 }, 

280 ], 

281 ], 

282] 

283 

284 

285def test_tutorial(print_mock: PrintMock, mod: ModuleType): 1abcdefghi

286 mod.main() 1jklmnopqr

287 assert print_mock.calls == expected_calls 1jklmnopqr