Update tests for new proto shapes

Signed-off-by: joshvanl <[email protected]>

Author: JoshVanL

client/worker_grpc.go

@@ -60,7 +60,6 @@ func (c *TaskHubGrpcClient) StartWorkItemListener(ctx context.Context, r *task.T
 			}
 		}()
 		for {
-			// TODO: Manage concurrency
 			workItem, err := stream.Recv()
 
 			if err != nil {

tests/task_executor_test.go

@@ -54,8 +54,8 @@ func Test_Executor_WaitForEventSchedulesTimer(t *testing.T) {
 	executor := task.NewTaskExecutor(r)
 	results, err := executor.ExecuteOrchestrator(ctx, iid, oldEvents, newEvents)
 	require.NoError(t, err)
-	require.Equal(t, 1, len(results.Response.Actions), "Expected a single action to be scheduled")
-	createTimerAction := results.Response.Actions[0].GetCreateTimer()
+	require.Equal(t, 1, len(results.Actions), "Expected a single action to be scheduled")
+	createTimerAction := results.Actions[0].GetCreateTimer()
 	require.NotNil(t, createTimerAction, "Expected the scheduled action to be a timer")
 	require.WithinDuration(t, startEvent.Timestamp.AsTime().Add(timerDuration), createTimerAction.FireAt.AsTime(), 0)
 	require.Equal(t, "MyEvent", createTimerAction.GetName())
@@ -109,5 +109,5 @@ func Test_Executor_SuspendStopsAllActions(t *testing.T) {
 	// Execute the orchestrator function and expect to get back no actions
 	results, err := executor.ExecuteOrchestrator(ctx, iid, oldEvents, newEvents)
 	require.NoError(t, err)
-	require.Empty(t, results.Response.Actions, "Suspended orchestrations should not have any actions")
+	require.Empty(t, results.Actions, "Suspended orchestrations should not have any actions")
 }

tests/worker_test.go

@@ -46,7 +46,7 @@ func Test_TryProcessSingleOrchestrationWorkItem_BasicFlow(t *testing.T) {
 		},
 	}
 	state := &backend.OrchestrationRuntimeState{}
-	result := &backend.ExecutionResults{Response: &protos.OrchestratorResponse{}}
+	result := &protos.OrchestratorResponse{}
 
 	ctx, cancel := context.WithCancel(ctx)
 	completed := atomic.Bool{}
@@ -116,16 +116,14 @@ func Test_TryProcessSingleOrchestrationWorkItem_ExecutionStartedAndCompleted(t *
 
 	// Return an execution completed action to simulate the completion of the orchestration (a no-op)
 	resultValue := "done"
-	result := &backend.ExecutionResults{
-		Response: &protos.OrchestratorResponse{
-			Actions: []*protos.OrchestratorAction{
-				{
-					Id: -1,
-					OrchestratorActionType: &protos.OrchestratorAction_CompleteOrchestration{
-						CompleteOrchestration: &protos.CompleteOrchestrationAction{
-							OrchestrationStatus: protos.OrchestrationStatus_ORCHESTRATION_STATUS_COMPLETED,
-							Result:              wrapperspb.String(resultValue),
-						},
+	result := &protos.OrchestratorResponse{
+		Actions: []*protos.OrchestratorAction{
+			{
+				Id: -1,
+				OrchestratorActionType: &protos.OrchestratorAction_CompleteOrchestration{
+					CompleteOrchestration: &protos.CompleteOrchestrationAction{
+						OrchestrationStatus: protos.OrchestrationStatus_ORCHESTRATION_STATUS_COMPLETED,
+						Result:              wrapperspb.String(resultValue),
 					},
 				},
 			},