@@ -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