Internal Changes

PiperOrigin-RevId: 808012281

Author: ChinmayMadeshi

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

@@ -18,6 +18,7 @@
 
 import com.google.auto.value.AutoValue;
 import com.google.common.collect.ImmutableList;
+import com.google.common.io.Files;
 import com.google.errorprone.annotations.CanIgnoreReturnValue;
 import javax.annotation.concurrent.ThreadSafe;
 import dev.cel.common.CelAbstractSyntaxTree;
@@ -28,6 +29,8 @@
 import dev.cel.common.types.CelKind;
 import dev.cel.parser.CelUnparserVisitor;
 import dev.cel.runtime.CelEvaluationListener;
+import java.io.File;
+import java.io.IOException;
 import java.io.UnsupportedEncodingException;
 import java.net.URLEncoder;
 import java.util.Map;
@@ -392,4 +395,21 @@ private String escapeSpecialCharacters(String exprText) {
         .replace("{", "\\{")
         .replace("}", "\\}");
   }
+
+  private void writeDotGraphToArtifact(String dotGraph) {
+    String testOutputsDir = System.getenv("TEST_UNDECLARED_OUTPUTS_DIR");
+    if (testOutputsDir == null) {
+      // Only for OSS users, we write to a subdirectory under the cwd.
+      testOutputsDir = "cel_artifacts";
+    }
+    File outputDir = new File(testOutputsDir, "cel_test_coverage");
+    if (!outputDir.exists()) {
+      outputDir.mkdirs();
+    }
+    try {
+      Files.asCharSink(new File(outputDir, "coverage_graph.txt"), UTF_8).write(dotGraph);
+    } catch (IOException e) {
+      throw new IllegalStateException(e);
+    }
+  }
 }

testing/src/test/java/dev/cel/testing/testrunner/CelCoverageIndexTest.java

@@ -14,8 +14,10 @@
 package dev.cel.testing.testrunner;
 
 import static com.google.common.truth.Truth.assertThat;
+import static java.nio.charset.StandardCharsets.UTF_8;
 
 import com.google.common.collect.ImmutableMap;
+import com.google.common.io.Files;
 import dev.cel.bundle.Cel;
 import dev.cel.bundle.CelFactory;
 import dev.cel.common.CelAbstractSyntaxTree;
@@ -27,6 +29,7 @@
 import dev.cel.runtime.CelEvaluationListener;
 import dev.cel.runtime.CelRuntime;
 import dev.cel.testing.testrunner.CelCoverageIndex.CoverageReport;
+import java.io.File;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;