@@ -243,7 +243,8 @@ def run(arg_, *, kwarg_=None):
243
243
pretend .call ("warehouse.task.run" , tags = ["task:warehouse.test.task" ])
244
244
]
245
245
assert metrics .increment .calls == [
246
- pretend .call ("warehouse.task.complete" , tags = ["task:warehouse.test.task" ])
246
+ pretend .call ("warehouse.task.start" , tags = ["task:warehouse.test.task" ]),
247
+ pretend .call ("warehouse.task.complete" , tags = ["task:warehouse.test.task" ]),
247
248
]
248
249
249
250
def test_run_retries_failed_transaction (self , metrics ):
@@ -288,7 +289,8 @@ def run():
288
289
pretend .call ("warehouse.task.run" , tags = ["task:warehouse.test.task" ])
289
290
]
290
291
assert metrics .increment .calls == [
291
- pretend .call ("warehouse.task.retried" , tags = ["task:warehouse.test.task" ])
292
+ pretend .call ("warehouse.task.start" , tags = ["task:warehouse.test.task" ]),
293
+ pretend .call ("warehouse.task.retried" , tags = ["task:warehouse.test.task" ]),
292
294
]
293
295
294
296
def test_run_doesnt_retries_failed_transaction (self , metrics ):
@@ -326,7 +328,8 @@ def run():
326
328
pretend .call ("warehouse.task.run" , tags = ["task:warehouse.test.task" ])
327
329
]
328
330
assert metrics .increment .calls == [
329
- pretend .call ("warehouse.task.failed" , tags = ["task:warehouse.test.task" ])
331
+ pretend .call ("warehouse.task.start" , tags = ["task:warehouse.test.task" ]),
332
+ pretend .call ("warehouse.task.failed" , tags = ["task:warehouse.test.task" ]),
330
333
]
331
334
332
335
def test_after_return_without_pyramid_env (self ):
0 commit comments