Function TestTrimToBlockOverlap

pkg/traceql/engine_metrics_test.go:329–402

(t *testing.T)

}

func TestTrimToBlockOverlap(t *testing.T) {
	tc := []struct {
		start1, end1               string
		step                       time.Duration
		start2, end2               string
		expectedStart, expectedEnd string
		expectedStep               time.Duration
	}{
		{
			// Block fully within range
			// Left border is extended to the next step.
			// Right border is extended to the next step.
			"2024-01-01T01:00:00Z", "2024-01-01T02:00:00Z", 5 * time.Minute,
			"2024-01-01T01:33:00Z", "2024-01-01T01:38:00Z",
			"2024-01-01T01:30:00Z", "2024-01-01T01:40:00Z", 5 * time.Minute,
		},
		{
			// Block overlapping right border.
			// Left border is extended to the next step.
			// Right border preserved.
			"2024-01-01T01:01:00Z", "2024-01-01T02:01:00.123Z", 5 * time.Minute,
			"2024-01-01T01:31:00Z", "2024-01-01T02:31:00Z",
			"2024-01-01T01:30:00Z", "2024-01-01T02:01:00.123Z", 5 * time.Minute,
		},
		{
			// Block overlapping left border.
			// Left border preserved.
			// Right border extended to the next step.
			"2024-01-01T01:01:00.123Z", "2024-01-01T02:00:00Z", 5 * time.Minute,
			"2024-01-01T00:31:00Z", "2024-01-01T01:31:00Z",
			"2024-01-01T01:01:00.123Z", "2024-01-01T01:35:00Z", 5 * time.Minute,
		},
		{
			// Block larger than range
			// Neither border is extended. Nanoseconds preserved.
			"2024-01-01T01:00:01.123Z", "2024-01-01T01:15:01.123Z", 5 * time.Minute,
			"2024-01-01T00:00:00Z", "2024-01-01T02:00:00Z",
			"2024-01-01T01:00:01.123Z", "2024-01-01T01:15:01.123Z", 5 * time.Minute,
		},
		{
			// Instant query, block overlaps right border.
			// Original range is 1h
			// Right border isn't extended past request range.
			// Left border is able to be extended.
			"2024-01-01T01:00:00.123Z", "2024-01-01T02:00:00.123Z", time.Hour,
			"2024-01-01T01:30:00.123Z", "2024-01-01T02:30:00.123Z",
			// we subtract a nanosecond to the start to make sure it's before the block start
			"2024-01-01T01:30:00.122999999Z", "2024-01-01T02:00:00.123Z", 30*time.Minute + time.Nanosecond,
		},
	}

	for _, c := range tc {
		start1, _ := time.Parse(time.RFC3339Nano, c.start1)
		end1, _ := time.Parse(time.RFC3339Nano, c.end1)
		start2, _ := time.Parse(time.RFC3339Nano, c.start2)
		end2, _ := time.Parse(time.RFC3339Nano, c.end2)

		req := &tempopb.QueryRangeRequest{