Fix flaky test (#1623)

open-telemetry · Dec 24, 2024 · 0d77587 · 0d77587
1 parent b600c74
commit 0d77587
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...pentelemetry/contrib/sampler/consistent/ConsistentReservoirSamplingSpanProcessorTest.java b/...pentelemetry/contrib/sampler/consistent/ConsistentReservoirSamplingSpanProcessorTest.java
@@ -243,9 +243,11 @@ void exportSpansToMultipleServices() {
 
   @Test
   void ignoresNullSpans() {
+    SpanExporter mockSpanExporter = mock(SpanExporter.class);
+    when(mockSpanExporter.shutdown()).thenReturn(CompletableResultCode.ofSuccess());
     SpanProcessor processor =
         ConsistentReservoirSamplingSpanProcessor.create(
-            mock(SpanExporter.class), RESERVOIR_SIZE, EXPORT_PERIOD_100_MILLIS_AS_NANOS);
+            mockSpanExporter, RESERVOIR_SIZE, EXPORT_PERIOD_100_MILLIS_AS_NANOS);
     assertThatCode(
             () -> {
               processor.onStart(null, null);