mirror of
https://github.com/chrislusf/seaweedfs
synced 2024-09-10 02:50:35 +02:00
150 lines
3.3 KiB
Go
150 lines
3.3 KiB
Go
package memdb
|
|
|
|
import (
|
|
"context"
|
|
"github.com/chrislusf/seaweedfs/weed/filer2"
|
|
"testing"
|
|
)
|
|
|
|
func TestCreateAndFind(t *testing.T) {
|
|
filer := filer2.NewFiler(nil, nil)
|
|
store := &MemDbStore{}
|
|
store.Initialize(nil)
|
|
filer.SetStore(store)
|
|
filer.DisableDirectoryCache()
|
|
|
|
ctx := context.Background()
|
|
|
|
fullpath := filer2.FullPath("/home/chris/this/is/one/file1.jpg")
|
|
|
|
entry1 := &filer2.Entry{
|
|
FullPath: fullpath,
|
|
Attr: filer2.Attr{
|
|
Mode: 0440,
|
|
Uid: 1234,
|
|
Gid: 5678,
|
|
},
|
|
}
|
|
|
|
if err := filer.CreateEntry(ctx, entry1); err != nil {
|
|
t.Errorf("create entry %v: %v", entry1.FullPath, err)
|
|
return
|
|
}
|
|
|
|
entry, err := filer.FindEntry(ctx, fullpath)
|
|
|
|
if err != nil {
|
|
t.Errorf("find entry: %v", err)
|
|
return
|
|
}
|
|
|
|
if entry.FullPath != entry1.FullPath {
|
|
t.Errorf("find wrong entry: %v", entry.FullPath)
|
|
return
|
|
}
|
|
|
|
}
|
|
|
|
func TestCreateFileAndList(t *testing.T) {
|
|
filer := filer2.NewFiler(nil, nil)
|
|
store := &MemDbStore{}
|
|
store.Initialize(nil)
|
|
filer.SetStore(store)
|
|
filer.DisableDirectoryCache()
|
|
|
|
ctx := context.Background()
|
|
|
|
entry1 := &filer2.Entry{
|
|
FullPath: filer2.FullPath("/home/chris/this/is/one/file1.jpg"),
|
|
Attr: filer2.Attr{
|
|
Mode: 0440,
|
|
Uid: 1234,
|
|
Gid: 5678,
|
|
},
|
|
}
|
|
|
|
entry2 := &filer2.Entry{
|
|
FullPath: filer2.FullPath("/home/chris/this/is/one/file2.jpg"),
|
|
Attr: filer2.Attr{
|
|
Mode: 0440,
|
|
Uid: 1234,
|
|
Gid: 5678,
|
|
},
|
|
}
|
|
|
|
filer.CreateEntry(ctx, entry1)
|
|
filer.CreateEntry(ctx, entry2)
|
|
|
|
// checking the 2 files
|
|
entries, err := filer.ListDirectoryEntries(ctx, filer2.FullPath("/home/chris/this/is/one/"), "", false, 100)
|
|
|
|
if err != nil {
|
|
t.Errorf("list entries: %v", err)
|
|
return
|
|
}
|
|
|
|
if len(entries) != 2 {
|
|
t.Errorf("list entries count: %v", len(entries))
|
|
return
|
|
}
|
|
|
|
if entries[0].FullPath != entry1.FullPath {
|
|
t.Errorf("find wrong entry 1: %v", entries[0].FullPath)
|
|
return
|
|
}
|
|
|
|
if entries[1].FullPath != entry2.FullPath {
|
|
t.Errorf("find wrong entry 2: %v", entries[1].FullPath)
|
|
return
|
|
}
|
|
|
|
// checking the offset
|
|
entries, err = filer.ListDirectoryEntries(ctx, filer2.FullPath("/home/chris/this/is/one/"), "file1.jpg", false, 100)
|
|
if len(entries) != 1 {
|
|
t.Errorf("list entries count: %v", len(entries))
|
|
return
|
|
}
|
|
|
|
// checking one upper directory
|
|
entries, _ = filer.ListDirectoryEntries(ctx, filer2.FullPath("/home/chris/this/is"), "", false, 100)
|
|
if len(entries) != 1 {
|
|
t.Errorf("list entries count: %v", len(entries))
|
|
return
|
|
}
|
|
|
|
// checking root directory
|
|
entries, _ = filer.ListDirectoryEntries(ctx, filer2.FullPath("/"), "", false, 100)
|
|
if len(entries) != 1 {
|
|
t.Errorf("list entries count: %v", len(entries))
|
|
return
|
|
}
|
|
|
|
// add file3
|
|
file3Path := filer2.FullPath("/home/chris/this/is/file3.jpg")
|
|
entry3 := &filer2.Entry{
|
|
FullPath: file3Path,
|
|
Attr: filer2.Attr{
|
|
Mode: 0440,
|
|
Uid: 1234,
|
|
Gid: 5678,
|
|
},
|
|
}
|
|
filer.CreateEntry(ctx, entry3)
|
|
|
|
// checking one upper directory
|
|
entries, _ = filer.ListDirectoryEntries(ctx, filer2.FullPath("/home/chris/this/is"), "", false, 100)
|
|
if len(entries) != 2 {
|
|
t.Errorf("list entries count: %v", len(entries))
|
|
return
|
|
}
|
|
|
|
// delete file and count
|
|
filer.DeleteEntryMetaAndData(ctx, file3Path, false, false, false)
|
|
entries, _ = filer.ListDirectoryEntries(ctx, filer2.FullPath("/home/chris/this/is"), "", false, 100)
|
|
if len(entries) != 1 {
|
|
t.Errorf("list entries count: %v", len(entries))
|
|
return
|
|
}
|
|
|
|
}
|