Skip to content

Commit 027e4d7

Browse files
committed
WIP
1 parent 81b390a commit 027e4d7

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

tests/worker/test_nexus.py

+10-3
Original file line numberDiff line numberDiff line change
@@ -211,8 +211,12 @@ async def run(
211211
print(f"🌈 {'cancel returned':<24}: {cancel_ret}")
212212

213213
print(f"🌈 {'before await op_handle':<24}: {op_handle}")
214-
output = await op_handle
215-
print(f"🌈 {'after await op_handle':<24}: {op_handle}")
214+
try:
215+
output = await op_handle
216+
except Exception as e:
217+
print(f"🌈 {'await op_handle failed':<24}: {e.__class__}({e})")
218+
raise
219+
print(f"🌈 {'after await op_handle':<24}: {op_handle} -> {output}")
216220
return output
217221

218222
@workflow.update
@@ -274,6 +278,9 @@ async def test_async_response(
274278
request_cancel: bool,
275279
use_shorthand_defined_operation: bool,
276280
):
281+
print(
282+
f"🌈 {'test_async_response':<24}: {request_cancel=} {use_shorthand_defined_operation=}"
283+
)
277284
task_queue = str(uuid.uuid4())
278285
async with Worker(
279286
client,
@@ -283,7 +290,7 @@ async def test_async_response(
283290
workflow_runner=UnsandboxedWorkflowRunner(),
284291
):
285292
await create_nexus_endpoint(task_queue, client)
286-
operation_workflow_id = str(uuid.uuid4())
293+
operation_workflow_id = "default-workflow-id"
287294

288295
# Start the caller workflow.
289296
block_forever_waiting_for_cancellation = request_cancel

0 commit comments

Comments
 (0)