Enable conformance tests for two-var comprehensions.

Add existsOne (preferred alternative to older exists_one)

PiperOrigin-RevId: 853942944

Author: jnthntatum

common/src/main/java/dev/cel/common/Operator.java

@@ -45,7 +45,9 @@ public enum Operator {
   HAS("has"),
   ALL("all"),
   EXISTS("exists"),
+  // Prefer EXISTS_ONE_NEW in new usages.
   EXISTS_ONE("exists_one"),
+  EXISTS_ONE_NEW("existsOne"),
   MAP("map"),
   FILTER("filter"),
   NOT_STRICTLY_FALSE("@not_strictly_false"),
@@ -109,6 +111,7 @@ public static Optional<Operator> find(String text) {
           .put(EQUALS.getFunction(), EQUALS)
           .put(EXISTS.getFunction(), EXISTS)
           .put(EXISTS_ONE.getFunction(), EXISTS_ONE)
+          .put(EXISTS_ONE_NEW.getFunction(), EXISTS_ONE_NEW)
           .put(FILTER.getFunction(), FILTER)
           .put(GREATER.getFunction(), GREATER)
           .put(GREATER_EQUALS.getFunction(), GREATER_EQUALS)

conformance/src/test/java/dev/cel/conformance/BUILD.bazel

@@ -75,6 +75,7 @@ _ALL_TESTS = [
     "@cel_spec//tests/simple:testdata/lists.textproto",
     "@cel_spec//tests/simple:testdata/logic.textproto",
     "@cel_spec//tests/simple:testdata/macros.textproto",
+    "@cel_spec//tests/simple:testdata/macros2.textproto",
     "@cel_spec//tests/simple:testdata/math_ext.textproto",
     "@cel_spec//tests/simple:testdata/namespace.textproto",
     "@cel_spec//tests/simple:testdata/optionals.textproto",

conformance/src/test/java/dev/cel/conformance/ConformanceTest.java

@@ -66,6 +66,7 @@ public final class ConformanceTest extends Statement {
           .setOptions(OPTIONS)
           .addLibraries(
               CelExtensions.bindings(),
+              CelExtensions.comprehensions(),
               CelExtensions.encoders(OPTIONS),
               CelExtensions.math(OPTIONS),
               CelExtensions.protos(),
@@ -80,6 +81,7 @@ public final class ConformanceTest extends Statement {
           .setOptions(OPTIONS)
           .addLibraries(
               CelExtensions.bindings(),
+              CelExtensions.comprehensions(),
               CelExtensions.encoders(OPTIONS),
               CelExtensions.math(OPTIONS),
               CelExtensions.protos(),
@@ -106,10 +108,12 @@ private static CelChecker getChecker(SimpleTest test) throws Exception {
         .addFileTypes(dev.cel.expr.conformance.proto2.TestAllTypesExtensions.getDescriptor())
         .addLibraries(
             CelExtensions.bindings(),
+            CelExtensions.comprehensions(),
             CelExtensions.encoders(OPTIONS),
             CelExtensions.math(OPTIONS),
             CelExtensions.sets(OPTIONS),
             CelExtensions.strings(),
+            CelExtensions.comprehensions(),
             CelOptionalLibrary.INSTANCE)
         .addMessageTypes(dev.cel.expr.conformance.proto2.TestAllTypes.getDescriptor())
         .addMessageTypes(dev.cel.expr.conformance.proto3.TestAllTypes.getDescriptor())
@@ -120,6 +124,7 @@ private static CelChecker getChecker(SimpleTest test) throws Exception {
       CelRuntimeFactory.standardCelRuntimeBuilder()
           .setOptions(OPTIONS)
           .addLibraries(
+              CelExtensions.comprehensions(),
               CelExtensions.encoders(OPTIONS),
               CelExtensions.math(OPTIONS),
               CelExtensions.sets(OPTIONS),
@@ -182,6 +187,7 @@ public void evaluate() throws Throwable {
     CelValidationResult response = getParser(test).parse(test.getExpr(), test.getName());
     assertThat(response.hasError()).isFalse();
     response = getChecker(test).check(response.getAst());
+    System.out.println("response: " + response.getIssueString());
     assertThat(response.hasError()).isFalse();
     Type resultType = CelProtoTypes.celTypeToType(response.getAst().getResultType());
 

extensions/src/main/java/dev/cel/extensions/CelComprehensionsExtensions.java

@@ -159,6 +159,10 @@ public ImmutableSet<CelMacro> macros() {
             Operator.EXISTS_ONE.getFunction(),
             3,
             CelComprehensionsExtensions::expandExistsOneMacro),
+        CelMacro.newReceiverMacro(
+            Operator.EXISTS_ONE_NEW.getFunction(),
+            3,
+            CelComprehensionsExtensions::expandExistsOneMacro),
         CelMacro.newReceiverMacro(
             "transformList", 3, CelComprehensionsExtensions::transformListMacro),
         CelMacro.newReceiverMacro(

parser/src/main/java/dev/cel/parser/CelStandardMacro.java

@@ -48,12 +48,20 @@ public enum CelStandardMacro {
 
   /**
    * Boolean comprehension which asserts that a predicate holds true for exactly one element in the
-   * input range.
+   * input range. Prefer EXISTS_ONE_NEW.
    */
   EXISTS_ONE(
       CelMacro.newReceiverMacro(
           Operator.EXISTS_ONE.getFunction(), 2, CelStandardMacro::expandExistsOneMacro)),
 
+  /**
+   * Boolean comprehension which asserts that a predicate holds true for exactly one element in the
+   * input range.
+   */
+  EXISTS_ONE_NEW(
+      CelMacro.newReceiverMacro(
+          Operator.EXISTS_ONE_NEW.getFunction(), 2, CelStandardMacro::expandExistsOneMacro)),
+
   /**
    * Comprehension which applies a transform to each element in the input range and produces a list
    * of equivalent size as output.