Function TestCountOverTimeInstantNsWithCutoff

pkg/traceql/engine_metrics_test.go:1055–1167

TestCountOverTimeInstantNsWithCutoff simulates merge behavior in L2 and L3.

(t *testing.T)

// TestCountOverTimeInstantNsWithCutoff simulates merge behavior in L2 and L3.
func TestCountOverTimeInstantNsWithCutoff(t *testing.T) {
	start := 1*time.Second + 300*time.Nanosecond // additional 300ns that can be accidentally dropped by ms conversion
	end := 3 * time.Second
	step := end - start // for instant queries step == end-start
	req := &tempopb.QueryRangeRequest{
		Start: uint64(start),
		End:   uint64(end),
		Step:  uint64(step),
		Query: "{ } | count_over_time()",
	}
	req.SetInstant(true)

	cutoff := 2*time.Second + 300*time.Nanosecond
	// from start to cutoff
	req1 := *req
	req1.End = uint64(cutoff)
	req1.Step = req1.End - req1.Start
	// from cutoff to end
	req2 := *req
	req2.Start = uint64(cutoff)
	req2.Step = req2.End - req2.Start

	in1 := []Span{
		// outside of the range but within the range for ms. Should be ignored.
		newMockSpan(nil).WithStartTime(uint64(start - 20*time.Nanosecond)).WithDuration(1),
		newMockSpan(nil).WithStartTime(uint64(start - time.Nanosecond)).WithDuration(1),

		// within the range
		newMockSpan(nil).WithStartTime(uint64(start)).WithDuration(1),
		newMockSpan(nil).WithStartTime(uint64(start + time.Nanosecond)).WithDuration(1),
		newMockSpan(nil).WithStartTime(uint64(cutoff - time.Nanosecond)).WithDuration(1),
		newMockSpan(nil).WithStartTime(uint64(cutoff)).WithDuration(1),

		// outside of cutoff
		newMockSpan(nil).WithStartTime(uint64(cutoff + time.Nanosecond)).WithDuration(1),
	}

	in2 := []Span{
		// outside of cutoff
		newMockSpan(nil).WithStartTime(uint64(cutoff - time.Nanosecond)).WithDuration(1),

		// within the range
		newMockSpan(nil).WithStartTime(uint64(cutoff)).WithDuration(1),
		newMockSpan(nil).WithStartTime(uint64(cutoff + time.Nanosecond)).WithDuration(1),
		newMockSpan(nil).WithStartTime(uint64(end - time.Nanosecond)).WithDuration(1),
		newMockSpan(nil).WithStartTime(uint64(end)).WithDuration(10),

		// outside of the range but within the range for ms. Should be ignored.
		newMockSpan(nil).WithStartTime(uint64(end + time.Nanosecond)).WithDuration(1),
		newMockSpan(nil).WithStartTime(uint64(end + 20*time.Nanosecond)).WithDuration(1),
	}

	out := []TimeSeries{
		{
			Labels: []Label{
				{Name: "__name__", Value: NewStaticString("count_over_time")},
			},
			Values:    []float64{8},