Function TestFirstIndexInBudget

pkg/compaction/compaction_test.go:314–362

(t *testing.T)

}

func TestFirstIndexInBudget(t *testing.T) {
	t.Parallel()

	msg := func(role chat.MessageRole, content string) chat.Message {
		return chat.Message{Role: role, Content: content}
	}

	tests := []struct {
		name      string
		messages  []chat.Message
		budget    int64
		wantFirst int
	}{
		{
			name:      "empty",
			messages:  nil,
			budget:    1000,
			wantFirst: 0,
		},
		{
			name: "everything fits",
			messages: []chat.Message{
				msg(chat.MessageRoleUser, "short"),
				msg(chat.MessageRoleAssistant, "short"),
			},
			budget:    1000,
			wantFirst: 0,
		},
		{
			name: "tight budget keeps tail starting on a user/assistant turn",
			messages: []chat.Message{
				msg(chat.MessageRoleUser, strings.Repeat("a", 4000)),
				msg(chat.MessageRoleAssistant, strings.Repeat("b", 4000)),
				msg(chat.MessageRoleUser, strings.Repeat("c", 4000)),
				msg(chat.MessageRoleAssistant, strings.Repeat("d", 4000)),
			},
			budget:    2100, // ~2 messages worth
			wantFirst: 2,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()
			got := FirstIndexInBudget(tt.messages, tt.budget)
			assert.Equal(t, tt.wantFirst, got)
		})
	}
}