You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@rocketmq.apache.org by yu...@apache.org on 2017/01/17 13:15:02 UTC
[3/6] incubator-rocketmq git commit: [ROCKETMQ-52] Add unit tests for
LocalFileOffsetStore
[ROCKETMQ-52] Add unit tests for LocalFileOffsetStore
Project: http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/repo
Commit: http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/commit/6b5ab40c
Tree: http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/tree/6b5ab40c
Diff: http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/diff/6b5ab40c
Branch: refs/heads/ROCKETMQ-52
Commit: 6b5ab40ced9f3f40f0acc18ba4eaf64b76686701
Parents: c0e7629
Author: yukon <yu...@apache.org>
Authored: Tue Jan 17 15:15:16 2017 +0800
Committer: yukon <yu...@apache.org>
Committed: Tue Jan 17 15:15:16 2017 +0800
----------------------------------------------------------------------
.../store/LocalFileOffsetStoreTest.java | 74 ++++++++++++++++++++
1 file changed, 74 insertions(+)
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/incubator-rocketmq/blob/6b5ab40c/client/src/test/java/org/apache/rocketmq/client/consumer/store/LocalFileOffsetStoreTest.java
----------------------------------------------------------------------
diff --git a/client/src/test/java/org/apache/rocketmq/client/consumer/store/LocalFileOffsetStoreTest.java b/client/src/test/java/org/apache/rocketmq/client/consumer/store/LocalFileOffsetStoreTest.java
new file mode 100644
index 0000000..58d99d6
--- /dev/null
+++ b/client/src/test/java/org/apache/rocketmq/client/consumer/store/LocalFileOffsetStoreTest.java
@@ -0,0 +1,74 @@
+/*
+ * 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.store;
+
+import java.util.Collections;
+import java.util.HashSet;
+import org.apache.rocketmq.client.ClientConfig;
+import org.apache.rocketmq.client.impl.factory.MQClientInstance;
+import org.apache.rocketmq.common.message.MessageQueue;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.mockito.Mockito.when;
+
+public class LocalFileOffsetStoreTest {
+
+ @Mock
+ private static MQClientInstance mQClientFactory;
+ private static String group = "FooBarGroup";
+ private static String topic = "FooBar";
+ private static String brokerName = "DefaultBrokerName";
+
+ @BeforeClass
+ public static void init() {
+ System.setProperty("rocketmq.client.localOffsetStoreDir", System.getProperty("java.io.tmpdir") + ".rocketmq_offsets");
+ mQClientFactory = Mockito.mock(MQClientInstance.class);
+ String clientId = new ClientConfig().buildMQClientId() + "#TestNamespace" + System.currentTimeMillis();
+ when(mQClientFactory.getClientId()).thenReturn(clientId);
+ }
+
+ @Test
+ public void testUpdateOffset() throws Exception {
+ OffsetStore offsetStore = new LocalFileOffsetStore(mQClientFactory, group);
+ MessageQueue messageQueue = new MessageQueue(topic, brokerName, 1);
+ offsetStore.updateOffset(messageQueue, 1024, false);
+
+ assertThat(offsetStore.readOffset(messageQueue, ReadOffsetType.READ_FROM_MEMORY)).isEqualTo(1024);
+
+ offsetStore.updateOffset(messageQueue, 1023, false);
+ assertThat(offsetStore.readOffset(messageQueue, ReadOffsetType.READ_FROM_MEMORY)).isEqualTo(1023);
+
+ offsetStore.updateOffset(messageQueue, 1022, true);
+ assertThat(offsetStore.readOffset(messageQueue, ReadOffsetType.READ_FROM_MEMORY)).isEqualTo(1023);
+ }
+
+ @Test
+ public void testReadOffset_FromStore() throws Exception {
+ OffsetStore offsetStore = new LocalFileOffsetStore(mQClientFactory, group);
+ MessageQueue messageQueue = new MessageQueue(topic, brokerName, 2);
+
+ offsetStore.updateOffset(messageQueue, 1024, false);
+ assertThat(offsetStore.readOffset(messageQueue, ReadOffsetType.READ_FROM_STORE)).isEqualTo(-1);
+
+ offsetStore.persistAll(new HashSet<>(Collections.singletonList(messageQueue)));
+ assertThat(offsetStore.readOffset(messageQueue, ReadOffsetType.READ_FROM_STORE)).isEqualTo(1024);
+ }
+}
\ No newline at end of file