Function TestMetrics_PushBytesTracking

modules/livestore/metrics_test.go:107–149

(t *testing.T)

}

func TestMetrics_PushBytesTracking(t *testing.T) {
	setup := setupTest(t)
	defer setup.cleanup()

	// Create a trace and push it
	traceID := test.ValidTraceID(nil)
	trace := test.MakeTrace(10, traceID)
	traceData, err := trace.Marshal()
	require.NoError(t, err)

	req := &tempopb.PushBytesRequest{
		Traces: []tempopb.PreallocBytes{{Slice: traceData}},
		Ids:    [][]byte{traceID},
	}

	expectedBytes := float64(len(traceData))

	// Record initial values to ensure we're measuring the delta correctly
	initialBytesReceived, err := getCounterVecValue(metricBytesReceivedTotal, testTenant, "trace")
	require.NoError(t, err)
	initialTracesCreated, err := test.GetCounterValue(setup.instance.tracesCreatedTotal)
	require.NoError(t, err)

	// Push bytes
	setup.instance.pushBytes(t.Context(), time.Now(), req)

	// Verify bytes received metric increased by expected amount
	finalBytesReceived, err := getCounterVecValue(metricBytesReceivedTotal, testTenant, "trace")
	require.NoError(t, err)
	assert.Equal(t, initialBytesReceived+expectedBytes, finalBytesReceived,
		"bytes received should increase by trace data size")

	// Check live traces metric after cutting
	drained, err := setup.instance.cutIdleTraces(t.Context(), true) // immediate cut
	require.NoError(t, err)
	require.True(t, drained, "should drain live traces in one iteration")

	// Verify traces were created
	finalTracesCreated, err := test.GetCounterValue(setup.instance.tracesCreatedTotal)
	require.NoError(t, err)
	assert.Equal(t, initialTracesCreated+1.0, finalTracesCreated,
		"one trace should be created after cutting")
}

func TestMetrics_CompletionFlow(t *testing.T) {
	setup := setupTest(t)