Function TestExecutorWithCallbacks

pkg/executionplan/executor_test.go:87–140

(t *testing.T)

}

func TestExecutorWithCallbacks(t *testing.T) {
	toolMap := map[string]tools.Tool{
		"tool1": newMockTool("tool1", "result1", nil),
	}

	var stepStartCalled, stepCompleteCalled, planCompleteCalled bool
	var startedPlan, completedPlan *ExecutionPlan
	var startedStep, completedStep *Step

	executor := NewExecutor(
		toolMap,
		WithOnStepStart(func(p *ExecutionPlan, s *Step) {
			stepStartCalled = true
			startedPlan = p
			startedStep = s
		}),
		WithOnStepComplete(func(p *ExecutionPlan, s *Step) {
			stepCompleteCalled = true
			completedPlan = p
			completedStep = s
		}),
		WithOnPlanComplete(func(p *ExecutionPlan) {
			planCompleteCalled = true
		}),
	)

	plan := NewExecutionPlan("Test plan")
	plan.AddStep("tool1", "Test step", nil)
	plan.Options.RequireApproval = false

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

	err := executor.Execute(ctx, plan, toolCtx)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}

	if !stepStartCalled {
		t.Error("OnStepStart callback not called")
	}
	if !stepCompleteCalled {
		t.Error("OnStepComplete callback not called")
	}
	if !planCompleteCalled {
		t.Error("OnPlanComplete callback not called")
	}
	if startedPlan == nil || startedStep == nil {
		t.Error("start callback received nil plan or step")
	}
	if completedPlan == nil || completedStep == nil {
		t.Error("complete callback received nil plan or step")
	}
}

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