format/lint

Author: nagl-temporal

resource_pool/pool_client/resource_pool_client.py

@@ -31,12 +31,9 @@ def register_signal_handler(self) -> None:
             )
 
     async def send_acquire_signal(self) -> None:
-        handle = workflow.get_external_workflow_handle_for(
+        await workflow.get_external_workflow_handle_for(
             ResourcePoolWorkflow.run, self.pool_workflow_id
-        )
-        await handle.signal(
-            "acquire_resource", AcquireRequest(workflow.info().workflow_id)
-        )
+        ).signal("acquire_resource", AcquireRequest(workflow.info().workflow_id))
 
     async def send_release_signal(self, acquired_resource: AcquiredResource) -> None:
         handle = workflow.get_external_workflow_handle_for(

tests/resource_pool/workflow_test.py

@@ -70,10 +70,9 @@ async def use_resource_mock(input: UseResourceActivityInput) -> None:
                     holder = None
 
         # Are all the resources free, per the query?
-        pool_handle = client.get_workflow_handle_for(
+        query_result = await client.get_workflow_handle_for(
             ResourcePoolWorkflow.run, RESOURCE_POOL_WORKFLOW_ID
-        )
-        query_result = await pool_handle.query(ResourcePoolWorkflow.get_current_holders)
+        ).query(ResourcePoolWorkflow.get_current_holders)
         assert query_result == {"r_a": None, "r_b": None, "r_c": None}