Function TestInstanceSearchMetrics

modules/livestore/instance_search_test.go:868–925

(t *testing.T)

}

func TestInstanceSearchMetrics(t *testing.T) {
	t.Parallel()
	i, ls := defaultInstance(t)

	numTraces := uint32(500)
	numBytes := uint64(0)
	for j := uint32(0); j < numTraces; j++ {
		id := test.ValidTraceID(nil)

		// Trace bytes have to be pushed as raw tempopb.Trace bytes
		trace := test.MakeTrace(10, id)

		traceBytes, err := trace.Marshal()
		require.NoError(t, err)

		// Create a push request for livestore
		req := &tempopb.PushBytesRequest{
			Traces: []tempopb.PreallocBytes{{Slice: traceBytes}},
			Ids:    [][]byte{id},
		}
		i.pushBytes(t.Context(), time.Now(), req)
	}

	search := func() *tempopb.SearchMetrics {
		sr, err := i.Search(t.Context(), &tempopb.SearchRequest{
			Query: fmt.Sprintf(`{ span.%s = "%s" }`, "foo", "bar"),
		})
		require.NoError(t, err)
		return sr.Metrics
	}

	// Live traces
	m := search()
	require.Equal(t, uint32(0), m.InspectedTraces) // we don't search live traces
	require.Equal(t, uint64(0), m.InspectedBytes)  // we don't search live traces

	// Test after appending to WAL
	drained, err := i.cutIdleTraces(t.Context(), true)
	require.NoError(t, err)
	require.True(t, drained, "should drain live traces in one iteration")
	m = search()
	require.Less(t, numBytes, m.InspectedBytes)

	// Test after cutting new headblock
	blockID, err := i.cutBlocks(t.Context(), true)
	require.NoError(t, err)
	m = search()
	require.Less(t, numBytes, m.InspectedBytes)

	// Test after completing a block
	_, err = i.completeBlock(t.Context(), blockID)
	require.NoError(t, err)
	m = search()
	require.Less(t, numBytes, m.InspectedBytes)

	err = services.StopAndAwaitTerminated(t.Context(), ls)
	require.NoError(t, err)
}