UT Fix

Signed-off-by: Dev Agarwal <[email protected]>
opensearch-project · Jan 10, 2024 · 94849e9 · 94849e9
1 parent 28f1720
commit 94849e9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...st/java/org/opensearch/telemetry/tracing/ThreadContextBasedTracerContextStorageTests.java b/...st/java/org/opensearch/telemetry/tracing/ThreadContextBasedTracerContextStorageTests.java
@@ -31,6 +31,7 @@
 
 import static org.opensearch.telemetry.TelemetrySettings.TRACER_ENABLED_SETTING;
 import static org.opensearch.telemetry.TelemetrySettings.TRACER_FEATURE_ENABLED_SETTING;
+import static org.opensearch.telemetry.TelemetrySettings.TRACER_SAMPLER_ACTION_PROBABILITY;
 import static org.opensearch.telemetry.TelemetrySettings.TRACER_SAMPLER_PROBABILITY;
 import static org.hamcrest.CoreMatchers.instanceOf;
 import static org.hamcrest.CoreMatchers.is;
@@ -56,7 +57,10 @@ public void setUp() throws Exception {
 
         final TelemetrySettings telemetrySettings = new TelemetrySettings(
             settings,
-            new ClusterSettings(Settings.EMPTY, Set.of(TRACER_ENABLED_SETTING, TRACER_SAMPLER_PROBABILITY))
+            new ClusterSettings(
+                Settings.EMPTY,
+                Set.of(TRACER_ENABLED_SETTING, TRACER_SAMPLER_PROBABILITY, TRACER_SAMPLER_ACTION_PROBABILITY)
+            )
         );
 
         final TracingTelemetry tracingTelemetry = new MockTracingTelemetry();