Skip to content

Commit d690ddd

Browse files
committed
fix tests in celery-library
1 parent e9aa9ba commit d690ddd

File tree

1 file changed

+8
-10
lines changed

1 file changed

+8
-10
lines changed

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

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ async def test_submitting_task_calling_async_function_results_with_success_state
103103
celery_task_manager: CeleryTaskManager,
104104
with_celery_worker: WorkController,
105105
):
106-
task_filter = MyTaskFilter(user_id=42)
106+
task_filter = MyTaskFilter(user_id=42, owner="test-owner")
107107

108108
task_uuid = await celery_task_manager.submit_task(
109109
ExecutionMetadata(
@@ -134,7 +134,7 @@ async def test_submitting_task_with_failure_results_with_error(
134134
celery_task_manager: CeleryTaskManager,
135135
with_celery_worker: WorkController,
136136
):
137-
task_filter = MyTaskFilter(user_id=42)
137+
task_filter = MyTaskFilter(user_id=42, owner="test-owner")
138138

139139
task_uuid = await celery_task_manager.submit_task(
140140
ExecutionMetadata(
@@ -163,7 +163,7 @@ async def test_cancelling_a_running_task_aborts_and_deletes(
163163
celery_task_manager: CeleryTaskManager,
164164
with_celery_worker: WorkController,
165165
):
166-
task_filter = MyTaskFilter(user_id=42)
166+
task_filter = MyTaskFilter(user_id=42, owner="test-owner")
167167

168168
task_uuid = await celery_task_manager.submit_task(
169169
ExecutionMetadata(
@@ -186,7 +186,7 @@ async def test_listing_task_uuids_contains_submitted_task(
186186
celery_task_manager: CeleryTaskManager,
187187
with_celery_worker: WorkController,
188188
):
189-
task_filter = MyTaskFilter(user_id=42)
189+
task_filter = MyTaskFilter(user_id=42, owner="test-owner")
190190

191191
task_uuid = await celery_task_manager.submit_task(
192192
ExecutionMetadata(
@@ -215,18 +215,16 @@ async def test_filtering_listing_tasks(
215215
class MyFilter(OwnerMetadata):
216216
user_id: int
217217
product_name: str | Wildcard
218-
client_app: str | Wildcard
219218

220219
user_id = 42
220+
_owner = "test-owner"
221221
expected_task_uuids: set[TaskUUID] = set()
222222
all_tasks: list[tuple[TaskUUID, MyFilter]] = []
223223

224224
try:
225225
for _ in range(5):
226226
task_filter = MyFilter(
227-
user_id=user_id,
228-
product_name=_faker.word(),
229-
client_app=_faker.word(),
227+
user_id=user_id, product_name=_faker.word(), owner=_owner
230228
)
231229
task_uuid = await celery_task_manager.submit_task(
232230
ExecutionMetadata(
@@ -241,7 +239,7 @@ class MyFilter(OwnerMetadata):
241239
task_filter = MyFilter(
242240
user_id=_faker.pyint(min_value=100, max_value=200),
243241
product_name=_faker.word(),
244-
client_app=_faker.word(),
242+
owner=_owner,
245243
)
246244
task_uuid = await celery_task_manager.submit_task(
247245
ExecutionMetadata(
@@ -254,7 +252,7 @@ class MyFilter(OwnerMetadata):
254252
search_filter = MyFilter(
255253
user_id=user_id,
256254
product_name=Wildcard(),
257-
client_app=Wildcard(),
255+
owner=_owner,
258256
)
259257
tasks = await celery_task_manager.list_tasks(search_filter)
260258
assert expected_task_uuids == {task.uuid for task in tasks}

0 commit comments

Comments
 (0)