format/lint

Author: nagl-temporal

resource_pool/pool_client/resource_pool_client.py

@@ -31,18 +31,14 @@ 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(
+        await workflow.get_external_workflow_handle_for(
             ResourcePoolWorkflow.run, self.pool_workflow_id
-        )
-        await handle.signal(
+        ).signal(
             "release_resource",
             AcquireResponse(
                 resource=acquired_resource.resource,

tests/resource_pool/workflow_test.py

@@ -70,10 +70,12 @@ 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(
+        handle: WorkflowHandle[
+            ResourcePoolWorkflow, None
+        ] = client.get_workflow_handle_for(
             ResourcePoolWorkflow.run, RESOURCE_POOL_WORKFLOW_ID
         )
-        query_result = await pool_handle.query(ResourcePoolWorkflow.get_current_holders)
+        query_result = await handle.query(ResourcePoolWorkflow.get_current_holders)
         assert query_result == {"r_a": None, "r_b": None, "r_c": None}