Add and wire in condition probing configuration

1 files changed, 19 insertions, 0 deletions

diff --git a/storage/src/tests/distributor/distributor_stripe_test.cpp b/storage/src/tests/distributor/distributor_stripe_test.cpp
index 73ff0f22614..2130c6ae62d 100644
--- a/storage/src/tests/distributor/distributor_stripe_test.cpp
+++ b/storage/src/tests/distributor/distributor_stripe_test.cpp
@@ -197,6 +197,12 @@ struct DistributorStripeTest : Test, DistributorStripeTestUtil {
         configure_stripe(builder);
     }
 
+    void configure_enable_condition_probing(bool enable_probing) {
+        ConfigBuilder builder;
+        builder.enableConditionProbing = enable_probing;
+        configure_stripe(builder);
+    }
+
     bool scheduler_has_implicitly_clear_priority_on_schedule_set() const noexcept {
         return _stripe->_scheduler->implicitly_clear_priority_on_schedule();
     }
@@ -1031,4 +1037,17 @@ TEST_F(DistributorStripeTest, enable_two_phase_gc_config_is_propagated_to_intern
     EXPECT_FALSE(getConfig().enable_two_phase_garbage_collection());
 }
 
+TEST_F(DistributorStripeTest, enable_condition_probing_config_is_propagated_to_internal_config)
+{
+    setup_stripe(Redundancy(1), NodeCount(1), "distributor:1 storage:1");
+
+    EXPECT_FALSE(getConfig().enable_condition_probing());
+
+    configure_enable_condition_probing(true);
+    EXPECT_TRUE(getConfig().enable_condition_probing());
+
+    configure_enable_condition_probing(false);
+    EXPECT_FALSE(getConfig().enable_condition_probing());
+}
+
 }