Support triggering runner library programmatically.

PiperOrigin-RevId: 777700797

Author: ChinmayMadeshi

testing/src/main/java/dev/cel/testing/testrunner/BUILD.bazel

@@ -42,6 +42,7 @@ java_library(
     tags = [
     ],
     deps = [
+        ":cel_expression_source",
         ":cel_test_context",
         ":cel_test_suite",
         ":test_runner_library",
@@ -55,11 +56,11 @@ java_library(
     tags = [
     ],
     deps = [
+        ":cel_expression_source",
         ":cel_test_context",
         ":cel_test_suite",
         ":registry_utils",
         ":result_matcher",
-        "//:auto_value",
         "//bundle:cel",
         "//bundle:environment",
         "//bundle:environment_yaml_parser",
@@ -129,6 +130,7 @@ java_library(
     tags = [
     ],
     deps = [
+        ":cel_expression_source",
         ":default_result_matcher",
         ":result_matcher",
         "//:auto_value",
@@ -207,6 +209,16 @@ java_library(
     ],
 )
 
+java_library(
+    name = "cel_expression_source",
+    srcs = ["CelExpressionSource.java"],
+    tags = [
+    ],
+    deps = [
+        "//:auto_value",
+    ],
+)
+
 filegroup(
     name = "test_runner_binary",
     srcs = [

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

@@ -0,0 +1,75 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+package dev.cel.testing.testrunner;
+
+import com.google.auto.value.AutoValue;
+
+/**
+ * CelExpressionSource is an encapsulation around cel_expr file format argument accepted in
+ * cel_java_test/cel_test bzl macro. It either holds a {@link CheckedExpr} in binarypb/textproto
+ * format, a serialized {@link CelPolicy} file in yaml/celpolicy format or a raw cel expression in
+ * cel file format or string format.
+ */
+@AutoValue
+public abstract class CelExpressionSource {
+
+  /** Returns the value of the expression source. This can be a file path or a raw expression. */
+  public abstract String value();
+
+  /** Returns the type of the expression source. */
+  public abstract ExpressionSourceType type();
+
+  /**
+   * Creates a {@link CelExpressionSource} from a file path. The type of the expression source is
+   * inferred from the file extension.
+   */
+  public static CelExpressionSource fromSource(String value) {
+    return new AutoValue_CelExpressionSource(value, ExpressionSourceType.fromSource(value));
+  }
+
+  /** Creates a {@link CelExpressionSource} from a raw CEL expression string. */
+  public static CelExpressionSource fromRawExpr(String value) {
+    return new AutoValue_CelExpressionSource(value, ExpressionSourceType.RAW_EXPR);
+  }
+
+  /**
+   * ExpressionSourceType is an enumeration of the supported expression file types.
+   *
+   * <p>This enumeration is used to determine the type of the expression file based on the file
+   * extension.
+   */
+  public enum ExpressionSourceType {
+    BINARYPB,
+    TEXTPROTO,
+    POLICY,
+    CEL,
+    RAW_EXPR;
+
+    private static ExpressionSourceType fromSource(String filePath) {
+      if (filePath.endsWith(".binarypb")) {
+        return BINARYPB;
+      }
+      if (filePath.endsWith(".textproto")) {
+        return TEXTPROTO;
+      }
+      if (filePath.endsWith(".yaml") || filePath.endsWith(".celpolicy")) {
+        return POLICY;
+      }
+      if (filePath.endsWith(".cel")) {
+        return CEL;
+      }
+      throw new IllegalArgumentException("Unsupported expression file type: " + filePath);
+    }
+  }
+}

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

@@ -79,6 +79,26 @@ public abstract class CelTestContext {
    */
   public abstract ResultMatcher resultMatcher();
 
+  /**
+   * The CEL expression to be tested. Could be a expression string or a policy/cel file path. This
+   * should only be used when invoking the runner library directly.
+   */
+  public abstract Optional<CelExpressionSource> celExpression();
+
+  /**
+   * The config file for the CEL test.
+   *
+   * <p>The config file is used to provide a custom environment for the CEL test.
+   */
+  public abstract Optional<String> configFile();
+
+  /**
+   * The file descriptor set path for the CEL test.
+   *
+   * <p>The file descriptor set path is used to provide proto descriptors for the CEL test.
+   */
+  public abstract Optional<String> fileDescriptorSetPath();
+
   /** Returns a builder for {@link CelTestContext} with the current instance's values. */
   public abstract Builder toBuilder();
 
@@ -107,6 +127,12 @@ public abstract Builder setCelLateFunctionBindings(
 
     public abstract Builder setResultMatcher(ResultMatcher resultMatcher);
 
+    public abstract Builder setCelExpression(CelExpressionSource celExpression);
+
+    public abstract Builder setConfigFile(String configFile);
+
+    public abstract Builder setFileDescriptorSetPath(String fileDescriptorSetPath);
+
     public abstract CelTestContext build();
   }
 }

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

@@ -53,8 +53,8 @@ private TestSuite parseTestSuite(String textProto) throws IOException {
     TypeRegistry typeRegistry = TypeRegistry.getEmptyTypeRegistry();
     ExtensionRegistry extensionRegistry = ExtensionRegistry.getEmptyRegistry();
     if (fileDescriptorSetPath != null) {
-      extensionRegistry = RegistryUtils.getExtensionRegistry();
-      typeRegistry = RegistryUtils.getTypeRegistry();
+      extensionRegistry = RegistryUtils.getExtensionRegistry(fileDescriptorSetPath);
+      typeRegistry = RegistryUtils.getTypeRegistry(fileDescriptorSetPath);
     }
     TextFormat.Parser parser = TextFormat.Parser.newBuilder().setTypeRegistry(typeRegistry).build();
     TestSuite.Builder builder = TestSuite.newBuilder();

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

@@ -36,6 +36,38 @@ public CelUserTestTemplate(CelTestContext celTestContext) {
 
   @Test
   public void test() throws Exception {
-    TestRunnerLibrary.runTest(testCase, celTestContext);
+    TestRunnerLibrary.runTest(testCase, updateCelTestContext(celTestContext));
+  }
+
+  /**
+   * Updates the CEL test context based on the system properties.
+   *
+   * <p>This method is used to update the CEL test context based on the system properties. It checks
+   * if the runner library is triggered via blaze macro or via JUnit and assigns values accordingly.
+   *
+   * @param celTestContext The CEL test context to update.
+   * @return The updated CEL test context.
+   */
+  private CelTestContext updateCelTestContext(CelTestContext celTestContext) {
+    String celExpr = System.getProperty("cel_expr");
+    String configPath = System.getProperty("config_path");
+    String fileDescriptorSetPath = System.getProperty("file_descriptor_set_path");
+    String isRawExpr = System.getProperty("is_raw_expr");
+
+    CelTestContext.Builder celTestContextBuilder = celTestContext.toBuilder();
+    if (celExpr != null) {
+      if (isRawExpr.equals("True")) {
+        celTestContextBuilder.setCelExpression(CelExpressionSource.fromRawExpr(celExpr));
+      } else {
+        celTestContextBuilder.setCelExpression(CelExpressionSource.fromSource(celExpr));
+      }
+    }
+    if (configPath != null) {
+      celTestContextBuilder.setConfigFile(configPath);
+    }
+    if (fileDescriptorSetPath != null) {
+      celTestContextBuilder.setFileDescriptorSetPath(fileDescriptorSetPath);
+    }
+    return celTestContextBuilder.build();
   }
 }

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

@@ -88,7 +88,9 @@ private static void assertExprValue(ExprValue exprValue, ExprValue expectedExprV
       assertThat(exprValue)
           .ignoringRepeatedFieldOrderOfFieldDescriptors(
               MapValue.getDescriptor().findFieldByName("entries"))
-          .unpackingAnyUsing(RegistryUtils.getTypeRegistry(), RegistryUtils.getExtensionRegistry())
+          .unpackingAnyUsing(
+              RegistryUtils.getTypeRegistry(fileDescriptorSetPath),
+              RegistryUtils.getExtensionRegistry(fileDescriptorSetPath))
           .isEqualTo(expectedExprValue);
     } else {
       assertThat(exprValue)

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

@@ -27,17 +27,18 @@
 public final class RegistryUtils {
 
   /** Returns the {@link TypeRegistry} for the given file descriptor set. */
-  public static TypeRegistry getTypeRegistry() throws IOException {
+  public static TypeRegistry getTypeRegistry(String fileDescriptorSetPath) throws IOException {
     return TypeRegistry.newBuilder()
-        .add(getAllDescriptorsFromJvm().messageTypeDescriptors())
+        .add(getAllDescriptorsFromJvm(fileDescriptorSetPath).messageTypeDescriptors())
         .build();
   }
 
   /** Returns the {@link ExtensionRegistry} for the given file descriptor set. */
-  public static ExtensionRegistry getExtensionRegistry() throws IOException {
+  public static ExtensionRegistry getExtensionRegistry(String fileDescriptorSetPath)
+      throws IOException {
     ExtensionRegistry extensionRegistry = ExtensionRegistry.newInstance();
 
-    getAllDescriptorsFromJvm()
+    getAllDescriptorsFromJvm(fileDescriptorSetPath)
         .extensionDescriptors()
         .forEach(
             (descriptorName, descriptor) -> {