Function TestCache_TrimSpaces

pkg/runtime/cache_test.go:121–152

TestCache_TrimSpaces verifies that whitespace trimming is applied when TrimSpaces is enabled.

(t *testing.T)

// TestCache_TrimSpaces verifies that whitespace trimming is applied when
// TrimSpaces is enabled.
func TestCache_TrimSpaces(t *testing.T) {
	t.Parallel()

	c, err := cache.New(cache.Config{Enabled: true, TrimSpaces: true})
	require.NoError(t, err)

	stream := newStreamBuilder().
		AddContent("Trimmed!").
		AddStopWithUsage(5, 3).
		Build()
	prov := &messageRecordingProvider{
		id:      "test/mock-model",
		streams: []*mockStream{stream},
	}

	// First turn: question with surrounding whitespace.
	sess1 := session.New(session.WithUserMessage("  Hello  "))
	runWithCache(t, c, prov, sess1)

	prov.mu.Lock()
	require.Len(t, prov.recordedMessages, 1)
	prov.mu.Unlock()

	// Second turn: same question without whitespace must hit the cache.
	sess2 := session.New(session.WithUserMessage("Hello"))
	runWithCache(t, c, prov, sess2)

	prov.mu.Lock()
	defer prov.mu.Unlock()
	assert.Len(t, prov.recordedMessages, 1, "trim-enabled cache must hit when whitespace differs")
	assert.True(t, hasAssistantMessage(sess2, "Trimmed!"))
}

// TestCache_DisabledHasNoEffect verifies that when the agent has no cache
// attached, every call goes through to the model.