From 2409330af76acc19539df276b132963f09afbe38 Mon Sep 17 00:00:00 2001 From: John Yang Date: Mon, 19 May 2025 10:39:19 -0700 Subject: [PATCH] test(taskworker): Allow test tasks to have arbitary arguments --- src/sentry/taskworker/tasks/examples.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/sentry/taskworker/tasks/examples.py b/src/sentry/taskworker/tasks/examples.py index 81b6de9b993353..82792d245cffe3 100644 --- a/src/sentry/taskworker/tasks/examples.py +++ b/src/sentry/taskworker/tasks/examples.py @@ -2,6 +2,7 @@ import logging from time import sleep +from typing import Any from sentry.taskworker.namespaces import exampletasks from sentry.taskworker.retry import LastAction, NoRetriesRemainingError, Retry, RetryError @@ -12,7 +13,7 @@ @exampletasks.register(name="examples.say_hello") -def say_hello(name: str) -> None: +def say_hello(name: str, *args: list[Any], **kwargs: dict[str, Any]) -> None: logger.debug("Hello %s", name) @@ -83,6 +84,6 @@ def at_most_once_task() -> None: @exampletasks.register(name="examples.timed") -def timed_task(sleep_seconds: float | str) -> None: +def timed_task(sleep_seconds: float | str, *args: list[Any], **kwargs: dict[str, Any]) -> None: sleep(float(sleep_seconds)) logger.debug("timed_task complete")