Function TestParseRewrites

pkg/traceql/parse_test.go:1961–2006

(t *testing.T)

}

func TestParseRewrites(t *testing.T) {
	tests := []struct {
		name  string
		query string
		want  string
	}{
		{
			name:  "no rewrites",
			query: "{ .attr = `foo` }",
			want:  "{ .attr = `foo` }",
		},
		{
			name:  "query with rewrite OR",
			query: "{ .attr = `foo` || .attr = `bar` } | rate()",
			want:  "{ .attr IN [`foo`, `bar`] } | rate()",
		},
		{
			name:  "query with rewrite AND",
			query: "{ .attr != `foo` && .attr != `bar` }",
			want:  "{ .attr NOT IN [`foo`, `bar`] }",
		},
		{
			name:  "query with rewrite OR regex",
			query: "{ .attr =~ `foo` || .attr =~ `bar` }",
			want:  "{ .attr MATCH ANY [`foo`, `bar`] }",
		},
		{
			name:  "query with rewrite AND regex",
			query: "{ .attr !~ `foo` && .attr !~ `bar` }",
			want:  "{ .attr MATCH NONE [`foo`, `bar`] }",
		},
		{
			name:  "skip rewrites via unsafe hint",
			query: "{ .attr = `foo` || .attr = `bar` } | rate() with(skip_ast_transformations=`or_to_in`)",
			want:  "{ (.attr = `foo`) || (.attr = `bar`) } | rate() with(skip_ast_transformations=`or_to_in`)",
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			actual, err := Parse(tc.query, WithUnsafeHints(true))
			require.NoError(t, err)
			require.Equal(t, tc.want, actual.String())
		})
	}
}

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