Reach 100% patch coverage: CapturedContext & LogProbe

Co-authored-by: albabn <[email protected]>

Author: datadog-datadog-prod-us1[bot]

dd-java-agent/agent-debugger/src/test/java/com/datadog/debugger/probe/LogProbeCaptureExpressionsUndefinedTest.java

@@ -0,0 +1,43 @@
+package com.datadog.debugger.probe;
+
+import static java.util.Collections.emptyList;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+import com.datadog.debugger.el.DSL;
+import com.datadog.debugger.el.ValueScript;
+import datadog.trace.bootstrap.debugger.CapturedContext;
+import datadog.trace.bootstrap.debugger.EvaluationError;
+import datadog.trace.bootstrap.debugger.MethodLocation;
+import datadog.trace.bootstrap.debugger.ProbeId;
+import datadog.trace.bootstrap.debugger.el.Values;
+import java.util.Arrays;
+import org.junit.jupiter.api.Test;
+
+class LogProbeCaptureExpressionsUndefinedTest {
+  @Test
+  void captureExpressionUndefinedAddsError() {
+    LogProbe.CaptureExpression ce =
+        new LogProbe.CaptureExpression(
+            "ce", new ValueScript(DSL.value(Values.UNDEFINED_OBJECT), "exprUndefined"), null);
+
+    LogProbe probe =
+        LogProbe.builder()
+            .language("java")
+            .probeId(ProbeId.newId())
+            .template(null, emptyList())
+            .captureExpressions(Arrays.asList(ce))
+            .build();
+
+    CapturedContext context = new CapturedContext();
+    LogProbe.LogStatus status = new LogProbe.LogStatus(probe);
+
+    probe.evaluate(context, status, MethodLocation.DEFAULT);
+
+    assertTrue(status.hasLogTemplateErrors());
+    assertEquals(1, status.getErrors().size());
+    EvaluationError err = status.getErrors().get(0);
+    assertEquals("exprUndefined", err.getExpr());
+    assertEquals("UNDEFINED", err.getMessage());
+  }
+}

dd-java-agent/agent-debugger/src/test/java/datadog/trace/bootstrap/debugger/CapturedContextCaptureExpressionsTest.java

@@ -0,0 +1,46 @@
+package datadog.trace.bootstrap.debugger;
+
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.junit.jupiter.api.Assertions.assertSame;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+import com.datadog.debugger.agent.JsonSnapshotSerializer;
+import datadog.trace.bootstrap.debugger.util.TimeoutChecker;
+import java.time.Duration;
+import java.time.temporal.ChronoUnit;
+import java.util.Map;
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.Test;
+
+class CapturedContextCaptureExpressionsTest {
+  @BeforeAll
+  static void initSerializer() {
+    // Ensure freeze() has a serializer to produce strValue
+    DebuggerContext.initValueSerializer(new JsonSnapshotSerializer());
+  }
+
+  @Test
+  void addGetAndFreezeCaptureExpressions() {
+    CapturedContext ctx = new CapturedContext();
+
+    CapturedContext.CapturedValue exprVal =
+        CapturedContext.CapturedValue.of("expr1", Object.class.getTypeName(), 42);
+
+    // addCaptureExpression should lazily init the map and put the value
+    ctx.addCaptureExpression(exprVal);
+
+    // getCaptureExpressions should return the map (covering getter line)
+    Map<String, CapturedContext.CapturedValue> exprs = ctx.getCaptureExpressions();
+    assertNotNull(exprs);
+    assertTrue(exprs.containsKey("expr1"));
+    assertSame(exprVal, exprs.get("expr1"));
+
+    // freeze should only freeze captureExpressions branch when present
+    ctx.freeze(new TimeoutChecker(Duration.of(50, ChronoUnit.MILLIS)));
+
+    CapturedContext.CapturedValue frozen = exprs.get("expr1");
+    assertNotNull(frozen.getStrValue());
+    assertNull(frozen.getValue()); // value released after serialization
+  }
+}