Preserve the original CelExpr from symmbol table for comprehensions in verifier

PiperOrigin-RevId: 951504755

Author: l46kok

verifier/src/main/java/dev/cel/verifier/CelAstToZ3Translator.java

@@ -644,6 +644,12 @@ private <T> T withScope(String varName, TranslatedValue value, Supplier<T> actio
   private TranslatedValue translateComprehension(CelExpr celExpr, CelAbstractSyntaxTree ast) {
     CelComprehension comp = celExpr.comprehension();
     CelExpr iterRangeExpr = comp.iterRange();
+    if (iterRangeExpr.exprKind().getKind() == ExprKind.Kind.IDENT) {
+      TranslatedValue boundTv = symbolTable.get(iterRangeExpr.ident().name());
+      if (boundTv != null) {
+        iterRangeExpr = boundTv.celExpr().orElse(iterRangeExpr);
+      }
+    }
     List<IterationElement> iterationElements = new ArrayList<>();
     List<BoolExpr> taints = new ArrayList<>();
     List<Expr<?>> allRangeElems = new ArrayList<>();