Merge pull request #1199 from z-y-hang/develop

test:add RejectedProxyInvocationHandlerTest

hippo4j-core/src/test/java/cn/hippo4j/core/proxy/RejectedProxyInvocationHandlerTest.java

@@ -0,0 +1,55 @@
+package cn.hippo4j.core.proxy;
+
+import cn.hippo4j.common.api.ThreadPoolCheckAlarm;
+import cn.hippo4j.common.config.ApplicationContextHolder;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.jupiter.api.Assertions;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.MockedStatic;
+import org.mockito.Mockito;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.springframework.context.ApplicationContext;
+import java.lang.reflect.InvocationTargetException;
+import java.lang.reflect.Method;
+import java.util.concurrent.atomic.AtomicLong;
+
+@RunWith(MockitoJUnitRunner.class)
+public class RejectedProxyInvocationHandlerTest {
+
+    @Mock
+    private ThreadPoolCheckAlarm mockAlarmHandler;
+
+    @Mock
+    private ApplicationContext applicationContext;
+
+    @Mock
+    private Object target;
+
+    @Mock
+    private Method mockMethod;
+
+    private RejectedProxyInvocationHandler handler;
+
+    private AtomicLong rejectCount;
+    @Before
+    public void setUp() {
+        String threadPoolId = "test-pool";
+        rejectCount = new AtomicLong(0);
+        handler = new RejectedProxyInvocationHandler(target, threadPoolId, rejectCount);
+    }
+
+    @Test
+    public void testInvoke() throws Throwable {
+        Object[] mockArgs = new Object[] {"arg1", "arg2"};
+        MockedStatic<ApplicationContextHolder> mockedStatic = Mockito.mockStatic(ApplicationContextHolder.class);
+        mockedStatic.when(ApplicationContextHolder::getInstance).thenReturn(applicationContext);
+        mockedStatic.when(() -> ApplicationContextHolder.getBean(ThreadPoolCheckAlarm.class)).thenReturn(mockAlarmHandler);
+        Mockito.doNothing().when(mockAlarmHandler).asyncSendRejectedAlarm("test-pool");
+        handler.invoke(null, mockMethod, mockArgs);
+        Mockito.doThrow(new InvocationTargetException(new Throwable())).when(mockMethod).invoke(target, mockArgs);
+        Assertions.assertThrows(Throwable.class, () -> handler.invoke(null, mockMethod, mockArgs));
+        Assertions.assertSame(rejectCount.get(), 2L);
+    }
+}