Function TestSearchByShortTraceID

tempodb/tempodb_search_test.go:2625–2764

(t *testing.T)

}

func TestSearchByShortTraceID(t *testing.T) {
	for _, v := range encoding.AllEncodingsForWrites() {
		blockVersion := v.Version()

		tempDir := t.TempDir()
		traceID := make([]byte, 8) // short trace ID
		_, err := crand.Read(traceID)
		require.NoError(t, err)
		traceID = append(make([]byte, 8), traceID...) // adding leading zeros to make it 16 bytes

		r, w, c, err := New(testingConfig(tempDir, blockVersion, nil), nil, log.NewNopLogger())
		require.NoError(t, err)

		err = c.EnableCompaction(context.Background(), testingCompactorConfig, &mockSharder{}, &mockOverrides{})
		require.NoError(t, err)

		ctx := t.Context()
		r.EnablePolling(ctx, &mockJobSharder{}, false)
		wantID, wantTr, start, end, wantMeta := makeExpectedTrace(traceID)

		// Write to wal
		wal := w.WAL()

		meta := &backend.BlockMeta{BlockID: backend.NewUUID(), TenantID: testTenantID}
		head, err := wal.NewBlock(meta, model.CurrentEncoding)
		require.NoError(t, err)
		dec := model.MustNewSegmentDecoder(model.CurrentEncoding)

		totalTraces := 50
		wantTrIdx := rand.Intn(totalTraces) // nolint:gosec // G404: Use of weak random number generator
		for i := 0; i < totalTraces; i++ {
			var tr *tempopb.Trace
			var id []byte
			if i == wantTrIdx {
				tr = wantTr
				id = wantID
			} else {
				id = test.ValidTraceID(nil)
				tr = test.MakeTrace(10, id)
			}
			b1, err := dec.PrepareForWrite(tr, start, end)
			require.NoError(t, err)

			b2, err := dec.ToObject([][]byte{b1})
			require.NoError(t, err)
			err = head.Append(id, b2, start, end, true)
			require.NoError(t, err)
		}

		// Complete block
		block, err := w.CompleteBlock(context.Background(), head)
		require.NoError(t, err)

		fetcher := traceql.NewSpansetFetcherWrapper(func(ctx context.Context, req traceql.FetchSpansRequest) (traceql.FetchSpansResponse, error) {
			return block.Fetch(ctx, req, common.DefaultSearchOptions())
		})

		for name, traceID := range map[string]string{