Function TestCompare

pkg/traceql/engine_metrics_compare_test.go:11–124

(t *testing.T)

)

func TestCompare(t *testing.T) {
	// Test that the compare function correctly multiplies results based on sampling multiplier
	// The multiplication happens in result() method: s.Values[i] *= multiplier

	// Instant query
	req := &tempopb.QueryRangeRequest{
		Start: 1,
		End:   uint64(3 * time.Second),
		Step:  uint64(2 * time.Second),
		Query: `{ } | compare({ .service="selected" })`,
	}

	// Test data with clear baseline vs selection distinction
	spans := []Span{
		// Baseline spans (service != "selected")
		newMockSpan(nil).WithStartTime(uint64(1*time.Second)).WithSpanString("service", "baseline1").WithSpanString("environment", "prod"),
		newMockSpan(nil).WithStartTime(uint64(1*time.Second)).WithSpanString("service", "baseline2").WithSpanString("environment", "dev"),
		newMockSpan(nil).WithStartTime(uint64(2*time.Second)).WithSpanString("service", "baseline1").WithSpanString("environment", "prod"),

		// Selection spans (service == "selected")
		newMockSpan(nil).WithStartTime(uint64(1*time.Second)).WithSpanString("service", "selected").WithSpanString("environment", "prod"),
		newMockSpan(nil).WithStartTime(uint64(2*time.Second)).WithSpanString("service", "selected").WithSpanString("environment", "dev"),
		newMockSpan(nil).WithStartTime(uint64(2*time.Second)).WithSpanString("service", "selected").WithSpanString("environment", "dev"),
		newMockSpan(nil).WithStartTime(uint64(3*time.Second)).WithSpanString("service", "selected").WithSpanString("environment", "prod"),
	}

	ss, _, err := runTraceQLMetric(req, spans)
	require.NoError(t, err)

	expected := []TimeSeries{
		// Baseline values
		{
			Labels: Labels{
				internalLabelTypeBaseline,
				Label{Name: "span.service", Value: NewStaticString("baseline1")},
			},
			Values: []float64{2, 0},
		},
		{
			Labels: Labels{
				internalLabelTypeBaseline,
				Label{Name: "span.service", Value: NewStaticString("baseline2")},
			},
			Values: []float64{1, 0},
		},
		{
			Labels: Labels{
				internalLabelTypeBaseline,
				Label{Name: "span.environment", Value: NewStaticString("prod")},
			},
			Values: []float64{2, 0},
		},
		{
			Labels: Labels{
				internalLabelTypeBaseline,
				Label{Name: "span.environment", Value: NewStaticString("dev")},
			},
			Values: []float64{1, 0},