Function TestMaxTraceSizeExceededWithAccumulation

pkg/livetraces/livetraces_test.go:120–151

(t *testing.T)

}

func TestMaxTraceSizeExceededWithAccumulation(t *testing.T) {
	var buf bytes.Buffer
	logger := kitlog.NewJSONLogger(kitlog.NewSyncWriter(&buf))

	originalLogger := log.Logger
	log.Logger = logger
	defer func() { log.Logger = originalLogger }()

	const (
		batchSize    = uint64(300)
		maxTraceSize = uint64(500)
	)

	lt := New(func(_ *v1.ResourceSpans) uint64 { return batchSize }, time.Hour, time.Hour, testTenantID)

	id := test.ValidTraceID(nil)
	tr := test.MakeTrace(1, id)

	// First push should succeed
	lt.Push(id, tr.ResourceSpans[0], 0)

	// Second push should fail: batchSize + batchSize > maxTraceSize
	err := lt.PushWithTimestampAndLimits(time.Now(), id, tr.ResourceSpans[0], 0, maxTraceSize)

	require.Equal(t, ErrMaxTraceSizeExceeded, err)

	logOutput := buf.String()
	require.Contains(t, logOutput, "max trace size exceeded")
	require.Contains(t, logOutput, fmt.Sprintf(`"max":%d`, maxTraceSize))
	require.Contains(t, logOutput, fmt.Sprintf(`"reqSize":%d`, batchSize))
	require.Contains(t, logOutput, fmt.Sprintf(`"totalSize":%d`, batchSize))
}

func BenchmarkLiveTracesWrite(b *testing.B) {
	lt := New(func(rs *v1.ResourceSpans) uint64 { return uint64(rs.Size()) }, 0, 0, testTenantID)