Function TestStaticOptimizerUpdateExpr

cel/optimizer_test.go:34–165

(t *testing.T)

)

func TestStaticOptimizerUpdateExpr(t *testing.T) {
	expr := `has(a.b)`
	inlined := `[x, y].filter(i, i.size() > 0)[0].z`

	e := optimizerEnv(t)
	exprAST, iss := e.Compile(expr)
	if iss.Err() != nil {
		t.Fatalf("Compile() failed: %v", iss.Err())
	}

	inlinedAST, iss := e.Compile(inlined)
	if iss.Err() != nil {
		t.Fatalf("Compile() failed: %v", iss.Err())
	}
	opt, err := cel.NewStaticOptimizer(&testOptimizer{t: t, inlineExpr: inlinedAST.NativeRep()})
	if err != nil {
		t.Fatalf("NewStaticOptimizer() failed: %v", err)
	}
	optAST, iss := opt.Optimize(e, exprAST)
	if iss.Err() != nil {
		t.Fatalf("Optimize() generated an invalid AST: %v", iss.Err())
	}
	optString, err := cel.AstToString(optAST)
	if err != nil {
		t.Fatalf("cel.AstToString() failed: %v", err)
	}
	sourceInfo := optAST.NativeRep().SourceInfo()
	sourceInfoPB, err := ast.SourceInfoToProto(sourceInfo)
	if err != nil {
		t.Fatalf("cel.AstToCheckedExpr() failed: %v", err)
	}
	sourceInfoPB.Positions = nil
	wantTextPB := `
		location:  "<input>"
        line_offsets:  9
        macro_calls:  {
          key:  1
          value:  {
            call_expr:  {
              function:  "has"
              args:  {
                id:  24
                select_expr:  {
                  operand:  {
                    id:  2
                    call_expr:  {
                      function:  "_[_]"
                      args:  {
                        id:  3
                      }
                      args:  {
                        id:  20
                        const_expr:  {
                          int64_value:  0
                        }
                      }
                    }
                  }