fmt

StackStorm · Sep 19, 2024 · 78c3248 · 78c3248
1 parent 223c7a6
commit 78c3248
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/st2actions/tests/unit/test_worker.py b/st2actions/tests/unit/test_worker.py
@@ -201,7 +201,9 @@ def test_worker_shutdown(self):
     @mock.patch.object(
         RedisDriver,
         "get_members",
-        mock.MagicMock(return_value=coordination.NoOpAsyncResult(("member-1", "member-2"))),
+        mock.MagicMock(
+            return_value=coordination.NoOpAsyncResult(("member-1", "member-2"))
+        ),
     )
     def test_worker_graceful_shutdown_with_multiple_runners(self):
         self.reset_config(

diff --git a/st2actions/tests/unit/test_workflow_engine.py b/st2actions/tests/unit/test_workflow_engine.py
@@ -104,7 +104,9 @@ def reset_config(
         tests_config.parse_args()
         if graceful_shutdown is not None:
             cfg.CONF.set_override(
-                name="graceful_shutdown", override=graceful_shutdown, group="actionrunner"
+                name="graceful_shutdown",
+                override=graceful_shutdown,
+                group="actionrunner",
             )
         if exit_still_active_check is not None:
             cfg.CONF.set_override(
@@ -361,7 +363,9 @@ def test_workflow_engine_shutdown(self):
     @mock.patch.object(
         RedisDriver,
         "get_members",
-        mock.MagicMock(return_value=coordination_service.NoOpAsyncResult(("member-1",))),
+        mock.MagicMock(
+            return_value=coordination_service.NoOpAsyncResult(("member-1",))
+        ),
     )
     def test_workflow_engine_shutdown_with_multiple_members(self):
         self.reset_config(service_registry=True)