@@ -243,7 +243,8 @@ def run(arg_, *, kwarg_=None):
243243 pretend .call ("warehouse.task.run" , tags = ["task:warehouse.test.task" ])
244244 ]
245245 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" ]),
247248 ]
248249
249250 def test_run_retries_failed_transaction (self , metrics ):
@@ -288,7 +289,8 @@ def run():
288289 pretend .call ("warehouse.task.run" , tags = ["task:warehouse.test.task" ])
289290 ]
290291 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" ]),
292294 ]
293295
294296 def test_run_doesnt_retries_failed_transaction (self , metrics ):
@@ -326,7 +328,8 @@ def run():
326328 pretend .call ("warehouse.task.run" , tags = ["task:warehouse.test.task" ])
327329 ]
328330 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" ]),
330333 ]
331334
332335 def test_after_return_without_pyramid_env (self ):
0 commit comments