Function TestSamplingError

tempodb/encoding/vparquet5/block_traceql_test.go:1906–2014

(t *testing.T)

}

func TestSamplingError(t *testing.T) {
	// Comment this out to run tests when developing.
	t.Skip("Skipping sampling error test")

	testQueries := []string{
		"{} | rate()",                            // Simple rate, most amount of data
		"{} | rate() by (resource.service.name)", // Same but with subseries
		"{status=error} | rate()",                // Somewhat rare
		"{resource.service.name=`loki-querier` && name=`chunksmemcache.store`} | rate()", // Very rare
		"{nestedSetParent=-1} >> {}| rate()",                                             // Structural (common)
		"{nestedSetParent=-1} >> {status=error} | rate()",                                // Structural (rare)
		"{} | quantile_over_time(duration, .99)",                                         // Quantile (common)
		"{resource.service.name=`tempo-gateway`} | quantile_over_time(duration, .99)",    // Quantile (rare)

		// Drilldown queries
		/*"{nestedSetParent<0 && true && status=error} | rate()",
		"{nestedSetParent<0 && true && resource.service.name != nil} | rate() by(resource.service.name)",
		"{nestedSetParent<0 && true} | histogram_over_time(duration)",
		`{nestedSetParent<0 && resource.service.name="gme-alertmanager" && resource.service.namespace != nil} | rate() by(resource.service.namespace)`,
		"{true && true && resource.service.name != nil} | rate() by(resource.service.name)",*/
	}

	options := []string{
		"",            // Control, no sampling
		"sample=true", // Automatic sampling
		"sample=0.01", // Aggressive sampling. Automatic should be better accuracy.
		"sample=0.1",  // Decent sampling. Automatic should be roughly equal.
		"sample=0.5",  // Lowest amount of sampling possible as whole fraction. Automatic should be better performance.
	}

	e := traceql.NewEngine()
	ctx := context.TODO()
	opts := common.DefaultSearchOptions()
	opts.TotalPages = 1

	block := blockForBenchmarks(t)
	_, _, err := block.openForSearch(ctx, opts)
	require.NoError(t, err)

	f := traceql.NewSpansetFetcherWrapperBoth(
		func(ctx context.Context, req traceql.FetchSpansRequest) (traceql.FetchSpansResponse, error) {
			return block.Fetch(ctx, req, opts)
		},
		func(ctx context.Context, req traceql.FetchSpansRequest) (traceql.FetchSpansOnlyResponse, error) {
			return block.FetchSpans(ctx, req, opts)
		},
	)

	executeQuery := func(t *testing.T, q string) (results traceql.SeriesSet, spanCount int) {
		st := uint64(block.meta.StartTime.UnixNano())
		end := uint64(block.meta.EndTime.UnixNano())

		req := &tempopb.QueryRangeRequest{
			Query:     q,
			Start:     st,
			End:       end,
			Step:      uint64(time.Second * 15),
			MaxSeries: 1000,