Merge pull request alibaba#1554 from keranbingaa/unit-test

[ISSUE alibaba#1528]modify unit test of producer
CST11021 · Nov 14, 2019 · 7028af6 · 7028af6
2 parents e735fff + b8e1be8
commit 7028af6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/client/src/test/java/org/apache/rocketmq/client/producer/DefaultMQProducerTest.java b/client/src/test/java/org/apache/rocketmq/client/producer/DefaultMQProducerTest.java
@@ -189,7 +189,7 @@ public void testSendMessageSync_WithBodyCompressed() throws RemotingException, I
     @Test
     public void testSendMessageAsync_Success() throws RemotingException, InterruptedException, MQBrokerException, MQClientException {
         final CountDownLatch countDownLatch = new CountDownLatch(1);
-//        final AtomicInteger cc = new AtomicInteger(0);
+        final AtomicInteger cc = new AtomicInteger(0);
         when(mQClientAPIImpl.getTopicRouteInfoFromNameServer(anyString(), anyLong())).thenReturn(createTopicRoute());
         producer.send(message, new SendCallback() {
             @Override
@@ -198,15 +198,15 @@ public void onSuccess(SendResult sendResult) {
                 assertThat(sendResult.getOffsetMsgId()).isEqualTo("123");
                 assertThat(sendResult.getQueueOffset()).isEqualTo(456L);
                 countDownLatch.countDown();
-//                cc.incrementAndGet();
+                cc.incrementAndGet();
             }
 
             @Override
             public void onException(Throwable e) {
             }
         });
-        countDownLatch.await(3000L, TimeUnit.MILLISECONDS);
-//        assertThat(cc.get()).isEqualTo(1);
+        countDownLatch.await(8000L, TimeUnit.MILLISECONDS);
+        assertThat(cc.get()).isEqualTo(1);
     }
 
     @Test