You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@rocketmq.apache.org by di...@apache.org on 2019/05/14 08:51:20 UTC
[rocketmq] branch develop updated: [RIP-10] Add test cases for
QueryMsgByUniqueKeySubCommand (#945)
This is an automated email from the ASF dual-hosted git repository.
dinglei pushed a commit to branch develop
in repository https://gitbox.apache.org/repos/asf/rocketmq.git
The following commit(s) were added to refs/heads/develop by this push:
new 26878e9 [RIP-10] Add test cases for QueryMsgByUniqueKeySubCommand (#945)
26878e9 is described below
commit 26878e9c79368503eebac58484f7240c3e57f330
Author: 蔡盛智 <36...@qq.com>
AuthorDate: Tue May 14 16:51:13 2019 +0800
[RIP-10] Add test cases for QueryMsgByUniqueKeySubCommand (#945)
---
.../message/QueryMsgByUniqueKeySubCommand.java | 23 ++-
.../message/QueryMsgByUniqueKeySubCommandTest.java | 230 +++++++++++++++++++++
2 files changed, 250 insertions(+), 3 deletions(-)
diff --git a/tools/src/main/java/org/apache/rocketmq/tools/command/message/QueryMsgByUniqueKeySubCommand.java b/tools/src/main/java/org/apache/rocketmq/tools/command/message/QueryMsgByUniqueKeySubCommand.java
index 0103b50..9ad0750 100644
--- a/tools/src/main/java/org/apache/rocketmq/tools/command/message/QueryMsgByUniqueKeySubCommand.java
+++ b/tools/src/main/java/org/apache/rocketmq/tools/command/message/QueryMsgByUniqueKeySubCommand.java
@@ -39,6 +39,24 @@ import org.apache.rocketmq.tools.command.SubCommandException;
public class QueryMsgByUniqueKeySubCommand implements SubCommand {
+ private DefaultMQAdminExt defaultMQAdminExt;
+
+ private DefaultMQAdminExt createMQAdminExt(RPCHook rpcHook) throws SubCommandException {
+ if (this.defaultMQAdminExt != null) {
+ return defaultMQAdminExt;
+ } else {
+ defaultMQAdminExt = new DefaultMQAdminExt(rpcHook);
+ defaultMQAdminExt.setInstanceName(Long.toString(System.currentTimeMillis()));
+ try {
+ defaultMQAdminExt.start();
+ }
+ catch (Exception e) {
+ throw new SubCommandException(this.getClass().getSimpleName() + " command failed", e);
+ }
+ return defaultMQAdminExt;
+ }
+ }
+
public static void queryById(final DefaultMQAdminExt admin, final String topic,
final String msgId) throws MQClientException,
RemotingException, MQBrokerException, InterruptedException, IOException {
@@ -182,11 +200,10 @@ public class QueryMsgByUniqueKeySubCommand implements SubCommand {
@Override
public void execute(CommandLine commandLine, Options options, RPCHook rpcHook) throws SubCommandException {
- DefaultMQAdminExt defaultMQAdminExt = new DefaultMQAdminExt(rpcHook);
- defaultMQAdminExt.setInstanceName(Long.toString(System.currentTimeMillis()));
try {
- defaultMQAdminExt.start();
+
+ defaultMQAdminExt = createMQAdminExt(rpcHook);
final String msgId = commandLine.getOptionValue('i').trim();
final String topic = commandLine.getOptionValue('t').trim();
diff --git a/tools/src/test/java/org/apache/rocketmq/tools/command/message/QueryMsgByUniqueKeySubCommandTest.java b/tools/src/test/java/org/apache/rocketmq/tools/command/message/QueryMsgByUniqueKeySubCommandTest.java
new file mode 100644
index 0000000..be6b636
--- /dev/null
+++ b/tools/src/test/java/org/apache/rocketmq/tools/command/message/QueryMsgByUniqueKeySubCommandTest.java
@@ -0,0 +1,230 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.rocketmq.tools.command.message;
+
+import org.apache.commons.cli.CommandLine;
+import org.apache.commons.cli.Options;
+import org.apache.commons.cli.PosixParser;
+import org.apache.rocketmq.client.ClientConfig;
+import org.apache.rocketmq.client.exception.MQBrokerException;
+import org.apache.rocketmq.client.exception.MQClientException;
+import org.apache.rocketmq.client.impl.MQAdminImpl;
+import org.apache.rocketmq.client.impl.MQClientAPIImpl;
+import org.apache.rocketmq.client.impl.MQClientManager;
+import org.apache.rocketmq.client.impl.factory.MQClientInstance;
+import org.apache.rocketmq.common.MixAll;
+import org.apache.rocketmq.common.admin.ConsumeStats;
+import org.apache.rocketmq.common.admin.OffsetWrapper;
+import org.apache.rocketmq.common.message.MessageExt;
+import org.apache.rocketmq.common.message.MessageQueue;
+import org.apache.rocketmq.common.protocol.body.*;
+import org.apache.rocketmq.common.protocol.heartbeat.ConsumeType;
+import org.apache.rocketmq.common.protocol.route.BrokerData;
+import org.apache.rocketmq.common.protocol.route.TopicRouteData;
+import org.apache.rocketmq.remoting.exception.RemotingConnectException;
+import org.apache.rocketmq.remoting.exception.RemotingException;
+import org.apache.rocketmq.remoting.exception.RemotingSendRequestException;
+import org.apache.rocketmq.remoting.exception.RemotingTimeoutException;
+import org.apache.rocketmq.remoting.protocol.LanguageCode;
+import org.apache.rocketmq.srvutil.ServerUtil;
+import org.apache.rocketmq.tools.admin.DefaultMQAdminExt;
+import org.apache.rocketmq.tools.admin.DefaultMQAdminExtImpl;
+import org.apache.rocketmq.tools.command.SubCommandException;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.lang.reflect.Field;
+import java.net.InetSocketAddress;
+import java.util.*;
+
+import static org.mockito.ArgumentMatchers.*;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+public class QueryMsgByUniqueKeySubCommandTest {
+
+ private static QueryMsgByUniqueKeySubCommand cmd = new QueryMsgByUniqueKeySubCommand();
+
+ private static DefaultMQAdminExt defaultMQAdminExt;
+ private static DefaultMQAdminExtImpl defaultMQAdminExtImpl;
+ private static MQClientInstance mqClientInstance = MQClientManager.getInstance().getAndCreateMQClientInstance(new ClientConfig());
+
+ private static MQClientAPIImpl mQClientAPIImpl;
+ private static MQAdminImpl mQAdminImpl;
+
+ @Before
+ public void before() throws NoSuchFieldException, IllegalAccessException, InterruptedException, RemotingException, MQClientException, MQBrokerException {
+
+ mQClientAPIImpl = mock(MQClientAPIImpl.class);
+ mQAdminImpl = mock(MQAdminImpl.class);
+
+ defaultMQAdminExt = new DefaultMQAdminExt();
+ defaultMQAdminExtImpl = new DefaultMQAdminExtImpl(defaultMQAdminExt, 1000);
+
+ Field field = DefaultMQAdminExtImpl.class.getDeclaredField("mqClientInstance");
+ field.setAccessible(true);
+ field.set(defaultMQAdminExtImpl, mqClientInstance);
+
+ field = MQClientInstance.class.getDeclaredField("mQClientAPIImpl");
+ field.setAccessible(true);
+ field.set(mqClientInstance, mQClientAPIImpl);
+
+ field = MQClientInstance.class.getDeclaredField("mQAdminImpl");
+ field.setAccessible(true);
+ field.set(mqClientInstance, mQAdminImpl);
+
+
+ field = DefaultMQAdminExt.class.getDeclaredField("defaultMQAdminExtImpl");
+ field.setAccessible(true);
+ field.set(defaultMQAdminExt, defaultMQAdminExtImpl);
+
+ ConsumeMessageDirectlyResult result = new ConsumeMessageDirectlyResult();
+ result.setConsumeResult(CMResult.CR_SUCCESS);
+ result.setRemark("customRemark_122333444");
+ when(mQClientAPIImpl.consumeMessageDirectly(anyString(), anyString(), anyString(), anyString(), anyLong())).thenReturn(result);
+
+ MessageExt retMsgExt = new MessageExt();
+ retMsgExt.setMsgId("0A3A54F7BF7D18B4AAC28A3FA2CF0000");
+ retMsgExt.setBody("this is message ext body".getBytes());
+ retMsgExt.setTopic("testTopic");
+ retMsgExt.setTags("testTags");
+ retMsgExt.setStoreHost(new InetSocketAddress("127.0.0.1", 8899));
+ retMsgExt.setBornHost(new InetSocketAddress("127.0.0.1", 7788));
+ retMsgExt.setQueueId(1);
+ retMsgExt.setQueueOffset(12L);
+ retMsgExt.setCommitLogOffset(123);
+ retMsgExt.setReconsumeTimes(2);
+ retMsgExt.setBornTimestamp(System.currentTimeMillis());
+ retMsgExt.setStoreTimestamp(System.currentTimeMillis());
+ when(mQAdminImpl.viewMessage(anyString())).thenReturn(retMsgExt);
+
+ when(mQAdminImpl.queryMessageByUniqKey(anyString(), anyString())).thenReturn(retMsgExt);
+
+ TopicRouteData topicRouteData = new TopicRouteData();
+ List<BrokerData> brokerDataList = new ArrayList<BrokerData>();
+ BrokerData brokerData = new BrokerData();
+ HashMap<Long, String> brokerAddrs = new HashMap<Long, String>();
+ brokerAddrs.put(MixAll.MASTER_ID, "127.0.0.1:9876");
+ brokerData.setBrokerAddrs(brokerAddrs);
+ brokerDataList.add(brokerData);
+ topicRouteData.setBrokerDatas(brokerDataList);
+ when(mQClientAPIImpl.getTopicRouteInfoFromNameServer(anyString(), anyLong())).thenReturn(topicRouteData);
+
+ GroupList groupList = new GroupList();
+ HashSet<String> groupSets = new HashSet<String>();
+ groupSets.add("testGroup");
+ groupList.setGroupList(groupSets);
+ when(mQClientAPIImpl.queryTopicConsumeByWho(anyString(), anyString(), anyLong())).thenReturn(groupList);
+
+
+ ConsumeStats consumeStats = new ConsumeStats();
+ consumeStats.setConsumeTps(100*10000);
+ HashMap<MessageQueue, OffsetWrapper> offsetTable = new HashMap<MessageQueue, OffsetWrapper>();
+ MessageQueue messageQueue = new MessageQueue();
+ messageQueue.setBrokerName("messageQueue BrokerName testing");
+ messageQueue.setTopic("messageQueue topic");
+ messageQueue.setQueueId(1);
+ OffsetWrapper offsetWrapper = new OffsetWrapper();
+ offsetWrapper.setBrokerOffset(100);
+ offsetWrapper.setConsumerOffset(200);
+ offsetWrapper.setLastTimestamp(System.currentTimeMillis());
+ offsetTable.put(messageQueue, offsetWrapper);
+ consumeStats.setOffsetTable(offsetTable);
+ when(mQClientAPIImpl.getConsumeStats(anyString(), anyString(), (String)isNull(), anyLong())).thenReturn(consumeStats);
+
+ ClusterInfo clusterInfo = new ClusterInfo();
+ HashMap<String, BrokerData> brokerAddrTable = new HashMap<String, BrokerData>();
+ brokerAddrTable.put("key", brokerData);
+ clusterInfo.setBrokerAddrTable(brokerAddrTable);
+ HashMap<String, Set<String>> clusterAddrTable = new HashMap<String, Set<String>>();
+ Set<String> addrSet = new HashSet<String>();
+ addrSet.add("127.0.0.1:9876");
+ clusterAddrTable.put("key", addrSet);
+ clusterInfo.setClusterAddrTable(clusterAddrTable);
+ when(mQClientAPIImpl.getBrokerClusterInfo(anyLong())).thenReturn(clusterInfo);
+
+ field = QueryMsgByUniqueKeySubCommand.class.getDeclaredField("defaultMQAdminExt");
+ field.setAccessible(true);
+ field.set(cmd, defaultMQAdminExt);
+
+ }
+
+ @Test
+ public void testExecuteConsumeActively() throws SubCommandException, InterruptedException, MQBrokerException, RemotingTimeoutException, RemotingSendRequestException, RemotingConnectException {
+
+ ConsumerConnection consumerConnection = new ConsumerConnection();
+ consumerConnection.setConsumeType(ConsumeType.CONSUME_ACTIVELY);
+ HashSet<Connection> connectionSet = new HashSet<>();
+ Connection conn = new Connection();
+ conn.setClientId("clientIdTest");
+ conn.setClientAddr("clientAddrTest");
+ conn.setLanguage(LanguageCode.JAVA);
+ conn.setVersion(1);
+ connectionSet.add(conn);
+ consumerConnection.setConnectionSet(connectionSet);
+ when(mQClientAPIImpl.getConsumerConnectionList(anyString(), anyString(), anyLong())).thenReturn(consumerConnection);
+
+ Options options = ServerUtil.buildCommandlineOptions(new Options());
+
+ String[] args = new String[]{"-t myTopicTest", "-i msgId"};
+ CommandLine commandLine = ServerUtil.parseCmdLine("mqadmin ", args, cmd.buildCommandlineOptions(options), new PosixParser());
+ cmd.execute(commandLine, options, null);
+
+ }
+
+ @Test
+ public void testExecuteConsumePassively() throws SubCommandException, InterruptedException, MQBrokerException, RemotingTimeoutException, RemotingSendRequestException, RemotingConnectException {
+
+ ConsumerConnection consumerConnection = new ConsumerConnection();
+ consumerConnection.setConsumeType(ConsumeType.CONSUME_PASSIVELY);
+ HashSet<Connection> connectionSet = new HashSet<>();
+ Connection conn = new Connection();
+ conn.setClientId("clientIdTestStr");
+ conn.setClientAddr("clientAddrTestStr");
+ conn.setLanguage(LanguageCode.JAVA);
+ conn.setVersion(2);
+ connectionSet.add(conn);
+ consumerConnection.setConnectionSet(connectionSet);
+ when(mQClientAPIImpl.getConsumerConnectionList(anyString(), anyString(), anyLong())).thenReturn(consumerConnection);
+
+ Options options = ServerUtil.buildCommandlineOptions(new Options());
+
+ String[] args = new String[]{"-t myTopicTest", "-i 7F000001000004D20000000000000066"};
+ CommandLine commandLine = ServerUtil.parseCmdLine("mqadmin ", args, cmd.buildCommandlineOptions(options), new PosixParser());
+ cmd.execute(commandLine, options, null);
+
+
+
+ }
+
+ @Test
+ public void testExecuteWithConsumerGroupAndClientId() throws SubCommandException {
+
+ Options options = ServerUtil.buildCommandlineOptions(new Options());
+
+ String[] args = new String[]{"-t myTopicTest", "-i 0A3A54F7BF7D18B4AAC28A3FA2CF0000", "-g producerGroupName", "-d clientId"};
+ CommandLine commandLine = ServerUtil.parseCmdLine("mqadmin ", args, cmd.buildCommandlineOptions(options), new PosixParser());
+ cmd.execute(commandLine, options, null);
+
+ System.out.println();
+ System.out.println("commandName=" + cmd.commandName());
+ System.out.println("commandDesc=" + cmd.commandDesc());
+
+ }
+
+
+}