提交 c99510f5 编写于 作者: Y yukon

[ROCKETMQ-52] Add unit tests for MessageQueueSelector

上级 edde487e
...@@ -23,7 +23,7 @@ import static org.assertj.core.api.Assertions.assertThat; ...@@ -23,7 +23,7 @@ import static org.assertj.core.api.Assertions.assertThat;
public class ThreadLocalIndexTest { public class ThreadLocalIndexTest {
@Test @Test
public void getAndIncrement() throws Exception { public void testGetAndIncrement() throws Exception {
ThreadLocalIndex localIndex = new ThreadLocalIndex(); ThreadLocalIndex localIndex = new ThreadLocalIndex();
int initialVal = localIndex.getAndIncrement(); int initialVal = localIndex.getAndIncrement();
......
/*
* 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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册