1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
package store_test
import (
"context"
"fmt"
"github.com/seaweedfs/seaweedfs/weed/filer"
"github.com/seaweedfs/seaweedfs/weed/util"
"github.com/stretchr/testify/assert"
"os"
"testing"
)
func TestFilerStore(t *testing.T, store filer.FilerStore) {
ctx := context.Background()
store.InsertEntry(ctx, makeEntry(util.FullPath("/"), true))
store.InsertEntry(ctx, makeEntry(util.FullPath("/a"), true))
store.InsertEntry(ctx, makeEntry(util.FullPath("/a/b"), true))
store.InsertEntry(ctx, makeEntry(util.FullPath("/a/b/c"), true))
for i := 0; i < 2000; i++ {
store.InsertEntry(ctx, makeEntry(util.FullPath(fmt.Sprintf("/a/b/c/f%05d", i)), false))
}
{
var counter int
lastFileName, err := store.ListDirectoryEntries(ctx, util.FullPath("/a/b/c"), "", false, 3, func(entry *filer.Entry) bool {
counter++
return true
})
assert.Nil(t, err, "list directory")
assert.Equal(t, 3, counter, "directory list counter")
assert.Equal(t, "f00002", lastFileName, "directory list last file")
lastFileName, err = store.ListDirectoryEntries(ctx, util.FullPath("/a/b/c"), lastFileName, false, 1024, func(entry *filer.Entry) bool {
counter++
return true
})
assert.Nil(t, err, "list directory")
assert.Equal(t, 1027, counter, "directory list counter")
assert.Equal(t, "f01026", lastFileName, "directory list last file")
}
testKey := []byte("test_key")
testValue1 := []byte("test_value1")
testValue2 := []byte("test_value2")
err := store.KvPut(ctx, testKey, testValue1)
assert.Nil(t, err, "KV put")
value, err := store.KvGet(ctx, testKey)
assert.Equal(t, value, testValue1, "KV get")
err = store.KvPut(ctx, testKey, testValue2)
assert.Nil(t, err, "KV update")
value, err = store.KvGet(ctx, testKey)
assert.Equal(t, value, testValue2, "KV get after update")
}
func makeEntry(fullPath util.FullPath, isDirectory bool) *filer.Entry {
var mode os.FileMode
if isDirectory {
mode = os.ModeDir
}
return &filer.Entry{
FullPath: fullPath,
Attr: filer.Attr{
Mode: mode,
},
}
}
|
