Skip to content

Commit 2599304

Browse files
authored
feat: add warehouse.task.start metric (#18439)
1 parent 793f0f1 commit 2599304

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

tests/unit/test_tasks.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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):

warehouse/tasks.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@ def run(*args, **kwargs):
7272
metric_tags = [f"task:{obj.name}"]
7373

7474
with request.tm, metrics.timed("warehouse.task.run", tags=metric_tags):
75+
metrics.increment("warehouse.task.start", tags=metric_tags)
7576
try:
7677
result = original_run(*args, **kwargs)
7778
metrics.increment("warehouse.task.complete", tags=metric_tags)

0 commit comments

Comments
 (0)