github.com/nutsdb/nutsdb@v1.0.4/iterator_test.go (about) 1 // Copyright 2022 The nutsdb Author. All rights reserved. 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 package nutsdb 16 17 import ( 18 "github.com/stretchr/testify/require" 19 "testing" 20 ) 21 22 func TestIterator(t *testing.T) { 23 bucket := "bucket" 24 25 runNutsDBTest(t, nil, func(t *testing.T, db *DB) { 26 txCreateBucket(t, db, DataStructureBTree, bucket, nil) 27 28 for i := 0; i < 100; i++ { 29 txPut(t, db, bucket, GetTestBytes(i), GetTestBytes(i), Persistent, nil, nil) 30 } 31 32 _ = db.View(func(tx *Tx) error { 33 iterator := NewIterator(tx, bucket, IteratorOptions{Reverse: false}) 34 35 i := 0 36 37 for { 38 value, err := iterator.Value() 39 require.NoError(t, err) 40 require.Equal(t, GetTestBytes(i), value) 41 if !iterator.Next() { 42 break 43 } 44 i++ 45 } 46 47 return nil 48 }) 49 }) 50 } 51 52 func TestIterator_Reverse(t *testing.T) { 53 bucket := "bucket" 54 55 runNutsDBTest(t, nil, func(t *testing.T, db *DB) { 56 txCreateBucket(t, db, DataStructureBTree, bucket, nil) 57 58 for i := 0; i < 100; i++ { 59 txPut(t, db, bucket, GetTestBytes(i), GetTestBytes(i), Persistent, nil, nil) 60 } 61 62 _ = db.View(func(tx *Tx) error { 63 iterator := NewIterator(tx, bucket, IteratorOptions{Reverse: true}) 64 65 i := 99 66 for { 67 value, err := iterator.Value() 68 require.NoError(t, err) 69 require.Equal(t, GetTestBytes(i), value) 70 if !iterator.Next() { 71 break 72 } 73 i-- 74 } 75 76 return nil 77 }) 78 }) 79 } 80 81 func TestIterator_Seek(t *testing.T) { 82 bucket := "bucket" 83 84 runNutsDBTest(t, nil, func(t *testing.T, db *DB) { 85 txCreateBucket(t, db, DataStructureBTree, bucket, nil) 86 87 for i := 0; i < 100; i++ { 88 txPut(t, db, bucket, GetTestBytes(i), GetTestBytes(i), Persistent, nil, nil) 89 } 90 91 _ = db.View(func(tx *Tx) error { 92 iterator := NewIterator(tx, bucket, IteratorOptions{Reverse: true}) 93 94 iterator.Seek(GetTestBytes(40)) 95 96 value, err := iterator.Value() 97 require.NoError(t, err) 98 require.Equal(t, GetTestBytes(40), value) 99 100 return nil 101 }) 102 }) 103 }