Add test case for LocalFileOffsetStore closes apache#59

NeoTalk · May 27, 2017 · 0adad6f · 0adad6f
1 parent e5d01b4
commit 0adad6f
Showing 1 changed file with 13 additions and 0 deletions.
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
@@ -19,6 +19,8 @@
 import java.io.File;
 import java.util.Collections;
 import java.util.HashSet;
+import java.util.Map;
+
 import org.apache.rocketmq.client.ClientConfig;
 import org.apache.rocketmq.client.impl.factory.MQClientInstance;
 import org.apache.rocketmq.common.message.MessageQueue;
@@ -72,4 +74,15 @@ public void testReadOffset_FromStore() throws Exception {
         offsetStore.persistAll(new HashSet<MessageQueue>(Collections.singletonList(messageQueue)));
         assertThat(offsetStore.readOffset(messageQueue, ReadOffsetType.READ_FROM_STORE)).isEqualTo(1024);
     }
+
+    @Test
+    public void testCloneOffset() throws Exception {
+        OffsetStore offsetStore = new LocalFileOffsetStore(mQClientFactory, group);
+        MessageQueue messageQueue = new MessageQueue(topic, brokerName, 3);
+        offsetStore.updateOffset(messageQueue, 1024, false);
+        Map<MessageQueue, Long> cloneOffsetTable = offsetStore.cloneOffsetTable(topic);
+
+        assertThat(cloneOffsetTable.size()).isEqualTo(1);
+        assertThat(cloneOffsetTable.get(messageQueue)).isEqualTo(1024);
+    }
 }