Thread pool unit test

dubbo-common/src/test/java/com/alibaba/dubbo/common/threadpool/AbortPolicyWithReportTest.java → dubbo-common/src/test/java/com/alibaba/dubbo/common/threadpool/support/AbortPolicyWithReportTest.java

@@ -6,15 +6,15 @@
  * (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
+ *      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 com.alibaba.dubbo.common.threadpool;
+package com.alibaba.dubbo.common.threadpool.support;
 
 import com.alibaba.dubbo.common.URL;
 import com.alibaba.dubbo.common.threadpool.support.AbortPolicyWithReport;
@@ -37,11 +37,11 @@ public void run() {
                     System.out.println("hello");
                 }
             }, (ThreadPoolExecutor) Executors.newFixedThreadPool(1));
-        }catch (RejectedExecutionException rj){
-
+        } catch (RejectedExecutionException rj) {
+            // ignore
         }
 
-        Thread.currentThread().sleep(1000);
+        Thread.sleep(1000);
 
     }
 }

dubbo-common/src/test/java/com/alibaba/dubbo/common/threadpool/support/cached/CachedThreadPoolTest.java

@@ -0,0 +1,80 @@
+/*
+ * 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 com.alibaba.dubbo.common.threadpool.support.cached;
+
+import com.alibaba.dubbo.common.Constants;
+import com.alibaba.dubbo.common.URL;
+import com.alibaba.dubbo.common.threadlocal.InternalThread;
+import com.alibaba.dubbo.common.threadpool.ThreadPool;
+import com.alibaba.dubbo.common.threadpool.support.AbortPolicyWithReport;
+import org.hamcrest.Matchers;
+import org.junit.Test;
+
+import java.util.concurrent.BlockingQueue;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.LinkedBlockingQueue;
+import java.util.concurrent.RejectedExecutionHandler;
+import java.util.concurrent.SynchronousQueue;
+import java.util.concurrent.ThreadPoolExecutor;
+import java.util.concurrent.TimeUnit;
+
+import static org.hamcrest.Matchers.instanceOf;
+import static org.hamcrest.Matchers.is;
+import static org.hamcrest.Matchers.startsWith;
+import static org.junit.Assert.assertThat;
+
+public class CachedThreadPoolTest {
+    @Test
+    public void getExecutor1() throws Exception {
+        URL url = URL.valueOf("dubbo://10.20.130.230:20880/context/path?" +
+                Constants.THREAD_NAME_KEY + "=demo&" +
+                Constants.CORE_THREADS_KEY + "=1&" +
+                Constants.THREADS_KEY + "=2&" +
+                Constants.ALIVE_KEY + "=1000&" +
+                Constants.QUEUES_KEY + "=0");
+        ThreadPool threadPool = new CachedThreadPool();
+        ThreadPoolExecutor executor = (ThreadPoolExecutor) threadPool.getExecutor(url);
+        assertThat(executor.getCorePoolSize(), is(1));
+        assertThat(executor.getMaximumPoolSize(), is(2));
+        assertThat(executor.getQueue(), Matchers.<BlockingQueue<Runnable>>instanceOf(SynchronousQueue.class));
+        assertThat(executor.getRejectedExecutionHandler(),
+                Matchers.<RejectedExecutionHandler>instanceOf(AbortPolicyWithReport.class));
+
+        final CountDownLatch latch = new CountDownLatch(1);
+        executor.execute(new Runnable() {
+            @Override
+            public void run() {
+                Thread thread = Thread.currentThread();
+                assertThat(thread, instanceOf(InternalThread.class));
+                assertThat(thread.getName(), startsWith("demo"));
+                latch.countDown();
+            }
+        });
+
+        latch.await(5000, TimeUnit.MICROSECONDS);
+        assertThat(latch.getCount(), is(0L));
+    }
+
+    @Test
+    public void getExecutor2() throws Exception {
+        URL url = URL.valueOf("dubbo://10.20.130.230:20880/context/path?" + Constants.QUEUES_KEY + "=1");
+        ThreadPool threadPool = new CachedThreadPool();
+        ThreadPoolExecutor executor = (ThreadPoolExecutor) threadPool.getExecutor(url);
+        assertThat(executor.getQueue(), Matchers.<BlockingQueue<Runnable>>instanceOf(LinkedBlockingQueue.class));
+    }
+}

dubbo-common/src/test/java/com/alibaba/dubbo/common/threadpool/support/eager/EagerThreadPoolTest.java

@@ -0,0 +1,82 @@
+/*
+ * 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 com.alibaba.dubbo.common.threadpool.support.eager;
+
+import com.alibaba.dubbo.common.Constants;
+import com.alibaba.dubbo.common.URL;
+import com.alibaba.dubbo.common.threadlocal.InternalThread;
+import com.alibaba.dubbo.common.threadpool.ThreadPool;
+import com.alibaba.dubbo.common.threadpool.support.AbortPolicyWithReport;
+import org.hamcrest.Matchers;
+import org.junit.Test;
+
+import java.util.concurrent.BlockingQueue;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.RejectedExecutionHandler;
+import java.util.concurrent.ThreadPoolExecutor;
+import java.util.concurrent.TimeUnit;
+
+import static org.hamcrest.Matchers.instanceOf;
+import static org.hamcrest.Matchers.is;
+import static org.hamcrest.Matchers.startsWith;
+import static org.junit.Assert.assertThat;
+
+public class EagerThreadPoolTest {
+    @Test
+    public void getExecutor1() throws Exception {
+        URL url = URL.valueOf("dubbo://10.20.130.230:20880/context/path?" +
+                Constants.THREAD_NAME_KEY + "=demo&" +
+                Constants.CORE_THREADS_KEY + "=1&" +
+                Constants.THREADS_KEY + "=2&" +
+                Constants.ALIVE_KEY + "=1000&" +
+                Constants.QUEUES_KEY + "=0");
+        ThreadPool threadPool = new EagerThreadPool();
+        ThreadPoolExecutor executor = (ThreadPoolExecutor) threadPool.getExecutor(url);
+        assertThat(executor, instanceOf(EagerThreadPoolExecutor.class));
+        assertThat(executor.getCorePoolSize(), is(1));
+        assertThat(executor.getMaximumPoolSize(), is(2));
+        assertThat(executor.getKeepAliveTime(TimeUnit.MILLISECONDS), is(1000L));
+        assertThat(executor.getQueue().remainingCapacity(), is(1));
+        assertThat(executor.getQueue(), Matchers.<BlockingQueue<Runnable>>instanceOf(TaskQueue.class));
+        assertThat(executor.getRejectedExecutionHandler(),
+                Matchers.<RejectedExecutionHandler>instanceOf(AbortPolicyWithReport.class));
+
+        final CountDownLatch latch = new CountDownLatch(1);
+        executor.execute(new Runnable() {
+            @Override
+            public void run() {
+                Thread thread = Thread.currentThread();
+                assertThat(thread, instanceOf(InternalThread.class));
+                assertThat(thread.getName(), startsWith("demo"));
+                latch.countDown();
+            }
+        });
+
+        latch.await(5000, TimeUnit.MICROSECONDS);
+        assertThat(latch.getCount(), is(0L));
+    }
+
+    @Test
+    public void getExecutor2() throws Exception {
+        URL url = URL.valueOf("dubbo://10.20.130.230:20880/context/path?" + Constants.QUEUES_KEY + "=2");
+        ThreadPool threadPool = new EagerThreadPool();
+        ThreadPoolExecutor executor = (ThreadPoolExecutor) threadPool.getExecutor(url);
+        assertThat(executor.getQueue().remainingCapacity(), is(2));
+    }
+
+}

dubbo-common/src/test/java/com/alibaba/dubbo/common/threadpool/support/eager/TaskQueueTest.java

@@ -0,0 +1,89 @@
+/*
+ * 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 com.alibaba.dubbo.common.threadpool.support.eager;
+
+import org.junit.Test;
+import org.mockito.Mockito;
+
+import java.util.concurrent.RejectedExecutionException;
+import java.util.concurrent.TimeUnit;
+
+import static org.hamcrest.Matchers.is;
+import static org.junit.Assert.assertThat;
+import static org.mockito.Mockito.mock;
+
+public class TaskQueueTest {
+
+    @Test(expected = RejectedExecutionException.class)
+    public void testOffer1() throws Exception {
+        TaskQueue<Runnable> queue = new TaskQueue<Runnable>(1);
+        queue.offer(mock(Runnable.class));
+    }
+
+    @Test
+    public void testOffer2() throws Exception {
+        TaskQueue<Runnable> queue = new TaskQueue<Runnable>(1);
+        EagerThreadPoolExecutor executor = mock(EagerThreadPoolExecutor.class);
+        Mockito.when(executor.getPoolSize()).thenReturn(2);
+        Mockito.when(executor.getSubmittedTaskCount()).thenReturn(1);
+        queue.setExecutor(executor);
+        assertThat(queue.offer(mock(Runnable.class)), is(true));
+    }
+
+    @Test
+    public void testOffer3() throws Exception {
+        TaskQueue<Runnable> queue = new TaskQueue<Runnable>(1);
+        EagerThreadPoolExecutor executor = mock(EagerThreadPoolExecutor.class);
+        Mockito.when(executor.getPoolSize()).thenReturn(2);
+        Mockito.when(executor.getSubmittedTaskCount()).thenReturn(2);
+        Mockito.when(executor.getMaximumPoolSize()).thenReturn(4);
+        queue.setExecutor(executor);
+        assertThat(queue.offer(mock(Runnable.class)), is(false));
+    }
+
+    @Test
+    public void testOffer4() throws Exception {
+        TaskQueue<Runnable> queue = new TaskQueue<Runnable>(1);
+        EagerThreadPoolExecutor executor = mock(EagerThreadPoolExecutor.class);
+        Mockito.when(executor.getPoolSize()).thenReturn(4);
+        Mockito.when(executor.getSubmittedTaskCount()).thenReturn(4);
+        Mockito.when(executor.getMaximumPoolSize()).thenReturn(4);
+        queue.setExecutor(executor);
+        assertThat(queue.offer(mock(Runnable.class)), is(true));
+    }
+
+    @Test(expected = RejectedExecutionException.class)
+    public void testRetryOffer1() throws Exception {
+        TaskQueue<Runnable> queue = new TaskQueue<Runnable>(1);
+        EagerThreadPoolExecutor executor = mock(EagerThreadPoolExecutor.class);
+        Mockito.when(executor.isShutdown()).thenReturn(true);
+        queue.setExecutor(executor);
+        queue.retryOffer(mock(Runnable.class), 1000, TimeUnit.MILLISECONDS);
+    }
+
+
+    @Test
+    public void testRetryOffer2() throws Exception {
+        TaskQueue<Runnable> queue = new TaskQueue<Runnable>(1);
+        EagerThreadPoolExecutor executor = mock(EagerThreadPoolExecutor.class);
+        Mockito.when(executor.isShutdown()).thenReturn(false);
+        queue.setExecutor(executor);
+        assertThat(queue.retryOffer(mock(Runnable.class), 1000, TimeUnit.MILLISECONDS), is(true));
+    }
+
+}

dubbo-common/src/test/java/com/alibaba/dubbo/common/threadpool/support/fixed/FixedThreadPoolTest.java

@@ -0,0 +1,81 @@
+/*
+ * 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 com.alibaba.dubbo.common.threadpool.support.fixed;
+
+import com.alibaba.dubbo.common.Constants;
+import com.alibaba.dubbo.common.URL;
+import com.alibaba.dubbo.common.threadlocal.InternalThread;
+import com.alibaba.dubbo.common.threadpool.ThreadPool;
+import com.alibaba.dubbo.common.threadpool.support.AbortPolicyWithReport;
+import com.alibaba.dubbo.common.threadpool.support.limited.LimitedThreadPool;
+import org.hamcrest.Matchers;
+import org.junit.Test;
+
+import java.util.concurrent.BlockingQueue;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.LinkedBlockingQueue;
+import java.util.concurrent.RejectedExecutionHandler;
+import java.util.concurrent.SynchronousQueue;
+import java.util.concurrent.ThreadPoolExecutor;
+import java.util.concurrent.TimeUnit;
+
+import static org.hamcrest.Matchers.instanceOf;
+import static org.hamcrest.Matchers.is;
+import static org.hamcrest.Matchers.startsWith;
+import static org.junit.Assert.assertThat;
+
+public class FixedThreadPoolTest {
+    @Test
+    public void getExecutor1() throws Exception {
+        URL url = URL.valueOf("dubbo://10.20.130.230:20880/context/path?" +
+                Constants.THREAD_NAME_KEY + "=demo&" +
+                Constants.CORE_THREADS_KEY + "=1&" +
+                Constants.THREADS_KEY + "=2&" +
+                Constants.QUEUES_KEY + "=0");
+        ThreadPool threadPool = new FixedThreadPool();
+        ThreadPoolExecutor executor = (ThreadPoolExecutor) threadPool.getExecutor(url);
+        assertThat(executor.getCorePoolSize(), is(2));
+        assertThat(executor.getMaximumPoolSize(), is(2));
+        assertThat(executor.getKeepAliveTime(TimeUnit.MILLISECONDS), is(0L));
+        assertThat(executor.getQueue(), Matchers.<BlockingQueue<Runnable>>instanceOf(SynchronousQueue.class));
+        assertThat(executor.getRejectedExecutionHandler(),
+                Matchers.<RejectedExecutionHandler>instanceOf(AbortPolicyWithReport.class));
+
+        final CountDownLatch latch = new CountDownLatch(1);
+        executor.execute(new Runnable() {
+            @Override
+            public void run() {
+                Thread thread = Thread.currentThread();
+                assertThat(thread, instanceOf(InternalThread.class));
+                assertThat(thread.getName(), startsWith("demo"));
+                latch.countDown();
+            }
+        });
+
+        latch.await(5000, TimeUnit.MICROSECONDS);
+        assertThat(latch.getCount(), is(0L));
+    }
+
+    @Test
+    public void getExecutor2() throws Exception {
+        URL url = URL.valueOf("dubbo://10.20.130.230:20880/context/path?" + Constants.QUEUES_KEY + "=1");
+        ThreadPool threadPool = new FixedThreadPool();
+        ThreadPoolExecutor executor = (ThreadPoolExecutor) threadPool.getExecutor(url);
+        assertThat(executor.getQueue(), Matchers.<BlockingQueue<Runnable>>instanceOf(LinkedBlockingQueue.class));
+    }
+}