mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-12-16 19:47:51 +08:00
52 lines
1.5 KiB
Go
52 lines
1.5 KiB
Go
|
package lock_manager
|
||
|
|
||
|
import (
|
||
|
"github.com/seaweedfs/seaweedfs/weed/pb"
|
||
|
"github.com/stretchr/testify/assert"
|
||
|
"testing"
|
||
|
"time"
|
||
|
)
|
||
|
|
||
|
func TestAddServer(t *testing.T) {
|
||
|
counter := 0
|
||
|
r := NewLockRing(100*time.Millisecond, func(snapshot []pb.ServerAddress) {
|
||
|
counter++
|
||
|
if counter == 1 {
|
||
|
assert.Equal(t, 1, len(snapshot))
|
||
|
} else if counter == 2 {
|
||
|
assert.Equal(t, 2, len(snapshot))
|
||
|
}
|
||
|
})
|
||
|
r.AddServer("localhost:8080")
|
||
|
assert.Equal(t, 1, len(r.snapshots))
|
||
|
r.AddServer("localhost:8081")
|
||
|
r.AddServer("localhost:8082")
|
||
|
r.AddServer("localhost:8083")
|
||
|
r.AddServer("localhost:8084")
|
||
|
r.RemoveServer("localhost:8084")
|
||
|
r.RemoveServer("localhost:8082")
|
||
|
r.RemoveServer("localhost:8080")
|
||
|
|
||
|
assert.Equal(t, 8, len(r.snapshots))
|
||
|
|
||
|
time.Sleep(110 * time.Millisecond)
|
||
|
|
||
|
assert.Equal(t, 2, len(r.snapshots))
|
||
|
|
||
|
}
|
||
|
|
||
|
func TestLockRing(t *testing.T) {
|
||
|
r := NewLockRing(100*time.Millisecond, nil)
|
||
|
r.SetSnapshot([]pb.ServerAddress{"localhost:8080", "localhost:8081"})
|
||
|
assert.Equal(t, 1, len(r.snapshots))
|
||
|
r.SetSnapshot([]pb.ServerAddress{"localhost:8080", "localhost:8081", "localhost:8082"})
|
||
|
assert.Equal(t, 2, len(r.snapshots))
|
||
|
time.Sleep(110 * time.Millisecond)
|
||
|
r.SetSnapshot([]pb.ServerAddress{"localhost:8080", "localhost:8081", "localhost:8082", "localhost:8083"})
|
||
|
assert.Equal(t, 3, len(r.snapshots))
|
||
|
time.Sleep(110 * time.Millisecond)
|
||
|
assert.Equal(t, 2, len(r.snapshots))
|
||
|
r.SetSnapshot([]pb.ServerAddress{"localhost:8080", "localhost:8081", "localhost:8082", "localhost:8083", "localhost:8084"})
|
||
|
assert.Equal(t, 3, len(r.snapshots))
|
||
|
}
|