diff --git a/dd-trace-core/src/test/java/datadog/trace/core/baggage/BaggagePropagatorTelemetryTest.java b/dd-trace-core/src/test/java/datadog/trace/core/baggage/BaggagePropagatorTelemetryTest.java index 0b2b1943e06..1780f541c5f 100644 --- a/dd-trace-core/src/test/java/datadog/trace/core/baggage/BaggagePropagatorTelemetryTest.java +++ b/dd-trace-core/src/test/java/datadog/trace/core/baggage/BaggagePropagatorTelemetryTest.java @@ -11,18 +11,25 @@ import datadog.trace.api.Config; import datadog.trace.api.metrics.BaggageMetrics; import datadog.trace.api.telemetry.CoreMetricCollector; -import datadog.trace.test.util.Flaky; import java.util.Collection; import java.util.Collections; import java.util.List; import java.util.Map; import java.util.stream.Collectors; +import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; class BaggagePropagatorTelemetryTest { private static final CarrierVisitor> MAP_VISITOR = Map::forEach; + @BeforeEach + void setup() { + // Drain any metrics accumulated by other tests + CoreMetricCollector.getInstance().prepareMetrics(); + CoreMetricCollector.getInstance().drain(); + } + @Test void shouldDirectlyIncrementBaggageMetrics() { BaggageMetrics baggageMetrics = BaggageMetrics.getInstance(); @@ -68,7 +75,6 @@ void shouldIncrementTelemetryCounterWhenBaggageIsSuccessfullyExtracted() { assertTrue(baggageMetric.tags.contains("header_style:baggage")); } - @Flaky @Test void shouldDirectlyIncrementAllBaggageMetrics() { BaggageMetrics baggageMetrics = BaggageMetrics.getInstance(); @@ -122,7 +128,6 @@ void shouldDirectlyIncrementAllBaggageMetrics() { assertEquals(1, itemsTruncatedMetric.value.longValue()); } - @Flaky @Test void shouldNotIncrementTelemetryCounterWhenBaggageExtractionFails() { Config config = mock(Config.class);