Function TestAvgOverTimeForDuration

pkg/traceql/engine_metrics_test.go:1427–1466

(t *testing.T)

}

func TestAvgOverTimeForDuration(t *testing.T) {
	req := &tempopb.QueryRangeRequest{
		Start: 1,
		End:   uint64(3 * time.Second),
		Step:  uint64(1 * time.Second),
		Query: "{ } | avg_over_time(duration) by (span.foo)",
	}

	// A variety of spans across times, durations, and series. All durations are powers of 2 for simplicity
	in := []Span{
		newMockSpan(nil).WithStartTime(uint64(1*time.Second)).WithSpanString("foo", "bar").WithDuration(100),
		newMockSpan(nil).WithStartTime(uint64(1*time.Second)).WithSpanString("foo", "bar").WithDuration(100),
		newMockSpan(nil).WithStartTime(uint64(1*time.Second)).WithSpanString("foo", "bar").WithDuration(100),

		newMockSpan(nil).WithStartTime(uint64(2*time.Second)).WithSpanString("foo", "bar").WithDuration(100),
		newMockSpan(nil).WithStartTime(uint64(2*time.Second)).WithSpanString("foo", "bar").WithDuration(100),
		newMockSpan(nil).WithStartTime(uint64(2*time.Second)).WithSpanString("foo", "bar").WithDuration(100),
		newMockSpan(nil).WithStartTime(uint64(2*time.Second)).WithSpanString("foo", "bar").WithDuration(500),

		newMockSpan(nil).WithStartTime(uint64(3*time.Second)).WithSpanString("foo", "baz").WithDuration(100),
		newMockSpan(nil).WithStartTime(uint64(3*time.Second)).WithSpanString("foo", "baz").WithDuration(200),
		newMockSpan(nil).WithStartTime(uint64(3*time.Second)).WithSpanString("foo", "baz").WithDuration(300),
	}

	result, seriesCount, err := runTraceQLMetric(req, in)
	require.NoError(t, err)
	require.Equal(t, len(result), seriesCount)

	fooBaz := result[LabelsFromArgs("span.foo", "baz").MapKey()]
	fooBar := result[LabelsFromArgs("span.foo", "bar").MapKey()]

	// We cannot compare with require.Equal because NaN != NaN
	assert.True(t, math.IsNaN(fooBaz.Values[0]))
	assert.True(t, math.IsNaN(fooBaz.Values[1]))
	assert.Equal(t, 200., fooBaz.Values[2]*float64(time.Second))

	assert.Equal(t, 100., fooBar.Values[0]*float64(time.Second))
	assert.Equal(t, 200., fooBar.Values[1]*float64(time.Second))
	assert.True(t, math.IsNaN(fooBar.Values[2]))
}

func TestAvgOverTimeForDurationWithSecondStage(t *testing.T) {
	req := &tempopb.QueryRangeRequest{