Function TestConfirmation

internal/controller/executor/executor_test.go:178–262

(t *testing.T)

}

func TestConfirmation(t *testing.T) {
	ctx := context.Background()

	var runCount int
	eventLog := memoryEventLog()

	confID := "test-conf-id"
	var childDone atomic.Bool
	registry := map[string]agent.Agent{
		"root": AgentFunc(func(inputs []*proto.Message, tm agent.Executor, o agent.OutputHandler) {
			if _, err := tm.Exec(ctx, "test-conv", "child-task", &proto.AgentStart{
				AgentId:  "child",
				Messages: inputs,
			}, o); err != nil {
				t.Fatal(err)
			}
		}),
		"child": AgentFunc(func(inputs []*proto.Message, tm agent.Executor, o agent.OutputHandler) {
			if runCount == 0 {
				runCount++
				log.Println("Asking for the question...")
				if o != nil {
					o(&proto.AgentOutputs{
						Messages: []*proto.Message{{
							Role: "model",
							Content: &proto.Content{
								Type: &proto.Content_Confirmation{
									Confirmation: &proto.ConfirmationContent{Id: confID, Question: "proceed?"},
								},
							},
						}},
					})
				}
				return
			}

			lastInput := inputs[len(inputs)-1]
			if lastInput.GetContent().GetConfirmation() == nil || lastInput.GetContent().GetConfirmation().GetDecision() == nil {
				t.Fatal("no decision in the incoming inputs")
			}

			childDone.Store(true)
			if o != nil {
				o(&proto.AgentOutputs{
					Messages: []*proto.Message{text("assistant", "child done")},
				})
			}
		}),
	}

	tm := DefaultExecutor(eventLog, registry)

	// First run: child returns a confirmation request.
	if _, err := tm.Exec(ctx, "test-conv", "root-task", &proto.AgentStart{
		AgentId:  "root",
		Messages: []*proto.Message{text("user", "hello!")},
	}, nil); err != nil {
		t.Fatal(err)