Support triggering runner library programmatically.

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) -> {

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) -> {

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) -> {

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) -> {

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) -> {

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) -> {

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) -> {

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) -> {