diff --git a/hippo4j-core/pom.xml b/hippo4j-core/pom.xml index 2429039c..4067d24d 100644 --- a/hippo4j-core/pom.xml +++ b/hippo4j-core/pom.xml @@ -23,6 +23,7 @@ org.mockito mockito-inline + test diff --git a/hippo4j-core/src/test/java/cn/hippo4j/core/proxy/RejectedProxyInvocationHandlerTest.java b/hippo4j-core/src/test/java/cn/hippo4j/core/proxy/RejectedProxyInvocationHandlerTest.java index 5e47748d..fecd2ad0 100644 --- a/hippo4j-core/src/test/java/cn/hippo4j/core/proxy/RejectedProxyInvocationHandlerTest.java +++ b/hippo4j-core/src/test/java/cn/hippo4j/core/proxy/RejectedProxyInvocationHandlerTest.java @@ -11,12 +11,10 @@ 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 { @@ -28,14 +26,17 @@ public class RejectedProxyInvocationHandlerTest { @Mock private Object target; + @Mock private Method mockMethod; + private RejectedProxyInvocationHandler handler; + private AtomicLong rejectCount; @Before public void setUp() { String threadPoolId = "test-pool"; - AtomicLong rejectCount = new AtomicLong(0); + rejectCount = new AtomicLong(0); handler = new RejectedProxyInvocationHandler(target, threadPoolId, rejectCount); } @@ -47,8 +48,8 @@ public class RejectedProxyInvocationHandlerTest { 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); } }