Function advancedTraceQLRunner

tempodb/tempodb_search_test.go:166–341

(t *testing.T, wantTr *tempopb.Trace, wantMeta *tempopb.TraceSearchMetadata, _, _ []*tempopb.SearchRequest, meta *backend.BlockMeta, r Reader, _ common.BackendBlock)

}

func advancedTraceQLRunner(t *testing.T, wantTr *tempopb.Trace, wantMeta *tempopb.TraceSearchMetadata, _, _ []*tempopb.SearchRequest, meta *backend.BlockMeta, r Reader, _ common.BackendBlock) {
	ctx := context.Background()
	e := traceql.NewEngine()

	// collect some info about wantTr to use below
	trueConditionsBySpan := [][]string{}
	durationBySpan := []uint64{}
	falseConditions := []string{
		fmt.Sprintf("name=`%v`", test.RandomString()),
		fmt.Sprintf("duration>%dh", rand.Intn(10)+1),
		fmt.Sprintf("rootServiceName=`%v`", test.RandomString()),
		// status? can't really construct a status condition that's false for all spans
	}
	trueTraceC := []string{
		fmt.Sprintf("traceDuration=%dms", wantMeta.DurationMs),
		fmt.Sprintf("rootServiceName=`%s`", wantMeta.RootServiceName),
		fmt.Sprintf("rootName=`%s`", wantMeta.RootTraceName),
	}
	totalSpans := 0
	for _, b := range wantTr.ResourceSpans {
		trueResourceC, falseResourceC := conditionsForAttributes(b.Resource.Attributes, "resource")
		falseConditions = append(falseConditions, falseResourceC...)

		for _, ss := range b.ScopeSpans {
			totalSpans += len(ss.Spans)
			for _, s := range ss.Spans {
				trueC, falseC := conditionsForAttributes(s.Attributes, "span")

				status := trace.StatusToString(s.Status.Code)
				kind := trace.KindToString(s.Kind)
				trueC = append(trueC, fmt.Sprintf("name=`%v`", s.Name))
				trueC = append(trueC, fmt.Sprintf("duration=%dns", s.EndTimeUnixNano-s.StartTimeUnixNano))
				trueC = append(trueC, fmt.Sprintf("status=%s", status))
				trueC = append(trueC, fmt.Sprintf("kind=%s", kind))
				trueC = append(trueC, trueResourceC...)
				trueC = append(trueC, trueTraceC...)

				trueConditionsBySpan = append(trueConditionsBySpan, trueC)
				falseConditions = append(falseConditions, falseC...)
				durationBySpan = append(durationBySpan, s.EndTimeUnixNano-s.StartTimeUnixNano)
			}
		}
	}

	rando := func(s []string) string {
		return s[rand.Intn(len(s))]
	}

	searchesThatMatch := []*tempopb.SearchRequest{
		{Query: "{}"},
		// conditions
		{Query: fmt.Sprintf("{%s && %s && %s && %s && %s}", rando(trueConditionsBySpan[0]), rando(trueConditionsBySpan[0]), rando(trueConditionsBySpan[0]), rando(trueConditionsBySpan[0]), rando(trueConditionsBySpan[0]))},
		{Query: fmt.Sprintf("{%s || %s || %s || %s || %s}", rando(falseConditions), rando(falseConditions), rando(falseConditions), rando(trueConditionsBySpan[0]), rando(falseConditions))},
		{Query: fmt.Sprintf("{(%s && %s) || %s}", rando(falseConditions), rando(falseConditions), rando(trueConditionsBySpan[0]))},
		// spansets
		{Query: fmt.Sprintf("{%s} && {%s}", rando(trueConditionsBySpan[0]), rando(trueConditionsBySpan[1]))},
		{Query: fmt.Sprintf("{%s} || {%s}", rando(trueConditionsBySpan[0]), rando(falseConditions))},
		{Query: fmt.Sprintf("{%s} && {%s} && {%s} && {%s} && {%s}", rando(trueConditionsBySpan[0]), rando(trueConditionsBySpan[0]), rando(trueConditionsBySpan[0]), rando(trueConditionsBySpan[0]), rando(trueConditionsBySpan[0]))},