From 73814f31d81edcc647e81290c016a59d0e5e74ca Mon Sep 17 00:00:00 2001
From: Mercy <mercyblitz@gmail.com>
Date: Fri, 10 Jan 2025 19:12:58 +0800
Subject: [PATCH] Polish #57

---
 .../webmvc/annotation/AbstractEnableWebMvcExtensionTest.java | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/microsphere-spring-webmvc/src/test/java/io/microsphere/spring/webmvc/annotation/AbstractEnableWebMvcExtensionTest.java b/microsphere-spring-webmvc/src/test/java/io/microsphere/spring/webmvc/annotation/AbstractEnableWebMvcExtensionTest.java
index 57e19b109..5bfbfe996 100644
--- a/microsphere-spring-webmvc/src/test/java/io/microsphere/spring/webmvc/annotation/AbstractEnableWebMvcExtensionTest.java
+++ b/microsphere-spring-webmvc/src/test/java/io/microsphere/spring/webmvc/annotation/AbstractEnableWebMvcExtensionTest.java
@@ -121,6 +121,8 @@ public void test() throws Exception {
      */
     @EventListener(WebEndpointMappingsReadyEvent.class)
     public void onWebEndpointMappingsReadyEvent(WebEndpointMappingsReadyEvent event) {
+        assertTrue(publishEvents);
+
         // Only TestController
         Collection<WebEndpointMapping> mappings = event.getMappings();
         assertEquals(1, mappings.size());
@@ -137,6 +139,8 @@ public void onWebEndpointMappingsReadyEvent(WebEndpointMappingsReadyEvent event)
      */
     @EventListener(HandlerMethodArgumentsResolvedEvent.class)
     public void onHandlerMethodArgumentsResolvedEvent(HandlerMethodArgumentsResolvedEvent event) {
+        assertTrue(publishEvents);
+
         Method method = event.getMethod();
         assertEquals("echo", method.getName());
         assertEquals(String.class, method.getReturnType());
@@ -156,5 +160,6 @@ public void onHandlerMethodArgumentsResolvedEvent(HandlerMethodArgumentsResolved
         Object[] arguments = event.getArguments();
         assertEquals(1, arguments.length);
         assertEquals("hello", arguments[0]);
+
     }
 }