Fix test & linters

seedofjoy · May 29, 2022 · f4940f7 · f4940f7
1 parent bac6e12
commit f4940f7
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/darq/app.py b/darq/app.py
@@ -295,7 +295,7 @@ async def delay(*args: t.Any, **kwargs: t.Any) -> t.Optional[Job]:
             self.registry.add(Task.new(
                 coroutine=function, name=name,
                 keep_result=keep_result, timeout=timeout, max_tries=max_tries,
-                with_ctx=with_ctx
+                with_ctx=with_ctx,
             ))
 
             return function

diff --git a/darq/worker.py b/darq/worker.py
@@ -68,7 +68,7 @@ def new(
         return cls(
             name=name, coroutine=coroutine, timeout_s=to_seconds(timeout),
             keep_result_s=to_seconds(keep_result), max_tries=max_tries,
-            with_ctx=with_ctx
+            with_ctx=with_ctx,
         )
 
 

diff --git a/tests/test_app.py b/tests/test_app.py
@@ -113,7 +113,7 @@ async def test_task_parametrized(darq):
     with_ctx = True
     foobar_task = darq.task(
         keep_result=keep_result, timeout=timeout,
-        max_tries=max_tries, queue=queue, with_ctx=with_ctx
+        max_tries=max_tries, queue=queue, with_ctx=with_ctx,
     )(foobar)
 
     task_name = 'tests.test_app.foobar'
@@ -399,7 +399,7 @@ async def foobar_with_ctx(ctx, a: int) -> int:
 ])
 async def test_run_task_with_ctx(
         func_args, func_kwargs, func_ctx, result,
-        arq_redis, caplog, worker_factory
+        arq_redis, caplog, worker_factory,
 ):
     caplog.set_level(logging.INFO)
 
@@ -419,7 +419,7 @@ async def on_worker_startup(ctx):
     job_id = 'testing'
     function_name = 'tests.test_app.foobar_with_ctx'
     await foobar_with_ctx_task.apply_async(
-        func_args, func_kwargs, job_id=job_id
+        func_args, func_kwargs, job_id=job_id,
     )
 
     worker = worker_factory(darq)

diff --git a/tests/test_registry.py b/tests/test_registry.py
@@ -15,12 +15,12 @@ def test_registry():
     task_name = 'foobar_task'
     worker_task = Task(
         name=task_name, coroutine=foobar, timeout_s=None, keep_result_s=None,
-        max_tries=None,
+        max_tries=None, with_ctx=False,
     )
     task_name_2 = 'foobar_2_task'
     worker_task_2 = Task(
         name=task_name_2, coroutine=foobar_2, timeout_s=None,
-        keep_result_s=None, max_tries=None,
+        keep_result_s=None, max_tries=None, with_ctx=False,
     )
 
     registry.add(worker_task)