Function TestExecuteContinueOnError

pkg/executionplan/executor_test.go:234–281

(t *testing.T)

}

func TestExecuteContinueOnError(t *testing.T) {
	testErr := errors.New("test error")
	tool1 := newMockTool("tool1", "result1", nil)
	tool2 := newMockTool("tool2", nil, testErr)
	tool3 := newMockTool("tool3", "result3", nil)

	toolMap := map[string]tools.Tool{
		"tool1": tool1,
		"tool2": tool2,
		"tool3": tool3,
	}

	executor := NewExecutor(toolMap)

	plan := NewExecutionPlan("Continue on error test")
	plan.AddStep("tool1", "Step 1", nil)
	plan.AddStep("tool2", "Step 2", nil)
	plan.AddStep("tool3", "Step 3", nil)
	plan.Options.RequireApproval = false
	plan.Options.StopOnError = false

	ctx := context.Background()
	toolCtx := &tools.ToolContext{AgentID: "test-agent"}

	err := executor.Execute(ctx, plan, toolCtx)
	if err == nil {
		t.Fatal("expected error to be returned")
	}

	if plan.Steps[0].Status != StepStatusCompleted {
		t.Errorf("step 0: expected %v, got %v", StepStatusCompleted, plan.Steps[0].Status)
	}
	if plan.Steps[1].Status != StepStatusFailed {
		t.Errorf("step 1: expected %v, got %v", StepStatusFailed, plan.Steps[1].Status)
	}
	if plan.Steps[2].Status != StepStatusCompleted {
		t.Errorf("step 2: expected %v, got %v", StepStatusCompleted, plan.Steps[2].Status)
	}

	// Tool3 should still execute
	if tool3.ExecutionCount() != 1 {
		t.Errorf("tool3 should have been executed once, got %d", tool3.ExecutionCount())
	}

	if plan.Status != StatusPartial {
		t.Errorf("expected status %v, got %v", StatusPartial, plan.Status)
	}
}

func TestExecuteParallel(t *testing.T) {
	// Use delays to verify parallel execution