Function TestFieldExpressionRewriter_VisitOrder

pkg/traceql/ast_rewriter_test.go:188–246

(t *testing.T)

}

func TestFieldExpressionRewriter_VisitOrder(t *testing.T) {
	t.Run("BinaryOperation", func(t *testing.T) {
		cases := []struct {
			name       string
			query      string
			visitOrder []FieldExpression
		}{
			{
				name:  "simple",
				query: `{ .attr1 = "val1" }`,
				visitOrder: []FieldExpression{
					Attribute{Name: "attr1"},
					NewStaticString("val1"),
					&BinaryOperation{Op: OpEqual},
				},
			},
			{
				name:  "and",
				query: `{ .attr1 = "val1" && .attr2 != "val2" }`,
				visitOrder: []FieldExpression{
					Attribute{Name: "attr1"},
					NewStaticString("val1"),
					&BinaryOperation{Op: OpEqual},
					Attribute{Name: "attr2"},
					NewStaticString("val2"),
					&BinaryOperation{Op: OpNotEqual},
					&BinaryOperation{Op: OpAnd},
				},
			},
			{
				name:  "count",
				query: `{} | count() > 11`,
				visitOrder: []FieldExpression{
					NewStaticBool(true),
					nil,
					NewStaticInt(11),
				},
			},
		}

		for _, tc := range cases {
			t.Run(tc.name, func(t *testing.T) {
				expr, err := Parse(tc.query)
				require.NoError(t, err)

				var index int
				rw := &fieldExpressionRewriter{rewriteFunctions: []fieldExpressionRewriteFn{func(op FieldExpression) (FieldExpression, int) {
					if index < len(tc.visitOrder) {
						requireMatchingFieldExpression(t, op, tc.visitOrder[index])
					}
					index++
					return op, 0
				}}}
				rw.RewriteRoot(expr)
				require.Equal(t, len(tc.visitOrder), index, "visited element count mismatch")
			})
		}
	})