Function TestBackendBlockSelectAll

tempodb/encoding/vparquet5/block_traceql_test.go:1418–1489

(t *testing.T)

}

func TestBackendBlockSelectAll(t *testing.T) {
	var (
		ctx          = context.Background()
		numTraces    = 250
		traces       = make([]*Trace, 0, numTraces)
		wantTraceIdx = rand.Intn(numTraces)
		wantTraceID  = test.ValidTraceID(nil)
		wantTrace    = fullyPopulatedTestTrace(wantTraceID)
		dc           = test.MakeDedicatedColumns()
		dcm          = dedicatedColumnsToColumnMapping(dc)
	)

	// TODO - This strips unsupported attributes types for now. Revisit when
	// add support for arrays/kvlists in the fetch layer.
	trimForSelectAll(wantTrace)

	for i := 0; i < numTraces; i++ {
		if i == wantTraceIdx {
			traces = append(traces, wantTrace)
			continue
		}

		id := test.ValidTraceID(nil)
		tr, _ := traceToParquet(&backend.BlockMeta{}, id, test.MakeTrace(1, id), nil)
		traces = append(traces, tr)
	}

	b := makeBackendBlockWithTraces(t, traces)

	_, eval, _, _, req, err := traceql.Compile("{}")
	require.NoError(t, err)
	req.SecondPass = func(inSS *traceql.Spanset) ([]*traceql.Spanset, error) { return eval([]*traceql.Spanset{inSS}) }
	req.SecondPassSelectAll = true

	resp, err := b.Fetch(ctx, *req, common.DefaultSearchOptions())
	require.NoError(t, err)
	defer resp.Results.Close()

	// This is a dump of all spans in the fully-populated test trace
	wantSS := flattenForSelectAll(wantTrace, dcm)

	for {
		// Seek to our desired trace
		ss, err := resp.Results.Next(ctx)
		require.NoError(t, err)
		if ss == nil {
			break
		}
		if !bytes.Equal(ss.TraceID, wantTraceID) {
			continue
		}

		// Cleanup found data for comparison
		// equal will fail on the rownum mismatches. this is an internal detail to the
		// fetch layer. just wipe them out here
		ss.ReleaseFn = nil
		ss.ServiceStats = nil
		for _, sp := range ss.Spans {