Fix flaky test in Md5UtilTest (#1494)

* Fix flaky test in Md5UtilTest * Remove unused imports --------- Co-authored-by: bbelide2 <[email protected]>
opengoofy · Oct 6, 2023 · cf854b3 · cf854b3
1 parent 79dc8f5
commit cf854b3
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/infra/common/src/test/java/cn/hippo4j/common/toolkit/Md5UtilTest.java b/infra/common/src/test/java/cn/hippo4j/common/toolkit/Md5UtilTest.java
@@ -19,12 +19,16 @@
 
 import cn.hippo4j.common.model.ThreadPoolParameterInfo;
 import org.junit.Test;
+import org.mockito.MockedStatic;
 
 import java.io.IOException;
 import java.security.NoSuchAlgorithmException;
 import java.util.ArrayList;
 import java.util.List;
 
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.mockStatic;
+
 public class Md5UtilTest {
 
     @Test
@@ -50,12 +54,15 @@ public void assetEncodeHexString() {
 
     @Test
     public void assetGetTpContentMd5() {
-        String md5Result = "ef5ea7cb47377fb9fb85a7125e76715d";
-        ThreadPoolParameterInfo threadPoolParameterInfo = ThreadPoolParameterInfo.builder().tenantId("prescription")
-                .itemId("dynamic-threadpool-example").tpId("message-consume").content("描述信息").corePoolSize(1)
-                .maximumPoolSize(2).queueType(1).capacity(4).keepAliveTime(513L).executeTimeOut(null).rejectedType(4)
-                .isAlarm(1).capacityAlarm(80).livenessAlarm(80).allowCoreThreadTimeOut(1).build();
-        Assert.isTrue(md5Result.equals(Md5Util.getTpContentMd5(threadPoolParameterInfo)));
+        final ThreadPoolParameterInfo threadPoolParameterInfo = new ThreadPoolParameterInfo();
+        final String mockContent = "mockContent";
+        final String mockContentMd5 = "34cf17bc632ece6e4c81a4ce8aa97d5e";
+        try (final MockedStatic<ContentUtil> mockedContentUtil = mockStatic(ContentUtil.class)) {
+            mockedContentUtil.when(() -> ContentUtil.getPoolContent(threadPoolParameterInfo)).thenReturn(mockContent);
+            final String result = Md5Util.getTpContentMd5(threadPoolParameterInfo);
+            Assert.isTrue(result.equals(mockContentMd5));
+            mockedContentUtil.verify(() -> ContentUtil.getPoolContent(threadPoolParameterInfo), times(1));
+        }
     }
 
     @Test