[rocketmq] branch develop updated: [ISSUE #3674]Add unit test for AllocateMessageQueueAveragelyByCircle (#3839)

[du...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

duhengforever 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 8b78f61  [ISSUE #3674]Add unit test for AllocateMessageQueueAveragelyByCircle (#3839)
8b78f61 is described below

commit 8b78f6114f92cc3e482eab602e4b5cdff9d79f0c
Author: zhangjidi2016 <10...@qq.com>
AuthorDate: Fri Feb 11 11:30:07 2022 +0800

    [ISSUE #3674]Add unit test for AllocateMessageQueueAveragelyByCircle (#3839)
    
    Co-authored-by: zhangjidi <zh...@cmss.chinamobile.com>
---
 .../AllocateMessageQueueAveragelyByCircleTest.java | 68 ++++++++++++++++++++++
 1 file changed, 68 insertions(+)

diff --git a/client/src/test/java/org/apache/rocketmq/client/consumer/rebalance/AllocateMessageQueueAveragelyByCircleTest.java b/client/src/test/java/org/apache/rocketmq/client/consumer/rebalance/AllocateMessageQueueAveragelyByCircleTest.java
new file mode 100644
index 0000000..497766d
--- /dev/null
+++ b/client/src/test/java/org/apache/rocketmq/client/consumer/rebalance/AllocateMessageQueueAveragelyByCircleTest.java
@@ -0,0 +1,68 @@
+/*
+ * 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.client.consumer.rebalance;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import junit.framework.TestCase;
+import org.apache.rocketmq.common.message.MessageQueue;
+import org.junit.Assert;
+
+public class AllocateMessageQueueAveragelyByCircleTest extends TestCase {
+
+    public void testAllocateMessageQueueAveragelyByCircle() {
+        List<String> consumerIdList = createConsumerIdList(4);
+        List<MessageQueue> messageQueueList = createMessageQueueList(10);
+        // the consumerId not in cidAll
+        List<MessageQueue> allocateQueues = new AllocateMessageQueueAveragelyByCircle().allocate("", "CID_PREFIX", messageQueueList, consumerIdList);
+        Assert.assertEquals(0, allocateQueues.size());
+
+        Map<String, int[]> consumerAllocateQueue = new HashMap<String, int[]>(consumerIdList.size());
+        for (String consumerId : consumerIdList) {
+            List<MessageQueue> queues = new AllocateMessageQueueAveragelyByCircle().allocate("", consumerId, messageQueueList, consumerIdList);
+            int[] queueIds = new int[queues.size()];
+            for (int i = 0; i < queues.size(); i++) {
+                queueIds[i] = queues.get(i).getQueueId();
+            }
+            consumerAllocateQueue.put(consumerId, queueIds);
+        }
+        Assert.assertArrayEquals(new int[] {0, 4, 8}, consumerAllocateQueue.get("CID_PREFIX0"));
+        Assert.assertArrayEquals(new int[] {1, 5, 9}, consumerAllocateQueue.get("CID_PREFIX1"));
+        Assert.assertArrayEquals(new int[] {2, 6}, consumerAllocateQueue.get("CID_PREFIX2"));
+        Assert.assertArrayEquals(new int[] {3, 7}, consumerAllocateQueue.get("CID_PREFIX3"));
+    }
+
+    private List<String> createConsumerIdList(int size) {
+        List<String> consumerIdList = new ArrayList<String>(size);
+        for (int i = 0; i < size; i++) {
+            consumerIdList.add("CID_PREFIX" + i);
+        }
+        return consumerIdList;
+    }
+
+    private List<MessageQueue> createMessageQueueList(int size) {
+        List<MessageQueue> messageQueueList = new ArrayList<MessageQueue>(size);
+        for (int i = 0; i < size; i++) {
+            MessageQueue mq = new MessageQueue("topic", "brokerName", i);
+            messageQueueList.add(mq);
+        }
+        return messageQueueList;
+    }
+}
+