Function TestResumeConfirmation

internal/controller/executor/executor_test.go:338–454

(t *testing.T)

}

func TestResumeConfirmation(t *testing.T) {
	ctx := context.Background()
	eventLog := memoryEventLog()

	msg := &proto.Message{
		Content: &proto.Content{
			Type: &proto.Content_Confirmation{
				Confirmation: &proto.ConfirmationContent{
					Id:       "test-conf-id",
					Question: "proceed?",
				},
			},
		},
	}

	var runCount int
	registry := map[string]agent.Agent{
		"root": AgentFunc(func(inputs []*proto.Message, tm agent.Executor, o agent.OutputHandler) {
			if conf := historyutil.WaitsForConfirmation(inputs); conf != nil {
				o(&proto.AgentOutputs{
					Messages: []*proto.Message{msg},
				})
				return
			}

			approved, _ := historyutil.HasConfirmationAnswer(inputs)
			if approved {
				if err := o(&proto.AgentOutputs{
					// "Hello!" is responded with a confirmation request.
					Messages: []*proto.Message{{
						Content: &proto.Content{
							Type: &proto.Content_Text{
								Text: &proto.TextContent{
									Text: "awesome!",
								},
							},
						},
					}},
				}); err != nil {
					t.Fatal(err)
				}
			}

			if runCount == 0 {
				runCount++
				if err := o(&proto.AgentOutputs{
					// "Hello!" is responded with a confirmation request.
					Messages: []*proto.Message{msg},
				}); err != nil {
					t.Fatal(err)
				}
			}
		}),
	}

	tm := DefaultExecutor(eventLog, registry)

	// First run: starts as "root"