Function testEvaluator

pkg/traceql/ast_execute_test.go:23–56

(t *testing.T, tc evalTC)

const testIterations = 5

func testEvaluator(t *testing.T, tc evalTC) {
	t.Helper()

	t.Run(tc.query, func(t *testing.T) {
		ast, err := Parse(tc.query)
		require.NoError(t, err)

		// run tests multiple times to ensure optimizations like regex caching are working
		for range testIterations {
			// clone input to confirm it doesn't get modified
			cloneIn := make([]*Spanset, len(tc.input))
			for i := range tc.input {
				cloneIn[i] = tc.input[i].clone()
				cloneIn[i].Spans = append([]Span(nil), tc.input[i].Spans...)
			}

			actual, err := ast.Pipeline.evaluate(tc.input)
			require.NoError(t, err)

			// sort expected/actual spansets. grouping requires this b/c map iteration makes the output
			// non-deterministic.
			makeSort := func(ss []*Spanset) func(i, j int) bool {
				return func(i, j int) bool {
					return bytes.Compare(ss[i].Spans[0].ID(), ss[j].Spans[0].ID()) < 0
				}
			}
			sort.Slice(actual, makeSort(actual))
			sort.Slice(tc.output, makeSort(tc.output))

			require.Equal(t, tc.output, actual)
			require.Equal(t, tc.input, cloneIn)
		}
	})
}

func TestSpansetFilter_matches(t *testing.T) {
	tests := []struct {