github.com/nutsdb/nutsdb@v1.0.4/tx_bucket_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 "testing" 19 20 "github.com/stretchr/testify/assert" 21 ) 22 23 var ( 24 setBucketName = "set_bucket" 25 zSetBucketName = "zset_bucket" 26 listBucketName = "list_bucket" 27 stringBucketName = "string_bucket" 28 ) 29 30 func setupBucket(t *testing.T, db *DB) { 31 key := GetTestBytes(0) 32 val := GetTestBytes(1) 33 txCreateBucket(t, db, DataStructureBTree, stringBucketName, nil) 34 txCreateBucket(t, db, DataStructureSet, setBucketName, nil) 35 txCreateBucket(t, db, DataStructureSortedSet, zSetBucketName, nil) 36 txCreateBucket(t, db, DataStructureList, listBucketName, nil) 37 38 txSAdd(t, db, setBucketName, key, val, nil, nil) 39 txZAdd(t, db, zSetBucketName, key, val, 80, nil, nil) 40 txPush(t, db, listBucketName, key, val, true, nil, nil) 41 txPut(t, db, stringBucketName, key, val, Persistent, nil, nil) 42 } 43 44 func TestBucket_IterateBuckets(t *testing.T) { 45 runNutsDBTest(t, nil, func(t *testing.T, db *DB) { 46 setupBucket(t, db) 47 48 txIterateBuckets(t, db, DataStructureSet, "*", func(bucket string) bool { 49 return true 50 }, nil, setBucketName) 51 52 txIterateBuckets(t, db, DataStructureSortedSet, "*", func(bucket string) bool { 53 return true 54 }, nil, zSetBucketName) 55 56 txIterateBuckets(t, db, DataStructureList, "*", func(bucket string) bool { 57 return true 58 }, nil, listBucketName) 59 60 txIterateBuckets(t, db, DataStructureBTree, "*", func(bucket string) bool { 61 return true 62 }, nil, stringBucketName) 63 64 matched := false 65 txIterateBuckets(t, db, DataStructureBTree, "str*", func(bucket string) bool { 66 matched = true 67 return true 68 }, nil) 69 assert.Equal(t, true, matched) 70 71 matched = false 72 txIterateBuckets(t, db, DataStructureList, "str*", func(bucket string) bool { 73 return true 74 }, nil) 75 assert.Equal(t, false, matched) 76 }) 77 } 78 79 func TestBucket_DeleteBucket(t *testing.T) { 80 runNutsDBTest(t, nil, func(t *testing.T, db *DB) { 81 setupBucket(t, db) 82 83 txDeleteBucket(t, db, DataStructureSet, setBucketName, nil) 84 txDeleteBucket(t, db, DataStructureSortedSet, zSetBucketName, nil) 85 txDeleteBucket(t, db, DataStructureList, listBucketName, nil) 86 txDeleteBucket(t, db, DataStructureBTree, stringBucketName, nil) 87 88 }) 89 }