Function TestTaskManager

internal/controller/executor/executor_test.go:68–102

(t *testing.T)

}

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

	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)
			}
			if o != nil {
				o(&proto.AgentOutputs{
					Messages: []*proto.Message{text("assistant", "root done")},
				})
			}
		}),
		"child": AgentFunc(func(inputs []*proto.Message, tm agent.Executor, o agent.OutputHandler) {
			time.Sleep(100 * time.Millisecond)
			if o != nil {
				o(&proto.AgentOutputs{
					Messages: []*proto.Message{text("assistant", "child done")},
				})
			}
		}),
	}

	tm := DefaultExecutor(memoryEventLog(), registry)
	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)
	}
}

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