test(debugger): budget limit for capture expressions (#7111)

Co-authored-by: jean-philippe bempel <jean-philippe.bempel@datadoghq.com>

Author: tylfin

manifests/java.yml

@@ -3047,6 +3047,7 @@ manifest:
         spring-boot-undertow: v1.38.0
         spring-boot-wildfly: v1.38.0
         uds-spring-boot: v1.38.0
+  tests/debugger/test_debugger_probe_budgets.py::Test_Debugger_Probe_Budgets::test_log_line_capture_expression_budgets: bug (DEBUG-5730)
   tests/debugger/test_debugger_probe_budgets.py::Test_Debugger_Probe_Budgets::test_span_probe_expression_budgets: bug (DEBUG-3797)
   tests/debugger/test_debugger_probe_snapshot.py::Test_Debugger_Line_Probe_Snaphots:
     - weblog_declaration:

manifests/php.yml

@@ -577,6 +577,7 @@ manifest:
   tests/debugger/test_debugger_pii.py::Test_Debugger_PII_Redaction::test_pii_redaction_method_full: v1.19.0
   tests/debugger/test_debugger_pii.py::Test_Debugger_PII_Redaction_Excluded_Identifiers: missing_feature (DD_DYNAMIC_INSTRUMENTATION_REDACTION_EXCLUDED_IDENTIFIERS config key not implemented; only REDACTED_IDENTIFIERS exists which adds names, not removes them from the built-in list)
   tests/debugger/test_debugger_probe_budgets.py::Test_Debugger_Probe_Budgets: v1.13.0+4663b2fa7c20c6920f347d059b57dc2a419cb7f7
+  tests/debugger/test_debugger_probe_budgets.py::Test_Debugger_Probe_Budgets::test_log_line_capture_expression_budgets: missing_feature
   tests/debugger/test_debugger_probe_budgets.py::Test_Debugger_Probe_Budgets::test_span_probe_expression_budgets: irrelevant
   tests/debugger/test_debugger_probe_snapshot.py::Test_Debugger_Line_Probe_Snaphots: missing_feature
   tests/debugger/test_debugger_probe_snapshot.py::Test_Debugger_Line_Probe_Snaphots::test_process_tags_snapshot: missing_feature (Not yet implemented)

manifests/python.yml

@@ -1258,6 +1258,7 @@ manifest:
     - weblog_declaration:
         "*": missing_feature
         *flask: v2.11.0
+  tests/debugger/test_debugger_probe_budgets.py::Test_Debugger_Probe_Budgets::test_log_line_capture_expression_budgets: bug (DEBUG-5743)
   tests/debugger/test_debugger_probe_budgets.py::Test_Debugger_Probe_Budgets::test_span_probe_expression_budgets: irrelevant (Python emits evaluation errors as span tags, not error snapshots to the debugger endpoint)
   tests/debugger/test_debugger_probe_snapshot.py::Test_Debugger_Line_Probe_Snaphots:
     - weblog_declaration:

tests/debugger/test_debugger_probe_budgets.py

@@ -89,6 +89,41 @@ def test_log_line_budgets(self):
                 f"Expected 1-20 snapshot with captures, got {snapshots_with_captures} in {self.total_request_time} seconds"
             )
 
+    def setup_log_line_capture_expression_budgets(self):
+        self._setup(
+            "probe_capture_expressions_budgets",
+            "/debugger/budgets/150",
+            lines=self.method_and_language_to_line_number("Budgets", self.get_tracer()["language"]),
+        )
+
+    def test_log_line_capture_expression_budgets(self):
+        self._assert()
+        self._validate_snapshots()
+
+        snapshots_with_captures = 0
+        for _id in self.probe_ids:
+            for span in self.probe_snapshots[_id]:
+                captures = span.get("debugger", {}).get("snapshot", {}).get("captures", None)
+                if captures is None:
+                    continue
+
+                # Capture expressions live under captures.lines.{line}.captureExpressions for line probes.
+                has_capture_expressions = any(
+                    isinstance(line_data, dict) and "captureExpressions" in line_data
+                    for line_data in captures.get("lines", {}).values()
+                )
+                if not has_capture_expressions:
+                    continue
+
+                snapshots_with_captures += 1
+
+            # Probe budgets aren't exact and can take time to be applied, so we allow a range of 1-20 snapshots with
+            # capture expressions for 150 requests, matching the 1/s snapshot limit.
+            assert 1 <= snapshots_with_captures <= 20, (
+                f"Expected 1-20 snapshots with capture expressions, got {snapshots_with_captures} in "
+                f"{self.total_request_time} seconds"
+            )
+
     def setup_span_probe_expression_budgets(self):
         self.initialize_weblog_remote_config()
 

tests/debugger/utils/probes/probe_capture_expressions_budgets.json

@@ -0,0 +1,31 @@
+[
+    {
+        "language": "",
+        "type": "",
+        "id": "",
+        "version": 0,
+        "where": {
+            "typeName": null,
+            "sourceFile": "ACTUAL_SOURCE_FILE",
+            "lines": [
+                "141"
+            ]
+        },
+        "captureSnapshot": false,
+        "capture": {
+            "maxReferenceDepth": 3
+        },
+        "captureExpressions": [
+            {
+                "name": "loops",
+                "expr": {
+                    "dsl": "loops",
+                    "json": {"ref": "loops"}
+                },
+                "capture": {
+                    "maxReferenceDepth": 1
+                }
+            }
+        ]
+    }
+]