opensource/rocketmq
3
0
Fork 0
mirror of https://github.com/apache/rocketmq.git synced 2026-09-21 05:44:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ISSUE #3949] Update LocalMessageService unit test

Browse Source
This commit is contained in:
zhouxiang
2022-07-13 11:29:38 +08:00
parent 41e5cdafe8
commit dfacfe607f
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
proxy/src/test/java/org/apache/rocketmq/proxy/service/message/LocalMessageServiceTest.java
+6 -6
View File
@@ -155,8 +155,8 @@ public class LocalMessageServiceTest extends InitConfigAndLoggerTest {
return null;
});
CompletableFuture<SendResult> future = localMessageService.sendMessage(proxyContext, null, messagesList, requestHeader, 1000L);
SendResult sendResult = future.get();
CompletableFuture<List<SendResult>> 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<SendResult> future = localMessageService.sendMessage(proxyContext, null, messagesList, requestHeader, 1000L);
SendResult sendResult = future.get();
CompletableFuture<List<SendResult>> 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<SendResult> future = localMessageService.sendMessage(proxyContext, null, messagesList, sendMessageRequestHeader, 1000L);
CompletableFuture<List<SendResult>> 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<Message> messagesList = Collections.singletonList(message);
SendMessageRequestHeader sendMessageRequestHeader = new SendMessageRequestHeader();
CompletableFuture<SendResult> future = localMessageService.sendMessage(proxyContext, null, messagesList, sendMessageRequestHeader, 1000L);
CompletableFuture<List<SendResult>> future = localMessageService.sendMessage(proxyContext, null, messagesList, sendMessageRequestHeader, 1000L);
ExecutionException exception = catchThrowableOfType(future::get, ExecutionException.class);
assertThat(exception.getCause()).isInstanceOf(RemotingCommandException.class);
}