Cleaned up observation code after listener tests

This commit is contained in:
Glenn Renfro
2023-07-26 14:21:29 -04:00
parent 16542604e9
commit 7f6eca70ea

View File

@@ -24,6 +24,9 @@ import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.Set; import java.util.Set;
import io.micrometer.observation.Observation;
import io.micrometer.observation.ObservationRegistry;
import io.micrometer.observation.tck.TestObservationRegistry;
import org.junit.jupiter.api.AfterEach; import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
@@ -86,6 +89,12 @@ public class TaskLifecycleListenerTests {
if (this.context != null && this.context.isActive()) { if (this.context != null && this.context.isActive()) {
this.context.close(); this.context.close();
} }
ObservationRegistry observationRegistry = TestObservationRegistry.create();
if (observationRegistry.getCurrentObservationScope() != null) {
Observation.Scope scope = observationRegistry.getCurrentObservationScope();
scope.close();
scope.getCurrentObservation().stop();
}
} }
@Test @Test