From 44418df94a6a238b156a37aa9c0dd92c826b5add Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=97=B6=E9=97=B4=E9=82=AE=E9=80=92=E5=91=98?=
 <78356082+barret-yzh@users.noreply.github.com>
Date: Fri, 26 May 2023 16:36:12 +0800
Subject: [PATCH] Update MessageWrapperTest

---
 .../hippo4j/common/toolkit/MessageWrapperTest | 20 +++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/infra/common/src/test/java/cn/hippo4j/common/toolkit/MessageWrapperTest b/infra/common/src/test/java/cn/hippo4j/common/toolkit/MessageWrapperTest
index 06d66c36..7a18ea75 100644
--- a/infra/common/src/test/java/cn/hippo4j/common/toolkit/MessageWrapperTest
+++ b/infra/common/src/test/java/cn/hippo4j/common/toolkit/MessageWrapperTest
@@ -19,10 +19,14 @@ package cn.hippo4j.common.monitor;
 
 import cn.hippo4j.common.monitor.MessageTypeEnum;
 import cn.hippo4j.common.monitor.MessageWrapper;
+import org.junit.Assert;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
-import java.util.*;
-import static org.junit.jupiter.api.Assertions.*;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Arrays;
+import java.util.List;
+import java.util.Collections;
 
 public class MessageWrapperTest {
 
@@ -48,17 +52,17 @@ public class MessageWrapperTest {
 
     @Test
     void testGetContentParams() {
-        assertEquals(contentParams, messageWrapper.getContentParams());
+        Assert.assertEquals(contentParams, messageWrapper.getContentParams());
     }
 
     @Test
     void testGetResponseClass() {
-        assertEquals(responseClass, messageWrapper.getResponseClass());
+        Assert.assertEquals(responseClass, messageWrapper.getResponseClass());
     }
 
     @Test
     void testGetMessageType() {
-        assertEquals(messageType, messageWrapper.getMessageType());
+        Assert.assertEquals(messageType, messageWrapper.getMessageType());
     }
 
     @Test
@@ -69,8 +73,8 @@ public class MessageWrapperTest {
         messageWrapper.setContentParams(newContentParams);
         messageWrapper.setResponseClass(newResponseClass);
         messageWrapper.setMessageType(newMessageType);
-        assertEquals(newContentParams, messageWrapper.getContentParams());
-        assertEquals(newResponseClass, messageWrapper.getResponseClass());
-        assertEquals(newMessageType, messageWrapper.getMessageType());
+        Assert.assertEquals(newContentParams, messageWrapper.getContentParams());
+        Assert.assertEquals(newResponseClass, messageWrapper.getResponseClass());
+        Assert.assertEquals(newMessageType, messageWrapper.getMessageType());
     }
 }