mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-12-17 12:07:50 +08:00
77 lines
1.5 KiB
Go
77 lines
1.5 KiB
Go
package wdclient
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
)
|
|
|
|
func TestLocationIndex(t *testing.T) {
|
|
vm := vidMap{}
|
|
// test must be failed
|
|
mustFailed := func(length int) {
|
|
_, err := vm.getLocationIndex(length)
|
|
if err == nil {
|
|
t.Errorf("length %d must be failed", length)
|
|
}
|
|
if err.Error() != fmt.Sprintf("invalid length: %d", length) {
|
|
t.Errorf("length %d must be failed. error: %v", length, err)
|
|
}
|
|
}
|
|
|
|
mustFailed(-1)
|
|
mustFailed(0)
|
|
|
|
mustOk := func(length, cursor, expect int) {
|
|
if length <= 0 {
|
|
t.Fatal("please don't do this")
|
|
}
|
|
vm.cursor = int32(cursor)
|
|
got, err := vm.getLocationIndex(length)
|
|
if err != nil {
|
|
t.Errorf("length: %d, why? %v\n", length, err)
|
|
return
|
|
}
|
|
if got != expect {
|
|
t.Errorf("cursor: %d, length: %d, expect: %d, got: %d\n", cursor, length, expect, got)
|
|
return
|
|
}
|
|
}
|
|
|
|
for i := -1; i < 100; i++ {
|
|
mustOk(7, i, (i+1)%7)
|
|
}
|
|
|
|
// when cursor reaches MaxInt64
|
|
mustOk(7, maxCursorIndex, 0)
|
|
|
|
// test with constructor
|
|
vm = newVidMap("")
|
|
length := 7
|
|
for i := 0; i < 100; i++ {
|
|
got, err := vm.getLocationIndex(length)
|
|
if err != nil {
|
|
t.Errorf("length: %d, why? %v\n", length, err)
|
|
return
|
|
}
|
|
if got != i%length {
|
|
t.Errorf("length: %d, i: %d, got: %d\n", length, i, got)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkLocationIndex(b *testing.B) {
|
|
b.SetParallelism(8)
|
|
vm := vidMap{
|
|
cursor: maxCursorIndex - 4000,
|
|
}
|
|
b.ResetTimer()
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
_, err := vm.getLocationIndex(3)
|
|
if err != nil {
|
|
b.Error(err)
|
|
}
|
|
}
|
|
})
|
|
}
|