diff --git a/platform/diagnostic/telemetry/rt/testSrc/com/intellij/platform/diagnostic/telemetry/rt/context/TelemetryContextConfiguringTestCase.kt b/platform/diagnostic/telemetry/rt/testSrc/com/intellij/platform/diagnostic/telemetry/rt/context/TelemetryContextConfiguringTestCase.kt index 4819d3edbe0d..edb6fe5e0b00 100644 --- a/platform/diagnostic/telemetry/rt/testSrc/com/intellij/platform/diagnostic/telemetry/rt/context/TelemetryContextConfiguringTestCase.kt +++ b/platform/diagnostic/telemetry/rt/testSrc/com/intellij/platform/diagnostic/telemetry/rt/context/TelemetryContextConfiguringTestCase.kt @@ -2,49 +2,22 @@ package com.intellij.platform.diagnostic.telemetry.rt.context import io.opentelemetry.api.GlobalOpenTelemetry -import io.opentelemetry.api.trace.Span -import io.opentelemetry.api.trace.SpanContext import io.opentelemetry.api.trace.propagation.W3CTraceContextPropagator import io.opentelemetry.context.propagation.ContextPropagators import io.opentelemetry.sdk.OpenTelemetrySdk -import org.junit.jupiter.api.AfterAll -import org.junit.jupiter.api.Assertions.assertEquals -import org.junit.jupiter.api.BeforeAll import org.junit.jupiter.api.BeforeEach abstract class TelemetryContextConfiguringTestCase { - companion object { - @BeforeAll - @JvmStatic - fun beforeAll() { - configureGlobalTelemetry() - } - - @AfterAll - @JvmStatic - fun afterAll() { - shutdownGlobalTelemetry() - } - - private fun configureGlobalTelemetry() { - OpenTelemetrySdk.builder() - .setPropagators(ContextPropagators.create(W3CTraceContextPropagator.getInstance())) - .buildAndRegisterGlobal() - } - - private fun shutdownGlobalTelemetry() { - GlobalOpenTelemetry.resetForTest() - } - } - @BeforeEach - fun beforeEach() { - assertCurrentContextIsEmpty() + open fun setUp() { + GlobalOpenTelemetry.resetForTest() + configureGlobalTelemetry() } - private fun assertCurrentContextIsEmpty() { - val currentContext = Span.current().spanContext - assertEquals(SpanContext.getInvalid(), currentContext) + private fun configureGlobalTelemetry() { + OpenTelemetrySdk.builder() + .setPropagators(ContextPropagators.create(W3CTraceContextPropagator.getInstance())) + .buildAndRegisterGlobal() } }