From dfacfe607fc8d073eadda4a7ed7ff045a599ebf4 Mon Sep 17 00:00:00 2001 From: zhouxiang Date: Mon, 23 May 2022 19:15:26 +0800 Subject: [PATCH] [ISSUE #3949] Update LocalMessageService unit test --- .../service/message/LocalMessageServiceTest.java | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/proxy/src/test/java/org/apache/rocketmq/proxy/service/message/LocalMessageServiceTest.java b/proxy/src/test/java/org/apache/rocketmq/proxy/service/message/LocalMessageServiceTest.java index 64ba5cba71..3058951a2c 100644 --- a/proxy/src/test/java/org/apache/rocketmq/proxy/service/message/LocalMessageServiceTest.java +++ b/proxy/src/test/java/org/apache/rocketmq/proxy/service/message/LocalMessageServiceTest.java @@ -155,8 +155,8 @@ public class LocalMessageServiceTest extends InitConfigAndLoggerTest { return null; }); - CompletableFuture future = localMessageService.sendMessage(proxyContext, null, messagesList, requestHeader, 1000L); - SendResult sendResult = future.get(); + CompletableFuture> future = localMessageService.sendMessage(proxyContext, null, messagesList, requestHeader, 1000L); + SendResult sendResult = future.get().get(0); assertThat(sendResult.getSendStatus()).isEqualTo(SendStatus.SEND_OK); assertThat(sendResult.getMsgId()).isEqualTo(MessageClientIDSetter.getUniqID(message)); assertThat(sendResult.getMessageQueue()) @@ -200,8 +200,8 @@ public class LocalMessageServiceTest extends InitConfigAndLoggerTest { return null; }); - CompletableFuture future = localMessageService.sendMessage(proxyContext, null, messagesList, requestHeader, 1000L); - SendResult sendResult = future.get(); + CompletableFuture> future = localMessageService.sendMessage(proxyContext, null, messagesList, requestHeader, 1000L); + SendResult sendResult = future.get().get(0); assertThat(sendResult.getSendStatus()).isEqualTo(SendStatus.SEND_OK); assertThat(sendResult.getMessageQueue()) .isEqualTo(new MessageQueue(topic, brokerControllerMock.getBrokerConfig().getBrokerName(), queueId)); @@ -224,7 +224,7 @@ public class LocalMessageServiceTest extends InitConfigAndLoggerTest { Mockito.when(sendMessageProcessorMock.processRequest(Mockito.any(SimpleChannelHandlerContext.class), Mockito.any(RemotingCommand.class))) .thenReturn(response); - CompletableFuture future = localMessageService.sendMessage(proxyContext, null, messagesList, sendMessageRequestHeader, 1000L); + CompletableFuture> future = localMessageService.sendMessage(proxyContext, null, messagesList, sendMessageRequestHeader, 1000L); ExecutionException exception = catchThrowableOfType(future::get, ExecutionException.class); assertThat(exception.getCause()).isInstanceOf(ProxyException.class); assertThat(((ProxyException) exception.getCause()).getCode()).isEqualTo(ProxyExceptionCode.ILLEGAL_MESSAGE); @@ -238,7 +238,7 @@ public class LocalMessageServiceTest extends InitConfigAndLoggerTest { MessageClientIDSetter.setUniqID(message); List messagesList = Collections.singletonList(message); SendMessageRequestHeader sendMessageRequestHeader = new SendMessageRequestHeader(); - CompletableFuture future = localMessageService.sendMessage(proxyContext, null, messagesList, sendMessageRequestHeader, 1000L); + CompletableFuture> future = localMessageService.sendMessage(proxyContext, null, messagesList, sendMessageRequestHeader, 1000L); ExecutionException exception = catchThrowableOfType(future::get, ExecutionException.class); assertThat(exception.getCause()).isInstanceOf(RemotingCommandException.class); }