Function TestExecuteWithError

pkg/executionplan/executor_test.go:190–232

(t *testing.T)

}

func TestExecuteWithError(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("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 = true

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

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

	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 != StepStatusSkipped {
		t.Errorf("step 2: expected %v, got %v", StepStatusSkipped, plan.Steps[2].Status)
	}

	if tool3.ExecutionCount() != 0 {
		t.Errorf("tool3 should not have been executed")
	}
}

func TestExecuteContinueOnError(t *testing.T) {
	testErr := errors.New("test error")