diff --git a/lib/temporal/testing/local_workflow_context.rb b/lib/temporal/testing/local_workflow_context.rb index 7d3321ae..b7036e92 100644 --- a/lib/temporal/testing/local_workflow_context.rb +++ b/lib/temporal/testing/local_workflow_context.rb @@ -199,9 +199,11 @@ def wait_for_any(*futures) end def wait_until(&unblock_condition) - raise 'You must pass an unblock condition block to wait_for' if unblock_condition.nil? + raise 'You must pass an unblock condition block to wait_until' if unblock_condition.nil? - Fiber.yield until unblock_condition.call + Fiber.new do + Fiber.yield until unblock_condition.call + end.resume return end