Function TestOptimizeWithSource

cel/optimizer_test.go:200–255

(t *testing.T)

}

func TestOptimizeWithSource(t *testing.T) {
	initial := `has(a.b)`
	replacement := `x["a"]`
	e := optimizerEnv(t)
	initialAST, iss := e.Compile(initial)
	if iss.Err() != nil {
		t.Fatalf("Compile(%q) failed: %v", initial, iss.Err())
	}
	replacementAST, iss := e.Compile(replacement)
	if iss.Err() != nil {
		t.Fatalf("Compile(%q) failed: %v", replacement, iss.Err())
	}

	opt, err := cel.NewStaticOptimizer(
		&replaceOptimizer{t: t, targetAST: replacementAST.NativeRep()},
		cel.OptimizeWithSource(replacementAST.Source()),
	)
	if err != nil {
		t.Fatalf("NewStaticOptimizer() failed: %v", err)
	}
	optAST, iss := opt.Optimize(e, initialAST)
	if iss.Err() != nil {
		t.Fatalf("Optimize() returned an error: %v", iss.Err())
	}

	if optAST.Source().Content() != replacement {
		t.Errorf("got source content %q, wanted %q", optAST.Source().Content(), replacement)
	}
	sourceInfoPB, err := ast.SourceInfoToProto(optAST.NativeRep().SourceInfo())
	if err != nil {
		t.Fatalf("cel.AstToCheckedExpr() failed: %v", err)
	}
	wantTextPB := `
		location: "<input>"
		line_offsets: 7
        positions: {
          key: 1
          value: 1
        }
        positions: {
          key: 2
          value: 0
        }
        positions: {
          key: 3
          value: 2
        }
	`
	var wantSourceInfoPB exprpb.SourceInfo
	if err := prototext.Unmarshal([]byte(wantTextPB), &wantSourceInfoPB); err != nil {
		t.Fatalf("prototext.Unmarshal() failed: %v", err)
	}
	if !proto.Equal(&wantSourceInfoPB, sourceInfoPB) {
		t.Errorf("got source info: %s, wanted %s", prototext.Format(sourceInfoPB), wantTextPB)
	}
}

func TestStaticOptimizerNilAST(t *testing.T) {