Fix planner to honor enableComprehensions flag

PiperOrigin-RevId: 992461826

Author: l46kok

bundle/src/test/java/dev/cel/bundle/CelImplTest.java

@@ -1982,10 +1982,13 @@ public void program_nativeTypeUnknownsEnabled_asCallArguments() throws Exception
 
   @Test
   public void program_comprehensionDisabled_throws() throws Exception {
-    // TODO: Planner ExecutionFrame ignores CelOptions.enableComprehension(false).
     Cel cel =
-        legacyCelBuilderWithMacros()
-            .setOptions(CelOptions.current().enableComprehension(false).build())
+        plannerCelBuilderWithMacros()
+            .setOptions(
+                CelOptions.current()
+                    .enableHeterogeneousNumericComparisons(true)
+                    .enableComprehension(false)
+                    .build())
             .build();
     CelAbstractSyntaxTree ast = cel.compile("['foo', 'bar'].map(x, x)").getAst();
 

runtime/src/main/java/dev/cel/runtime/planner/ExecutionFrame.java

@@ -46,7 +46,7 @@ static ExecutionFrame create(
       @Nullable CelEvaluationListener listener) {
     return new ExecutionFrame(
         functionResolver,
-        celOptions.comprehensionMaxIterations(),
+        getComprehensionMaxIterations(celOptions),
         partialVars,
         listener,
         /* asyncTracker= */ null);
@@ -61,12 +61,16 @@ static ExecutionFrame createForAsync(
     checkNotNull(asyncTracker, "asyncTracker");
     return new ExecutionFrame(
         functionResolver,
-        celOptions.comprehensionMaxIterations(),
+        getComprehensionMaxIterations(celOptions),
         partialVars,
         listener,
         asyncTracker);
   }
 
+  private static int getComprehensionMaxIterations(CelOptions celOptions) {
+    return celOptions.enableComprehension() ? celOptions.comprehensionMaxIterations() : 0;
+  }
+
   Optional<CelResolvedOverload> findOverload(
       String functionName, Collection<String> overloadIds, Object[] args)
       throws CelEvaluationException {