Avoid potential ClassCastException when handling context closed event. (

#24)
wiremock · Jan 15, 2024 · f0abfee · f0abfee
1 parent 39f15e6
commit f0abfee
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/...ing-boot/src/main/java/com/maciejwalkowiak/wiremock/spring/WireMockContextCustomizer.java b/...ing-boot/src/main/java/com/maciejwalkowiak/wiremock/spring/WireMockContextCustomizer.java
@@ -12,7 +12,6 @@
 import org.slf4j.LoggerFactory;
 
 import org.springframework.boot.test.util.TestPropertyValues;
-import org.springframework.context.ApplicationListener;
 import org.springframework.context.ConfigurableApplicationContext;
 import org.springframework.context.event.ContextClosedEvent;
 import org.springframework.core.env.Environment;
@@ -82,9 +81,11 @@ private void resolveOrCreateWireMockServer(ConfigurableApplicationContext contex
             Store.INSTANCE.store(context, options.name(), newServer);
 
             // add shutdown hook
-            context.addApplicationListener((ApplicationListener<ContextClosedEvent>) event -> {
-                LOGGER.info("Stopping WireMockServer with name '{}'", options.name());
-                newServer.stop();
+            context.addApplicationListener(event -> {
+                if (event instanceof ContextClosedEvent) {
+                    LOGGER.info("Stopping WireMockServer with name '{}'", options.name());
+                    newServer.stop();
+                }
             });
 
             // configure Spring environment property