add RejectedExecutionHandler test case

sofastack · Oct 3, 2024 · 577a0fd · 577a0fd
1 parent 3467886
commit 577a0fd
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/core/api/src/test/java/com/alipay/sofa/rpc/config/UserThreadPoolManagerTest.java b/core/api/src/test/java/com/alipay/sofa/rpc/config/UserThreadPoolManagerTest.java
@@ -18,6 +18,7 @@
 
 import com.alipay.sofa.rpc.server.UserThreadPool;
 import com.alipay.sofa.rpc.server.UserVirtualThreadPool;
+import com.alipay.sofa.rpc.server.SofaRejectedExecutionHandler;
 import org.junit.Assert;
 import org.junit.Test;
 import java.util.Set;
@@ -67,6 +68,14 @@ public void userThreadPoolBuildTest() {
         Assert.assertNull(result);
     }
 
+    @Test
+    public void testRejectedExecutionHandler(){
+        UserThreadPool userThreadPool = new UserThreadPool();
+        Executor executorService = userThreadPool.getUserExecutor();
+        Assert.assertTrue(executorService instanceof ThreadPoolExecutor);
+        Assert.assertTrue(((ThreadPoolExecutor) executorService).getRejectedExecutionHandler() instanceof SofaRejectedExecutionHandler);
+    }
+
     @Test
     public void userThreadPoolCompatibleTest() {
         UserThreadPool userThreadPool = new UserThreadPool();