@@ -161,7 +161,7 @@ async def test_retry_of_custom_exc_types_of_simple_middleware() -> None:
161
161
no_result_on_retry = True ,
162
162
default_retry_label = True ,
163
163
types_of_exceptions = (KeyError , ValueError ),
164
- )
164
+ ),
165
165
)
166
166
runs = 0
167
167
@@ -171,7 +171,7 @@ def run_task() -> None:
171
171
172
172
runs += 1
173
173
174
- raise ValueError ()
174
+ raise ValueError (runs )
175
175
176
176
task = await run_task .kiq ()
177
177
resp = await task .wait_result (timeout = 1 )
@@ -186,7 +186,7 @@ def run_task() -> None:
186
186
no_result_on_retry = True ,
187
187
default_retry_label = True ,
188
188
types_of_exceptions = (KeyError ,),
189
- )
189
+ ),
190
190
)
191
191
runs = 0
192
192
@@ -196,7 +196,7 @@ def run_task2() -> None:
196
196
197
197
runs += 1
198
198
199
- raise ValueError ()
199
+ raise ValueError (runs )
200
200
201
201
task = await run_task2 .kiq ()
202
202
resp = await task .wait_result (timeout = 1 )
@@ -214,7 +214,7 @@ async def test_retry_of_custom_exc_types_of_smart_middleware() -> None:
214
214
no_result_on_retry = True ,
215
215
default_retry_label = True ,
216
216
types_of_exceptions = (KeyError , ValueError ),
217
- )
217
+ ),
218
218
)
219
219
runs = 0
220
220
@@ -224,7 +224,7 @@ def run_task() -> None:
224
224
225
225
runs += 1
226
226
227
- raise ValueError ()
227
+ raise ValueError (runs )
228
228
229
229
task = await run_task .kiq ()
230
230
resp = await task .wait_result (timeout = 1 )
@@ -239,7 +239,7 @@ def run_task() -> None:
239
239
no_result_on_retry = True ,
240
240
default_retry_label = True ,
241
241
types_of_exceptions = (KeyError ,),
242
- )
242
+ ),
243
243
)
244
244
runs = 0
245
245
@@ -249,7 +249,7 @@ def run_task2() -> None:
249
249
250
250
runs += 1
251
251
252
- raise ValueError ()
252
+ raise ValueError (runs )
253
253
254
254
task = await run_task2 .kiq ()
255
255
resp = await task .wait_result (timeout = 1 )
0 commit comments