| // Copyright (c) 2011-present, Facebook, Inc. All rights reserved. |
| // This source code is licensed under both the GPLv2 (found in the |
| // COPYING file in the root directory) and Apache 2.0 License |
| // (found in the LICENSE.Apache file in the root directory). |
| package org.rocksdb; |
| |
| import org.junit.ClassRule; |
| import org.junit.Rule; |
| import org.junit.Test; |
| import org.junit.rules.TemporaryFolder; |
| |
| import static org.assertj.core.api.Assertions.assertThat; |
| |
| public class RocksIteratorTest { |
| |
| @ClassRule |
| public static final RocksMemoryResource rocksMemoryResource = |
| new RocksMemoryResource(); |
| |
| @Rule |
| public TemporaryFolder dbFolder = new TemporaryFolder(); |
| |
| @Test |
| public void rocksIterator() throws RocksDBException { |
| try (final Options options = new Options() |
| .setCreateIfMissing(true) |
| .setCreateMissingColumnFamilies(true); |
| final RocksDB db = RocksDB.open(options, |
| dbFolder.getRoot().getAbsolutePath())) { |
| db.put("key1".getBytes(), "value1".getBytes()); |
| db.put("key2".getBytes(), "value2".getBytes()); |
| |
| try (final RocksIterator iterator = db.newIterator()) { |
| iterator.seekToFirst(); |
| assertThat(iterator.isValid()).isTrue(); |
| assertThat(iterator.key()).isEqualTo("key1".getBytes()); |
| assertThat(iterator.value()).isEqualTo("value1".getBytes()); |
| iterator.next(); |
| assertThat(iterator.isValid()).isTrue(); |
| assertThat(iterator.key()).isEqualTo("key2".getBytes()); |
| assertThat(iterator.value()).isEqualTo("value2".getBytes()); |
| iterator.next(); |
| assertThat(iterator.isValid()).isFalse(); |
| iterator.seekToLast(); |
| iterator.prev(); |
| assertThat(iterator.isValid()).isTrue(); |
| assertThat(iterator.key()).isEqualTo("key1".getBytes()); |
| assertThat(iterator.value()).isEqualTo("value1".getBytes()); |
| iterator.seekToFirst(); |
| iterator.seekToLast(); |
| assertThat(iterator.isValid()).isTrue(); |
| assertThat(iterator.key()).isEqualTo("key2".getBytes()); |
| assertThat(iterator.value()).isEqualTo("value2".getBytes()); |
| iterator.status(); |
| } |
| } |
| } |
| } |