From c99510f57550d8fdbf36de691074187da0e8e153 Mon Sep 17 00:00:00 2001
From: yukon <yukon@apache.org>
Date: Tue, 17 Jan 2017 13:28:56 +0800
Subject: [PATCH] [ROCKETMQ-52] Add unit tests for MessageQueueSelector

---
 .../client/common/ThreadLocalIndexTest.java   |  2 +-
 .../SelectMessageQueueByHashTest.java         | 49 +++++++++++++++++++
 2 files changed, 50 insertions(+), 1 deletion(-)
 create mode 100644 client/src/test/java/org/apache/rocketmq/client/producer/selector/SelectMessageQueueByHashTest.java

diff --git a/client/src/test/java/org/apache/rocketmq/client/common/ThreadLocalIndexTest.java b/client/src/test/java/org/apache/rocketmq/client/common/ThreadLocalIndexTest.java
index b937e457..350cde76 100644
--- a/client/src/test/java/org/apache/rocketmq/client/common/ThreadLocalIndexTest.java
+++ b/client/src/test/java/org/apache/rocketmq/client/common/ThreadLocalIndexTest.java
@@ -23,7 +23,7 @@ import static org.assertj.core.api.Assertions.assertThat;
 public class ThreadLocalIndexTest {
 
     @Test
-    public void getAndIncrement() throws Exception {
+    public void testGetAndIncrement() throws Exception {
         ThreadLocalIndex localIndex = new ThreadLocalIndex();
         int initialVal = localIndex.getAndIncrement();
 
diff --git a/client/src/test/java/org/apache/rocketmq/client/producer/selector/SelectMessageQueueByHashTest.java b/client/src/test/java/org/apache/rocketmq/client/producer/selector/SelectMessageQueueByHashTest.java
new file mode 100644
index 00000000..64ef21de
--- /dev/null
+++ b/client/src/test/java/org/apache/rocketmq/client/producer/selector/SelectMessageQueueByHashTest.java
@@ -0,0 +1,49 @@
+/*
+ * 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.producer.selector;
+
+import java.util.ArrayList;
+import java.util.List;
+import org.apache.rocketmq.common.message.Message;
+import org.apache.rocketmq.common.message.MessageQueue;
+import org.junit.Test;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class SelectMessageQueueByHashTest {
+
+    private String topic = "FooBar";
+
+    @Test
+    public void testSelect() throws Exception {
+        SelectMessageQueueByHash selector = new SelectMessageQueueByHash();
+
+        Message message = new Message(topic, new byte[] {});
+
+        List<MessageQueue> messageQueues = new ArrayList<>();
+        for (int i = 0; i < 10; i++) {
+            MessageQueue messageQueue = new MessageQueue(topic, "DefaultBroker", i);
+            messageQueues.add(messageQueue);
+        }
+
+        String orderId = "123";
+        String anotherOrderId = "234";
+        MessageQueue selected = selector.select(messageQueues, message, orderId);
+        assertThat(selector.select(messageQueues, message, anotherOrderId)).isNotEqualTo(selected);
+    }
+
+}
\ No newline at end of file
-- 
GitLab