mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-12-11 15:59:06 +08:00
48 lines
1.4 KiB
Go
48 lines
1.4 KiB
Go
package cluster
|
|
|
|
import (
|
|
"github.com/chrislusf/seaweedfs/weed/pb"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
)
|
|
|
|
func TestClusterAddRemoveNodes(t *testing.T) {
|
|
c := NewCluster()
|
|
|
|
c.AddClusterNode("", "filer", pb.ServerAddress("111:1"), "23.45")
|
|
c.AddClusterNode("", "filer", pb.ServerAddress("111:2"), "23.45")
|
|
assert.Equal(t, []pb.ServerAddress{
|
|
pb.ServerAddress("111:1"),
|
|
pb.ServerAddress("111:2"),
|
|
}, c.getFilers("", false).leaders.GetLeaders())
|
|
|
|
c.AddClusterNode("", "filer", pb.ServerAddress("111:3"), "23.45")
|
|
c.AddClusterNode("", "filer", pb.ServerAddress("111:4"), "23.45")
|
|
assert.Equal(t, []pb.ServerAddress{
|
|
pb.ServerAddress("111:1"),
|
|
pb.ServerAddress("111:2"),
|
|
pb.ServerAddress("111:3"),
|
|
}, c.getFilers("", false).leaders.GetLeaders())
|
|
|
|
c.AddClusterNode("", "filer", pb.ServerAddress("111:5"), "23.45")
|
|
c.AddClusterNode("", "filer", pb.ServerAddress("111:6"), "23.45")
|
|
c.RemoveClusterNode("", "filer", pb.ServerAddress("111:4"))
|
|
assert.Equal(t, []pb.ServerAddress{
|
|
pb.ServerAddress("111:1"),
|
|
pb.ServerAddress("111:2"),
|
|
pb.ServerAddress("111:3"),
|
|
}, c.getFilers("", false).leaders.GetLeaders())
|
|
|
|
// remove oldest
|
|
c.RemoveClusterNode("", "filer", pb.ServerAddress("111:1"))
|
|
assert.Equal(t, []pb.ServerAddress{
|
|
pb.ServerAddress("111:6"),
|
|
pb.ServerAddress("111:2"),
|
|
pb.ServerAddress("111:3"),
|
|
}, c.getFilers("", false).leaders.GetLeaders())
|
|
|
|
// remove oldest
|
|
c.RemoveClusterNode("", "filer", pb.ServerAddress("111:1"))
|
|
|
|
}
|