From ad137617e146bf2d050747ca3a7fedbcef0e4018 Mon Sep 17 00:00:00 2001
From: Jonathan Tatum <jdtatum@google.com>
Date: Wed, 3 Jun 2026 14:00:23 -0700
Subject: [PATCH] Add memory test for managed constant folding arena

PiperOrigin-RevId: 926250668
---
 eval/tests/memory_safety_test.cc | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/eval/tests/memory_safety_test.cc b/eval/tests/memory_safety_test.cc
index 9c0a683e4..a88844fed 100644
--- a/eval/tests/memory_safety_test.cc
+++ b/eval/tests/memory_safety_test.cc
@@ -51,7 +51,12 @@ struct TestCase {
   bool reference_resolver_enabled = false;
 };
 
-enum Options { kDefault, kExhaustive, kFoldConstants };
+enum Options {
+  kDefault,
+  kExhaustive,
+  kFoldConstants,
+  kFoldConstantsManagedArena
+};
 
 using ParamType = std::tuple<TestCase, Options>;
 
@@ -68,6 +73,9 @@ std::string TestCaseName(const testing::TestParamInfo<ParamType>& param_info) {
     case Options::kFoldConstants:
       opt = "opt";
       break;
+    case Options::kFoldConstantsManagedArena:
+      opt = "opt_managed_arena";
+      break;
   }
 
   return absl::StrCat(std::get<0>(param).name, "_", opt);
@@ -110,6 +118,14 @@ class EvaluatorMemorySafetyTest : public testing::TestWithParam<ParamType> {
         options.enable_comprehension_vulnerability_check = false;
         options.short_circuiting = true;
         break;
+      case Options::kFoldConstantsManagedArena:
+        options.enable_regex_precompilation = true;
+        options.constant_folding = true;
+        options.enable_comprehension_list_append = true;
+        options.enable_comprehension_vulnerability_check = false;
+        options.short_circuiting = true;
+        options.constant_arena = nullptr;
+        break;
     }
 
     options.enable_qualified_identifier_rewrites =
@@ -295,7 +311,8 @@ INSTANTIATE_TEST_SUITE_P(
                 test::IsCelBool(true),
             }}),
         testing::Values(Options::kDefault, Options::kExhaustive,
-                        Options::kFoldConstants)),
+                        Options::kFoldConstants,
+                        Options::kFoldConstantsManagedArena)),
     &TestCaseName);
 
 }  // namespace