Internal Changes

PiperOrigin-RevId: 908917795

Author: l46kok

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

@@ -12,6 +12,7 @@ java_library(
     deps = [
         "//:auto_value",
         "//bundle:cel",
+        "//runtime:function_binding",
         "//testing/testrunner:cel_expression_source",
         "//testing/testrunner:cel_test_context",
         "//testing/testrunner:cel_test_suite",

conformance/src/test/java/dev/cel/conformance/policy/PolicyConformanceTest.java

@@ -19,6 +19,7 @@
 import com.google.protobuf.Value;
 import dev.cel.bundle.Cel;
 import dev.cel.bundle.CelFactory;
+import dev.cel.runtime.CelFunctionBinding;
 import dev.cel.testing.testrunner.CelExpressionSource;
 import dev.cel.testing.testrunner.CelTestContext;
 import dev.cel.testing.testrunner.CelTestSuite.CelTestSection.CelTestCase;
@@ -31,7 +32,25 @@
 /** Statement representing a single CEL policy conformance test case. */
 public final class PolicyConformanceTest extends Statement {
 
-  private static final Cel CEL = CelFactory.standardCelBuilder().build();
+  private static final Cel CEL =
+      CelFactory.standardCelBuilder()
+          .addFunctionBindings(
+              CelFunctionBinding.fromOverloads(
+                  "locationCode",
+                  CelFunctionBinding.from(
+                      "locationCode_string",
+                      String.class,
+                      (ip) -> {
+                        switch (ip) {
+                          case "10.0.0.1":
+                            return "us";
+                          case "10.0.0.2":
+                            return "de";
+                          default:
+                            return "ir";
+                        }
+                      })))
+          .build();
 
   private final String name;
   private final CelTestCase testCase;

testing/src/main/java/dev/cel/testing/testrunner/CelTestSuiteYamlParser.java

@@ -90,7 +90,7 @@ private CelTestSuite parseYaml(String celTestSuiteYamlContent, String descriptio
   }
 
   private CelTestSuite.Builder parseTestSuite(ParserContext<Node> ctx, Node node) {
-    CelTestSuite.Builder builder = CelTestSuite.newBuilder();
+    CelTestSuite.Builder builder = CelTestSuite.newBuilder().setName("").setDescription("");
     long id = ctx.collectMetadata(node);
     if (!assertYamlType(ctx, id, node, YamlNodeType.MAP)) {
       ctx.reportError(id, "Unknown test suite type: " + node.getTag());