Function TestController_Exec_LastSeq

internal/controller/controller_test.go:178–237

(t *testing.T)

}

func TestController_Exec_LastSeq(t *testing.T) {
	ctx := context.Background()
	cid := "test-conv-seq"

	log := &executortest.MemoryEventLog{}
	// Pre-populate history
	log.AllEvents = []*proto.ConversationEvent{
		{
			ConversationId: cid,
			Seq:            1,
			Messages: []*proto.Message{
				{Role: "user", Content: &proto.Content{Type: &proto.Content_Text{Text: &proto.TextContent{Text: "msg 1"}}}},
			},
			State: proto.State_STATE_COMPLETED,
		},
		{
			ConversationId: cid,
			Seq:            2,
			Messages: []*proto.Message{
				{Role: "assistant", Content: &proto.Content{Type: &proto.Content_Text{Text: &proto.TextContent{Text: "msg 2"}}}},
			},
			State: proto.State_STATE_COMPLETED,
		},
	}

	c, err := New(ctx, Config{
		EventLogBuilder: func() (executor.EventLog, error) {
			return log, nil
		},
		PlannerBuilder: func(ctx context.Context, r *Registry) (agent.Agent, error) {
			return &dummyAgent{}, nil
		},
	})
	if err != nil {
		t.Fatal(err)
	}
	defer c.Close()

	var msgs []*proto.Message
	handler := ExecHandler(func(resp *proto.ExecResponse) error {
		msgs = append(msgs, resp.Outputs...)
		return nil
	})

	err = c.Exec(ctx, &proto.ExecRequest{
		ConversationId: cid,
		LastSeq:        1,
	}, handler)
	if err != nil {
		t.Fatal(err)
	}

	// We expect to receive messages from Seq 2 (since LastSeq is 1).
	if len(msgs) != 1 {
		t.Fatalf("expected 1 message, got %d", len(msgs))
	}
	if msgs[0].GetContent().GetText().GetText() != "msg 2" {
		t.Fatalf("expected 'msg 2', got %v", msgs[0].GetContent().GetText().GetText())