Skip to content

Commit 0a4da16

Browse files
fix: add mytaskfilter
1 parent e5556d7 commit 0a4da16

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

packages/celery-library/tests/unit/test_tasks.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,10 @@
3434
pytest_simcore_ops_services_selection = []
3535

3636

37+
class MyTaskFilter(TaskFilter):
38+
user_id: int
39+
40+
3741
async def _fake_file_processor(
3842
celery_app: Celery, task_name: str, task_id: str, files: list[str]
3943
) -> str:
@@ -94,7 +98,7 @@ async def test_submitting_task_calling_async_function_results_with_success_state
9498
celery_task_manager: CeleryTaskManager,
9599
with_celery_worker: WorkController,
96100
):
97-
task_filter = TaskFilter(user_id=42)
101+
task_filter = MyTaskFilter(user_id=42)
98102

99103
task_uuid = await celery_task_manager.submit_task(
100104
TaskMetadata(
@@ -125,7 +129,7 @@ async def test_submitting_task_with_failure_results_with_error(
125129
celery_task_manager: CeleryTaskManager,
126130
with_celery_worker: WorkController,
127131
):
128-
task_filter = TaskFilter(user_id=42)
132+
task_filter = MyTaskFilter(user_id=42)
129133

130134
task_uuid = await celery_task_manager.submit_task(
131135
TaskMetadata(
@@ -154,7 +158,7 @@ async def test_cancelling_a_running_task_aborts_and_deletes(
154158
celery_task_manager: CeleryTaskManager,
155159
with_celery_worker: WorkController,
156160
):
157-
task_filter = TaskFilter(user_id=42)
161+
task_filter = MyTaskFilter(user_id=42)
158162

159163
task_uuid = await celery_task_manager.submit_task(
160164
TaskMetadata(
@@ -177,7 +181,7 @@ async def test_listing_task_uuids_contains_submitted_task(
177181
celery_task_manager: CeleryTaskManager,
178182
with_celery_worker: WorkController,
179183
):
180-
task_filter = TaskFilter(user_id=42)
184+
task_filter = MyTaskFilter(user_id=42)
181185

182186
task_uuid = await celery_task_manager.submit_task(
183187
TaskMetadata(

0 commit comments

Comments
 (0)