From b29209e59ac44760d071094c30a6507719e33121 Mon Sep 17 00:00:00 2001
From: anotherJJz <470623352@qq.com>
Date: Fri, 18 Sep 2020 13:57:38 +0800
Subject: [PATCH] [ISSUE #2152] Use mock style in unit test for time-consuming
 (#2308)

---
 .../consumer/DefaultLitePullConsumerTest.java | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

diff --git a/client/src/test/java/org/apache/rocketmq/client/consumer/DefaultLitePullConsumerTest.java b/client/src/test/java/org/apache/rocketmq/client/consumer/DefaultLitePullConsumerTest.java
index 3726a5b3..66832706 100644
--- a/client/src/test/java/org/apache/rocketmq/client/consumer/DefaultLitePullConsumerTest.java
+++ b/client/src/test/java/org/apache/rocketmq/client/consumer/DefaultLitePullConsumerTest.java
@@ -20,10 +20,8 @@ package org.apache.rocketmq.client.consumer;
 import java.io.ByteArrayOutputStream;
 import java.lang.reflect.Field;
 import java.net.InetSocketAddress;
-import java.util.Collections;
-import java.util.HashSet;
-import java.util.List;
-import java.util.Set;
+import java.util.*;
+
 import org.apache.rocketmq.client.ClientConfig;
 import org.apache.rocketmq.client.consumer.store.OffsetStore;
 import org.apache.rocketmq.client.consumer.store.ReadOffsetType;
@@ -515,11 +513,14 @@ public class DefaultLitePullConsumerTest {
 
     @Test
     public void testConsumerAfterShutdown() throws Exception {
-        DefaultLitePullConsumer defaultLitePullConsumer = createStartLitePullConsumer();
-        defaultLitePullConsumer.setNamesrvAddr("127.0.0.1:9876");
-        defaultLitePullConsumer.subscribe(topic, "*");
+        DefaultLitePullConsumer defaultLitePullConsumer = createSubscribeLitePullConsumer();
+
+        DefaultLitePullConsumer mockConsumer = spy(defaultLitePullConsumer);
+        when(mockConsumer.poll(anyLong())).thenReturn(new ArrayList<>());
+
         new AsyncConsumer().executeAsync(defaultLitePullConsumer);
-        Thread.sleep(10 * 1000);
+
+        Thread.sleep(100);
         defaultLitePullConsumer.shutdown();
         assertThat(defaultLitePullConsumer.isRunning()).isFalse();
     }
@@ -531,9 +532,7 @@ public class DefaultLitePullConsumerTest {
                 public void run() {
                     while (consumer.isRunning()) {
                         List<MessageExt> poll = consumer.poll(2 * 1000);
-                        System.out.println("consumer is still running");
                     }
-                    System.out.println("consumer shutdown");
                 }
             }).start();
         }
-- 
GitLab