Merge pull request #1554 from keranbingaa/unit-test
[ISSUE #1528]modify unit test of producer
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
index cc0b801..818c94a 100644
--- 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 @@
@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 @@
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