From 088970d80abcb5f6a644db5d676209a1e36332dc Mon Sep 17 00:00:00 2001
From: pan_yujie <646836760@qq.com>
Date: Thu, 27 Apr 2023 23:25:38 +0800
Subject: [PATCH] test: add ThreadPoolRunStateHandler

---
 hippo4j-core/pom.xml                          |   6 +-
 .../state/ThreadPoolRunStateHandlerTest.java  | 131 ++++++++++++++++++
 2 files changed, 136 insertions(+), 1 deletion(-)
 create mode 100644 hippo4j-core/src/test/java/cn/hippo4j/core/executor/state/ThreadPoolRunStateHandlerTest.java

diff --git a/hippo4j-core/pom.xml b/hippo4j-core/pom.xml
index 7088a3f1..b3fccec9 100644
--- a/hippo4j-core/pom.xml
+++ b/hippo4j-core/pom.xml
@@ -8,7 +8,6 @@
         <version>${revision}</version>
     </parent>
     <artifactId>hippo4j-core</artifactId>
-
     <dependencies>
         <dependency>
             <groupId>cn.hippo4j</groupId>
@@ -20,5 +19,10 @@
             <artifactId>spring-boot-starter-test</artifactId>
             <scope>test</scope>
         </dependency>
+        <dependency>
+            <groupId>org.mockito</groupId>
+            <artifactId>mockito-inline</artifactId>
+            <scope>test</scope>
+        </dependency>
     </dependencies>
 </project>
diff --git a/hippo4j-core/src/test/java/cn/hippo4j/core/executor/state/ThreadPoolRunStateHandlerTest.java b/hippo4j-core/src/test/java/cn/hippo4j/core/executor/state/ThreadPoolRunStateHandlerTest.java
new file mode 100644
index 00000000..96e24ef5
--- /dev/null
+++ b/hippo4j-core/src/test/java/cn/hippo4j/core/executor/state/ThreadPoolRunStateHandlerTest.java
@@ -0,0 +1,131 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package cn.hippo4j.core.executor.state;
+
+import cn.hippo4j.common.model.ManyThreadPoolRunStateInfo;
+import cn.hippo4j.common.model.ThreadPoolRunStateInfo;
+import cn.hippo4j.common.toolkit.BeanUtil;
+import cn.hippo4j.common.toolkit.ByteConvertUtil;
+import cn.hippo4j.common.toolkit.MemoryUtil;
+import cn.hippo4j.common.toolkit.StringUtil;
+import cn.hippo4j.core.executor.DynamicThreadPoolWrapper;
+import cn.hippo4j.core.executor.manage.GlobalThreadPoolManage;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.Mock;
+import org.mockito.MockedStatic;
+import org.mockito.MockitoAnnotations;
+import org.mockito.junit.jupiter.MockitoExtension;
+import org.springframework.core.env.ConfigurableEnvironment;
+
+import java.util.concurrent.SynchronousQueue;
+import java.util.concurrent.ThreadPoolExecutor;
+import java.util.concurrent.TimeUnit;
+
+import static cn.hippo4j.core.toolkit.IdentifyUtil.CLIENT_IDENTIFICATION_VALUE;
+import static org.mockito.Mockito.*;
+
+@ExtendWith(MockitoExtension.class)
+public class ThreadPoolRunStateHandlerTest {
+
+    @Mock
+    ConfigurableEnvironment environment;
+
+    ThreadPoolRunStateInfo poolRunStateInfo = new ThreadPoolRunStateInfo();
+
+    @BeforeEach
+    void setUp() {
+        MockitoAnnotations.openMocks(this);
+        poolRunStateInfo.setCurrentLoad(poolRunStateInfo.getCurrentLoad() + "%");
+        poolRunStateInfo.setPeakLoad(poolRunStateInfo.getPeakLoad() + "%");
+        poolRunStateInfo.setTpId("1");
+    }
+
+
+    @Test
+    void testSupplement(){
+        long used = MemoryUtil.heapMemoryUsed();
+        long max = MemoryUtil.heapMemoryMax();
+        String memoryProportion = StringUtil.newBuilder(
+                "Allocation: ",
+                ByteConvertUtil.getPrintSize(used),
+                " / Maximum available: ",
+                ByteConvertUtil.getPrintSize(max)
+        );
+
+        String ipAddress = "127.0.0.1";
+
+        poolRunStateInfo.setHost(ipAddress);
+        poolRunStateInfo.setMemoryProportion(memoryProportion);
+        poolRunStateInfo.setFreeMemory(ByteConvertUtil.getPrintSize(Math.subtractExact(max, used)));
+
+        String threadPoolId = poolRunStateInfo.getTpId();
+
+        try(final MockedStatic<GlobalThreadPoolManage> globalThreadPoolManage = mockStatic(GlobalThreadPoolManage.class)) {
+            globalThreadPoolManage.when(()->GlobalThreadPoolManage.getExecutorService("1")).thenReturn(new DynamicThreadPoolWrapper());
+            DynamicThreadPoolWrapper executorService = GlobalThreadPoolManage.getExecutorService(threadPoolId);
+            Assertions.assertNotNull(executorService);
+        }
+
+        DynamicThreadPoolWrapper dynamicThreadPoolWrapperMock = mock(DynamicThreadPoolWrapper.class);
+        when(dynamicThreadPoolWrapperMock.getExecutor()).thenReturn(new ThreadPoolExecutor(2, 2, 2000, TimeUnit.SECONDS, new SynchronousQueue<>()));
+        ThreadPoolExecutor pool = dynamicThreadPoolWrapperMock.getExecutor();
+        Assertions.assertNotNull(pool);
+
+        String rejectedName;
+        rejectedName="java.util.concurrent.ThreadPoolExecutor.AbortPolicy";
+        poolRunStateInfo.setRejectedName(rejectedName);
+
+        ManyThreadPoolRunStateInfo manyThreadPoolRunStateInfo = BeanUtil.convert(poolRunStateInfo, ManyThreadPoolRunStateInfo.class);
+        manyThreadPoolRunStateInfo.setIdentify(CLIENT_IDENTIFICATION_VALUE);
+        String active = environment.getProperty("spring.profiles.active", "UNKNOWN");
+        manyThreadPoolRunStateInfo.setActive("TRUE");
+        String threadPoolState = ThreadPoolStatusHandler.getThreadPoolState(pool);
+        manyThreadPoolRunStateInfo.setState(threadPoolState);
+        Assertions.assertNotNull(manyThreadPoolRunStateInfo);
+
+
+    }
+
+    @Test
+    void testGetheapMemory(){
+        try(MockedStatic<MemoryUtil> memoryUtil = mockStatic(MemoryUtil.class)) {
+            memoryUtil.when(MemoryUtil::heapMemoryUsed).thenReturn(57534464L);
+            memoryUtil.when(MemoryUtil::heapMemoryMax).thenReturn(8566865920L);
+            Assertions.assertEquals(8566865920L, MemoryUtil.heapMemoryMax());
+            Assertions.assertEquals(57534464L, MemoryUtil.heapMemoryUsed());
+        }
+    }
+
+    @Test
+    void testMemoryProportion(){
+        long used = 57534464L;
+        long max = 8566865920L;
+        String memoryProportion = StringUtil.newBuilder(
+                "Allocation: ",
+                ByteConvertUtil.getPrintSize(used),
+                " / Maximum available: ",
+                ByteConvertUtil.getPrintSize(max));
+        Assertions.assertEquals("Allocation: 54.87MB / Maximum available: 7.98GB", memoryProportion);
+    }
+
+}
+
+