Automated fix for refs/heads/experiments-update

src/core/BUILD

@@ -126,6 +126,7 @@ grpc_cc_library(
         "absl/strings",
     ],
     language = "c++",
+    tags = ["nofixdeps"],
     deps = [
         "no_destruct",
         "//:config_vars",

src/core/lib/experiments/config.cc

@@ -54,6 +54,31 @@ std::atomic<bool> g_loaded(false);
 absl::AnyInvocable<bool(struct ExperimentMetadata)>* g_check_constraints_cb =
     nullptr;
 
+class TestExperiments {
+ public:
+  TestExperiments(const ExperimentMetadata* experiment_metadata,
+                  size_t num_experiments) {
+    enabled_ = new bool[num_experiments];
+    for (size_t i = 0; i < num_experiments; i++) {
+      if (g_check_constraints_cb != nullptr) {
+        enabled_[i] = (*g_check_constraints_cb)(experiment_metadata[i]);
+      } else {
+        enabled_[i] = experiment_metadata[i].default_value;
+      }
+    }
+  }
+
+  // Overloading [] operator to access elements in array style
+  bool operator[](int index) { return enabled_[index]; }
+
+  ~TestExperiments() { delete enabled_; }
+
+ private:
+  bool* enabled_;
+};
+
+TestExperiments* g_test_experiments = nullptr;
+
 GPR_ATTRIBUTE_NOINLINE Experiments LoadExperimentsFromConfigVariable() {
   g_loaded.store(true, std::memory_order_relaxed);
   // Set defaults from metadata.
@@ -111,11 +136,20 @@ void TestOnlyReloadExperimentsFromConfigVariables() {
   PrintExperimentsList();
 }
 
+void LoadTestOnlyExperimentsFromMetadata(
+    const ExperimentMetadata* experiment_metadata, size_t num_experiments) {
+  g_test_experiments =
+      new TestExperiments(experiment_metadata, num_experiments);
+}
+
 bool IsExperimentEnabled(size_t experiment_id) {
-  // Normal path: just return the value;
   return ExperimentsSingleton().enabled[experiment_id];
 }
 
+bool IsTestExperimentEnabled(size_t experiment_id) {
+  return (*g_test_experiments)[experiment_id];
+}
+
 void PrintExperimentsList() {
   size_t max_experiment_length = 0;
   for (size_t i = 0; i < kNumExperiments; i++) {

src/core/lib/experiments/config.h

@@ -26,17 +26,37 @@
 
 namespace grpc_core {
 
+struct ExperimentMetadata {
+  const char* name;
+  const char* description;
+  const char* additional_constaints;
+  bool default_value;
+  bool allow_in_fuzzing_config;
+};
+
 #ifndef GRPC_EXPERIMENTS_ARE_FINAL
 // Return true if experiment \a experiment_id is enabled.
 // Experiments are numbered by their order in the g_experiment_metadata array
 // declared in experiments.h.
 bool IsExperimentEnabled(size_t experiment_id);
 
+// Given a test experiment id, returns true if the test experiment is enabled.
+// Test experiments can be loaded using the LoadTestOnlyExperimentsFromMetadata
+// method.
+bool IsTestExperimentEnabled(size_t experiment_id);
+
 // Reload experiment state from config variables.
 // Does not change ForceEnableExperiment state.
 // Expects the caller to handle global thread safety - so really only
 // appropriate for carefully written tests.
 void TestOnlyReloadExperimentsFromConfigVariables();
+
+// Reload experiment state from passed metadata.
+// Does not change ForceEnableExperiment state.
+// Expects the caller to handle global thread safety - so really only
+// appropriate for carefully written tests.
+void LoadTestOnlyExperimentsFromMetadata(
+    const ExperimentMetadata* experiment_metadata, size_t num_experiments);
 #endif
 
 // Print out a list of all experiments that are built into this binary.
@@ -49,14 +69,6 @@ void PrintExperimentsList();
 // If this is called twice for the same experiment, both calls must agree.
 void ForceEnableExperiment(absl::string_view experiment_name, bool enable);
 
-struct ExperimentMetadata {
-  const char* name;
-  const char* description;
-  const char* additional_constaints;
-  bool default_value;
-  bool allow_in_fuzzing_config;
-};
-
 // Register a function to be called to validate the value an experiment can
 // take subject to additional constraints.
 // The function will take the ExperimentMetadata as its argument. It will return

test/core/experiments/BUILD

@@ -0,0 +1,45 @@
+# Copyright 2023 gRPC authors.
+#
+# 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
+#
+#     http://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.
+
+load("//bazel:grpc_build_system.bzl", "grpc_cc_library", "grpc_cc_test")
+
+grpc_cc_library(
+    name = "experiments_lib",
+    srcs = [
+        "fixtures/experiments.cc",
+    ],
+    hdrs = [
+        "fixtures/experiments.h",
+    ],
+    language = "c++",
+    deps = [
+        "//:config_vars",
+        "//:gpr",
+        "//src/core:experiments",
+        "//src/core:no_destruct",
+    ],
+)
+
+grpc_cc_test(
+    name = "experiments_test",
+    srcs = ["experiments_test.cc"],
+    external_deps = ["gtest"],
+    language = "C++",
+    uses_event_engine = False,
+    uses_polling = False,
+    deps = [
+        ":experiments_lib",
+        "//test/core/util:grpc_test_util",
+    ],
+)

test/core/experiments/experiments_test.cc

@@ -0,0 +1,62 @@
+// Copyright 2023 gRPC authors.
+//
+// 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
+//
+//     http://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.
+
+// Auto generated by tools/codegen/core/gen_experiments.py
+
+#include <grpc/support/port_platform.h>
+
+#include "test/core/experiments/fixtures/experiments.h"
+
+#include "gtest/gtest.h"
+
+#include "src/core/lib/experiments/config.h"
+
+#ifndef GRPC_EXPERIMENTS_ARE_FINAL
+
+bool GetExperimentTestExperiment1ExpectedValue() { return false; }
+
+bool GetExperimentTestExperiment2ExpectedValue() { return false; }
+
+bool GetExperimentTestExperiment3ExpectedValue() {
+#ifdef NDEBUG
+  return false;
+#else
+  return true;
+#endif
+}
+
+bool GetExperimentTestExperiment4ExpectedValue() { return true; }
+
+TEST(ExperimentsTest, CheckExperimentValuesTest) {
+  ASSERT_EQ(grpc_core::IsTestExperiment1Enabled(),
+            GetExperimentTestExperiment1ExpectedValue());
+
+  ASSERT_EQ(grpc_core::IsTestExperiment2Enabled(),
+            GetExperimentTestExperiment2ExpectedValue());
+
+  ASSERT_EQ(grpc_core::IsTestExperiment3Enabled(),
+            GetExperimentTestExperiment3ExpectedValue());
+
+  ASSERT_EQ(grpc_core::IsTestExperiment4Enabled(),
+            GetExperimentTestExperiment4ExpectedValue());
+}
+
+#endif  // GRPC_EXPERIMENTS_ARE_FINAL
+
+int main(int argc, char** argv) {
+  testing::InitGoogleTest(&argc, argv);
+  grpc_core::LoadTestOnlyExperimentsFromMetadata(
+      grpc_core::g_test_experiment_metadata, grpc_core::kNumTestExperiments);
+  return RUN_ALL_TESTS();
+}

test/core/experiments/fixtures/experiments.cc

@@ -0,0 +1,52 @@
+// Copyright 2023 gRPC authors.
+//
+// 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
+//
+//     http://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.
+
+// Auto generated by tools/codegen/core/gen_experiments.py
+
+#include <grpc/support/port_platform.h>
+
+#include "test/core/experiments/fixtures/experiments.h"
+
+#ifndef GRPC_EXPERIMENTS_ARE_FINAL
+namespace {
+const char* const description_test_experiment_1 = "Test Experiment 1";
+const char* const additional_constraints_test_experiment_1 = "{}";
+const char* const description_test_experiment_2 = "Test Experiment 2";
+const char* const additional_constraints_test_experiment_2 = "{}";
+const char* const description_test_experiment_3 = "Test Experiment 3";
+const char* const additional_constraints_test_experiment_3 = "{}";
+const char* const description_test_experiment_4 = "Test Experiment 4";
+const char* const additional_constraints_test_experiment_4 = "{}";
+#ifdef NDEBUG
+const bool kDefaultForDebugOnly = false;
+#else
+const bool kDefaultForDebugOnly = true;
+#endif
+}  // namespace
+
+namespace grpc_core {
+
+const ExperimentMetadata g_test_experiment_metadata[] = {
+    {"test_experiment_1", description_test_experiment_1,
+     additional_constraints_test_experiment_1, false, true},
+    {"test_experiment_2", description_test_experiment_2,
+     additional_constraints_test_experiment_2, false, true},
+    {"test_experiment_3", description_test_experiment_3,
+     additional_constraints_test_experiment_3, kDefaultForDebugOnly, true},
+    {"test_experiment_4", description_test_experiment_4,
+     additional_constraints_test_experiment_4, true, true},
+};
+
+}  // namespace grpc_core
+#endif