Function TestOnStepFailedCallback

pkg/executionplan/executor_test.go:668–702

(t *testing.T)

}

func TestOnStepFailedCallback(t *testing.T) {
	testErr := errors.New("test error")
	tool1 := newMockTool("tool1", nil, testErr)

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

	var failedStep *Step
	var failedErr error

	executor := NewExecutor(
		toolMap,
		WithOnStepFailed(func(p *ExecutionPlan, s *Step, err error) {
			failedStep = s
			failedErr = err
		}),
	)

	plan := NewExecutionPlan("Failed callback test")
	plan.AddStep("tool1", "Step 1", nil)
	plan.Options.RequireApproval = false

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

	_ = executor.Execute(ctx, plan, toolCtx)

	if failedStep == nil {
		t.Fatal("OnStepFailed callback not called")
	}
	if failedErr == nil || failedErr.Error() != testErr.Error() {
		t.Errorf("expected error %v, got %v", testErr, failedErr)
	}
}

func TestExecuteWithInputParameters(t *testing.T) {
	tool1 := newMockTool("tool1", "result", nil)