Function TestCompareScalesResults

pkg/traceql/engine_metrics_compare_test.go:157–281

(t *testing.T)

}

func TestCompareScalesResults(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" })`,
	}

	a := newMetricsCompare(newSpansetFilter(
		newBinaryOperation(OpEqual,
			NewAttribute("service"),
			NewStaticString("selected"),
		)), 10, 0, 0)
	a.init(req, AggregateModeRaw)

	// 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"),
	}

	for _, span := range spans {
		a.observe(span)
	}

	// Double all counts
	ss := a.result(2.0)

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