for #3666 add unit test more

Author: tianliuliu

broker/src/test/java/org/apache/rocketmq/broker/BrokerPathConfigHelperTest.java

@@ -0,0 +1,42 @@
+/*
+ * 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.broker;
+
+import org.junit.Test;
+
+import static org.junit.Assert.*;
+
+public class BrokerPathConfigHelperTest {
+
+    @Test
+    public void testGetLmqConsumerOffsetPath() {
+        String lmqConsumerOffsetPath = BrokerPathConfigHelper.getLmqConsumerOffsetPath("/home/admin/store");
+        assertEquals("/home/admin/store/config/lmqConsumerOffset.json", lmqConsumerOffsetPath);
+
+
+        String consumerOffsetPath = BrokerPathConfigHelper.getConsumerOffsetPath("/home/admin/store");
+        assertEquals("/home/admin/store/config/consumerOffset.json", consumerOffsetPath);
+
+        String topicConfigPath = BrokerPathConfigHelper.getTopicConfigPath("/home/admin/store");
+        assertEquals("/home/admin/store/config/topics.json", topicConfigPath);
+
+        String subscriptionGroupPath = BrokerPathConfigHelper.getSubscriptionGroupPath("/home/admin/store");
+        assertEquals("/home/admin/store/config/subscriptionGroup.json", subscriptionGroupPath);
+
+    }
+}

store/src/test/java/org/apache/rocketmq/store/ConsumeQueueTest.java

@@ -27,6 +27,7 @@
 import java.util.Map;
 import org.apache.rocketmq.common.BrokerConfig;
 import org.apache.rocketmq.common.UtilAll;
+import org.apache.rocketmq.common.message.MessageConst;
 import org.apache.rocketmq.common.message.MessageDecoder;
 import org.apache.rocketmq.store.config.MessageStoreConfig;
 import org.apache.rocketmq.store.stats.BrokerStatsManager;
@@ -146,6 +147,34 @@ public void arriving(String topic, int queueId, long logicOffset, long tagsCode,
         return master;
     }
 
+    protected DefaultMessageStore genForMultiQueue() throws Exception {
+        MessageStoreConfig messageStoreConfig = buildStoreConfig(
+            commitLogFileSize, cqFileSize, true, cqExtFileSize
+        );
+
+        messageStoreConfig.setEnableLmq(true);
+        messageStoreConfig.setEnableMultiDispatch(true);
+
+        BrokerConfig brokerConfig = new BrokerConfig();
+
+        DefaultMessageStore master = new DefaultMessageStore(
+            messageStoreConfig,
+            new BrokerStatsManager(brokerConfig.getBrokerClusterName(), brokerConfig.isEnableDetailStat()),
+            new MessageArrivingListener() {
+                @Override
+                public void arriving(String topic, int queueId, long logicOffset, long tagsCode,
+                    long msgStoreTime, byte[] filterBitMap, Map<String, String> properties) {
+                }
+            }
+            , brokerConfig);
+
+        assertThat(master.load()).isTrue();
+
+        master.start();
+
+        return master;
+    }
+
     protected void putMsg(DefaultMessageStore master) throws Exception {
         long totalMsgs = 200;
 
@@ -158,6 +187,33 @@ protected void putMsg(DefaultMessageStore master) throws Exception {
         }
     }
 
+    protected void putMsgMultiQueue(DefaultMessageStore master) throws Exception {
+        for (long i = 0; i < 1; i++) {
+            master.putMessage(buildMessageMultiQueue());
+        }
+    }
+
+    private MessageExtBrokerInner buildMessageMultiQueue() {
+        MessageExtBrokerInner msg = new MessageExtBrokerInner();
+        msg.setTopic(topic);
+        msg.setTags("TAG1");
+        msg.setKeys("Hello");
+        msg.setBody(msgBody);
+        msg.setKeys(String.valueOf(System.currentTimeMillis()));
+        msg.setQueueId(queueId);
+        msg.setSysFlag(0);
+        msg.setBornTimestamp(System.currentTimeMillis());
+        msg.setStoreHost(StoreHost);
+        msg.setBornHost(BornHost);
+        for (int i = 0; i < 1; i++) {
+            msg.putUserProperty(MessageConst.PROPERTY_INNER_MULTI_DISPATCH, "%LMQ%123,%LMQ%456");
+            msg.putUserProperty(String.valueOf(i), "imagoodperson" + i);
+        }
+        msg.setPropertiesString(MessageDecoder.messageProperties2String(msg.getProperties()));
+
+        return msg;
+    }
+
     protected void deleteDirectory(String rootPath) {
         File file = new File(rootPath);
         deleteFile(file);
@@ -217,6 +273,41 @@ public void testPutMessagePositionInfo_buildCQRepeatedly() throws Exception {
 
     }
 
+    @Test
+    public void testPutMessagePositionInfoMultiQueue() throws Exception {
+        DefaultMessageStore messageStore = null;
+        try {
+
+            messageStore = genForMultiQueue();
+
+            int totalMessages = 10;
+
+            for (int i = 0; i < totalMessages; i++) {
+                putMsgMultiQueue(messageStore);
+            }
+            Thread.sleep(5);
+
+            ConsumeQueue cq = messageStore.getConsumeQueueTable().get(topic).get(queueId);
+
+            ConsumeQueue lmqCq1 = messageStore.getConsumeQueueTable().get("%LMQ%123").get(0);
+
+            ConsumeQueue lmqCq2 = messageStore.getConsumeQueueTable().get("%LMQ%456").get(0);
+
+            assertThat(cq).isNotNull();
+
+            assertThat(lmqCq1).isNotNull();
+
+            assertThat(lmqCq2).isNotNull();
+
+        } finally {
+            if (messageStore != null) {
+                messageStore.shutdown();
+                messageStore.destroy();
+            }
+            deleteDirectory(storePath);
+        }
+    }
+
     @Test
     public void testConsumeQueueWithExtendData() {
         DefaultMessageStore master = null;