@@ -61,7 +61,8 @@ public void ConfigureWorker(Runner runner, TemporalWorkerOptions options) =>
6161 runner . NewWorkflowOptions ( ) ) ;
6262
6363 // Start update
64- var updateHandle1 = await handle . StartUpdateAsync ( wf => wf . SuccessfulUpdateAsync ( ) ) ;
64+ var updateHandle1 = await handle . StartUpdateAsync (
65+ wf => wf . SuccessfulUpdateAsync ( ) , new ( WorkflowUpdateStage . Accepted ) ) ;
6566 // Send signal to finish the update
6667 await handle . SignalAsync ( wf => wf . FinishUpdateAsync ( ) ) ;
6768 // Confirm result
@@ -70,7 +71,8 @@ public void ConfigureWorker(Runner runner, TemporalWorkerOptions options) =>
7071 Assert . Equal ( 123 , await handle . GetUpdateHandle < int > ( updateHandle1 . Id ) . GetResultAsync ( ) ) ;
7172
7273 // Start a failed update
73- var updateHandle2 = await handle . StartUpdateAsync ( wf => wf . FailureUpdateAsync ( ) ) ;
74+ var updateHandle2 = await handle . StartUpdateAsync (
75+ wf => wf . FailureUpdateAsync ( ) , new ( WorkflowUpdateStage . Accepted ) ) ;
7476 // Send signal to finish the update
7577 await handle . SignalAsync ( wf => wf . FinishUpdateAsync ( ) ) ;
7678 // Confirm failure
@@ -79,7 +81,8 @@ public void ConfigureWorker(Runner runner, TemporalWorkerOptions options) =>
7981 Assert . Equal ( "Intentional failure" , exc . InnerException ? . Message ) ;
8082
8183 // Start an update but cancel/timeout waiting on its result
82- var updateHandle3 = await handle . StartUpdateAsync ( wf => wf . SuccessfulUpdateAsync ( ) ) ;
84+ var updateHandle3 = await handle . StartUpdateAsync (
85+ wf => wf . SuccessfulUpdateAsync ( ) , new ( WorkflowUpdateStage . Accepted ) ) ;
8386 // Wait for result only for 100ms
8487 using var tokenSource = new CancellationTokenSource ( ) ;
8588 tokenSource . CancelAfter ( TimeSpan . FromMilliseconds ( 100 ) ) ;
0 commit comments