|
@@ -31,13 +31,13 @@ import (
|
|
|
func TestClusterFromString(t *testing.T) {
|
|
func TestClusterFromString(t *testing.T) {
|
|
|
tests := []struct {
|
|
tests := []struct {
|
|
|
f string
|
|
f string
|
|
|
- mems []Member
|
|
|
|
|
|
|
+ mems []*Member
|
|
|
}{
|
|
}{
|
|
|
{
|
|
{
|
|
|
"mem1=http://10.0.0.1:2379,mem1=http://128.193.4.20:2379,mem2=http://10.0.0.2:2379,default=http://127.0.0.1:2379",
|
|
"mem1=http://10.0.0.1:2379,mem1=http://128.193.4.20:2379,mem2=http://10.0.0.2:2379,default=http://127.0.0.1:2379",
|
|
|
- []Member{
|
|
|
|
|
- newTestMember(4322322643958477905, []string{"http://10.0.0.1:2379", "http://128.193.4.20:2379"}, "mem1", nil),
|
|
|
|
|
|
|
+ []*Member{
|
|
|
newTestMember(3141198903430435750, []string{"http://10.0.0.2:2379"}, "mem2", nil),
|
|
newTestMember(3141198903430435750, []string{"http://10.0.0.2:2379"}, "mem2", nil),
|
|
|
|
|
+ newTestMember(4322322643958477905, []string{"http://10.0.0.1:2379", "http://128.193.4.20:2379"}, "mem1", nil),
|
|
|
newTestMember(12762790032478827328, []string{"http://127.0.0.1:2379"}, "default", nil),
|
|
newTestMember(12762790032478827328, []string{"http://127.0.0.1:2379"}, "default", nil),
|
|
|
},
|
|
},
|
|
|
},
|
|
},
|
|
@@ -50,9 +50,8 @@ func TestClusterFromString(t *testing.T) {
|
|
|
if c.token != "abc" {
|
|
if c.token != "abc" {
|
|
|
t.Errorf("#%d: token = %v, want abc", i, c.token)
|
|
t.Errorf("#%d: token = %v, want abc", i, c.token)
|
|
|
}
|
|
}
|
|
|
- wc := newTestCluster(tt.mems)
|
|
|
|
|
- if !reflect.DeepEqual(c.members, wc.members) {
|
|
|
|
|
- t.Errorf("#%d: members = %+v, want %+v", i, c.members, wc.members)
|
|
|
|
|
|
|
+ if !reflect.DeepEqual(c.Members(), tt.mems) {
|
|
|
|
|
+ t.Errorf("#%d: members = %+v, want %+v", i, c.Members(), tt.mems)
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
@@ -80,39 +79,39 @@ func TestClusterFromStringBad(t *testing.T) {
|
|
|
|
|
|
|
|
func TestClusterFromStore(t *testing.T) {
|
|
func TestClusterFromStore(t *testing.T) {
|
|
|
tests := []struct {
|
|
tests := []struct {
|
|
|
- mems []Member
|
|
|
|
|
|
|
+ mems []*Member
|
|
|
}{
|
|
}{
|
|
|
{
|
|
{
|
|
|
- []Member{newTestMember(1, nil, "node1", nil)},
|
|
|
|
|
|
|
+ []*Member{newTestMember(1, nil, "", nil)},
|
|
|
},
|
|
},
|
|
|
{
|
|
{
|
|
|
- []Member{},
|
|
|
|
|
|
|
+ nil,
|
|
|
},
|
|
},
|
|
|
{
|
|
{
|
|
|
- []Member{
|
|
|
|
|
- newTestMember(1, nil, "node1", nil),
|
|
|
|
|
- newTestMember(2, nil, "node2", nil),
|
|
|
|
|
|
|
+ []*Member{
|
|
|
|
|
+ newTestMember(1, nil, "", nil),
|
|
|
|
|
+ newTestMember(2, nil, "", nil),
|
|
|
},
|
|
},
|
|
|
},
|
|
},
|
|
|
}
|
|
}
|
|
|
for i, tt := range tests {
|
|
for i, tt := range tests {
|
|
|
hc := newTestCluster(nil)
|
|
hc := newTestCluster(nil)
|
|
|
|
|
+ hc.SetStore(store.New())
|
|
|
for _, m := range tt.mems {
|
|
for _, m := range tt.mems {
|
|
|
- hc.AddMember(&m)
|
|
|
|
|
|
|
+ hc.AddMember(m)
|
|
|
}
|
|
}
|
|
|
c := NewClusterFromStore("abc", hc.store)
|
|
c := NewClusterFromStore("abc", hc.store)
|
|
|
if c.token != "abc" {
|
|
if c.token != "abc" {
|
|
|
t.Errorf("#%d: token = %v, want %v", i, c.token, "abc")
|
|
t.Errorf("#%d: token = %v, want %v", i, c.token, "abc")
|
|
|
}
|
|
}
|
|
|
- wc := newTestCluster(tt.mems)
|
|
|
|
|
- if !reflect.DeepEqual(c.members, wc.members) {
|
|
|
|
|
- t.Errorf("#%d: members = %v, want %v", i, c.members, wc.members)
|
|
|
|
|
|
|
+ if !reflect.DeepEqual(c.Members(), tt.mems) {
|
|
|
|
|
+ t.Errorf("#%d: members = %v, want %v", i, c.Members(), tt.mems)
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
func TestClusterMember(t *testing.T) {
|
|
func TestClusterMember(t *testing.T) {
|
|
|
- membs := []Member{
|
|
|
|
|
|
|
+ membs := []*Member{
|
|
|
newTestMember(1, nil, "node1", nil),
|
|
newTestMember(1, nil, "node1", nil),
|
|
|
newTestMember(2, nil, "node2", nil),
|
|
newTestMember(2, nil, "node2", nil),
|
|
|
}
|
|
}
|
|
@@ -137,7 +136,7 @@ func TestClusterMember(t *testing.T) {
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
func TestClusterMemberByName(t *testing.T) {
|
|
func TestClusterMemberByName(t *testing.T) {
|
|
|
- membs := []Member{
|
|
|
|
|
|
|
+ membs := []*Member{
|
|
|
newTestMember(1, nil, "node1", nil),
|
|
newTestMember(1, nil, "node1", nil),
|
|
|
newTestMember(2, nil, "node2", nil),
|
|
newTestMember(2, nil, "node2", nil),
|
|
|
}
|
|
}
|
|
@@ -162,7 +161,7 @@ func TestClusterMemberByName(t *testing.T) {
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
func TestClusterMemberIDs(t *testing.T) {
|
|
func TestClusterMemberIDs(t *testing.T) {
|
|
|
- c := newTestCluster([]Member{
|
|
|
|
|
|
|
+ c := newTestCluster([]*Member{
|
|
|
newTestMember(1, nil, "", nil),
|
|
newTestMember(1, nil, "", nil),
|
|
|
newTestMember(4, nil, "", nil),
|
|
newTestMember(4, nil, "", nil),
|
|
|
newTestMember(100, nil, "", nil),
|
|
newTestMember(100, nil, "", nil),
|
|
@@ -176,12 +175,12 @@ func TestClusterMemberIDs(t *testing.T) {
|
|
|
|
|
|
|
|
func TestClusterPeerURLs(t *testing.T) {
|
|
func TestClusterPeerURLs(t *testing.T) {
|
|
|
tests := []struct {
|
|
tests := []struct {
|
|
|
- mems []Member
|
|
|
|
|
|
|
+ mems []*Member
|
|
|
wurls []string
|
|
wurls []string
|
|
|
}{
|
|
}{
|
|
|
// single peer with a single address
|
|
// single peer with a single address
|
|
|
{
|
|
{
|
|
|
- mems: []Member{
|
|
|
|
|
|
|
+ mems: []*Member{
|
|
|
newTestMember(1, []string{"http://192.0.2.1"}, "", nil),
|
|
newTestMember(1, []string{"http://192.0.2.1"}, "", nil),
|
|
|
},
|
|
},
|
|
|
wurls: []string{"http://192.0.2.1"},
|
|
wurls: []string{"http://192.0.2.1"},
|
|
@@ -189,7 +188,7 @@ func TestClusterPeerURLs(t *testing.T) {
|
|
|
|
|
|
|
|
// single peer with a single address with a port
|
|
// single peer with a single address with a port
|
|
|
{
|
|
{
|
|
|
- mems: []Member{
|
|
|
|
|
|
|
+ mems: []*Member{
|
|
|
newTestMember(1, []string{"http://192.0.2.1:8001"}, "", nil),
|
|
newTestMember(1, []string{"http://192.0.2.1:8001"}, "", nil),
|
|
|
},
|
|
},
|
|
|
wurls: []string{"http://192.0.2.1:8001"},
|
|
wurls: []string{"http://192.0.2.1:8001"},
|
|
@@ -197,7 +196,7 @@ func TestClusterPeerURLs(t *testing.T) {
|
|
|
|
|
|
|
|
// several members explicitly unsorted
|
|
// several members explicitly unsorted
|
|
|
{
|
|
{
|
|
|
- mems: []Member{
|
|
|
|
|
|
|
+ mems: []*Member{
|
|
|
newTestMember(2, []string{"http://192.0.2.3", "http://192.0.2.4"}, "", nil),
|
|
newTestMember(2, []string{"http://192.0.2.3", "http://192.0.2.4"}, "", nil),
|
|
|
newTestMember(3, []string{"http://192.0.2.5", "http://192.0.2.6"}, "", nil),
|
|
newTestMember(3, []string{"http://192.0.2.5", "http://192.0.2.6"}, "", nil),
|
|
|
newTestMember(1, []string{"http://192.0.2.1", "http://192.0.2.2"}, "", nil),
|
|
newTestMember(1, []string{"http://192.0.2.1", "http://192.0.2.2"}, "", nil),
|
|
@@ -207,13 +206,13 @@ func TestClusterPeerURLs(t *testing.T) {
|
|
|
|
|
|
|
|
// no members
|
|
// no members
|
|
|
{
|
|
{
|
|
|
- mems: []Member{},
|
|
|
|
|
|
|
+ mems: []*Member{},
|
|
|
wurls: []string{},
|
|
wurls: []string{},
|
|
|
},
|
|
},
|
|
|
|
|
|
|
|
// peer with no peer urls
|
|
// peer with no peer urls
|
|
|
{
|
|
{
|
|
|
- mems: []Member{
|
|
|
|
|
|
|
+ mems: []*Member{
|
|
|
newTestMember(3, []string{}, "", nil),
|
|
newTestMember(3, []string{}, "", nil),
|
|
|
},
|
|
},
|
|
|
wurls: []string{},
|
|
wurls: []string{},
|
|
@@ -231,12 +230,12 @@ func TestClusterPeerURLs(t *testing.T) {
|
|
|
|
|
|
|
|
func TestClusterClientURLs(t *testing.T) {
|
|
func TestClusterClientURLs(t *testing.T) {
|
|
|
tests := []struct {
|
|
tests := []struct {
|
|
|
- mems []Member
|
|
|
|
|
|
|
+ mems []*Member
|
|
|
wurls []string
|
|
wurls []string
|
|
|
}{
|
|
}{
|
|
|
// single peer with a single address
|
|
// single peer with a single address
|
|
|
{
|
|
{
|
|
|
- mems: []Member{
|
|
|
|
|
|
|
+ mems: []*Member{
|
|
|
newTestMember(1, nil, "", []string{"http://192.0.2.1"}),
|
|
newTestMember(1, nil, "", []string{"http://192.0.2.1"}),
|
|
|
},
|
|
},
|
|
|
wurls: []string{"http://192.0.2.1"},
|
|
wurls: []string{"http://192.0.2.1"},
|
|
@@ -244,7 +243,7 @@ func TestClusterClientURLs(t *testing.T) {
|
|
|
|
|
|
|
|
// single peer with a single address with a port
|
|
// single peer with a single address with a port
|
|
|
{
|
|
{
|
|
|
- mems: []Member{
|
|
|
|
|
|
|
+ mems: []*Member{
|
|
|
newTestMember(1, nil, "", []string{"http://192.0.2.1:8001"}),
|
|
newTestMember(1, nil, "", []string{"http://192.0.2.1:8001"}),
|
|
|
},
|
|
},
|
|
|
wurls: []string{"http://192.0.2.1:8001"},
|
|
wurls: []string{"http://192.0.2.1:8001"},
|
|
@@ -252,7 +251,7 @@ func TestClusterClientURLs(t *testing.T) {
|
|
|
|
|
|
|
|
// several members explicitly unsorted
|
|
// several members explicitly unsorted
|
|
|
{
|
|
{
|
|
|
- mems: []Member{
|
|
|
|
|
|
|
+ mems: []*Member{
|
|
|
newTestMember(2, nil, "", []string{"http://192.0.2.3", "http://192.0.2.4"}),
|
|
newTestMember(2, nil, "", []string{"http://192.0.2.3", "http://192.0.2.4"}),
|
|
|
newTestMember(3, nil, "", []string{"http://192.0.2.5", "http://192.0.2.6"}),
|
|
newTestMember(3, nil, "", []string{"http://192.0.2.5", "http://192.0.2.6"}),
|
|
|
newTestMember(1, nil, "", []string{"http://192.0.2.1", "http://192.0.2.2"}),
|
|
newTestMember(1, nil, "", []string{"http://192.0.2.1", "http://192.0.2.2"}),
|
|
@@ -262,13 +261,13 @@ func TestClusterClientURLs(t *testing.T) {
|
|
|
|
|
|
|
|
// no members
|
|
// no members
|
|
|
{
|
|
{
|
|
|
- mems: []Member{},
|
|
|
|
|
|
|
+ mems: []*Member{},
|
|
|
wurls: []string{},
|
|
wurls: []string{},
|
|
|
},
|
|
},
|
|
|
|
|
|
|
|
// peer with no client urls
|
|
// peer with no client urls
|
|
|
{
|
|
{
|
|
|
- mems: []Member{
|
|
|
|
|
|
|
+ mems: []*Member{
|
|
|
newTestMember(3, nil, "", []string{}),
|
|
newTestMember(3, nil, "", []string{}),
|
|
|
},
|
|
},
|
|
|
wurls: []string{},
|
|
wurls: []string{},
|
|
@@ -286,34 +285,34 @@ func TestClusterClientURLs(t *testing.T) {
|
|
|
|
|
|
|
|
func TestClusterValidateAndAssignIDsBad(t *testing.T) {
|
|
func TestClusterValidateAndAssignIDsBad(t *testing.T) {
|
|
|
tests := []struct {
|
|
tests := []struct {
|
|
|
- clmembs []Member
|
|
|
|
|
|
|
+ clmembs []*Member
|
|
|
membs []*Member
|
|
membs []*Member
|
|
|
}{
|
|
}{
|
|
|
{
|
|
{
|
|
|
// unmatched length
|
|
// unmatched length
|
|
|
- []Member{
|
|
|
|
|
|
|
+ []*Member{
|
|
|
newTestMember(1, []string{"http://127.0.0.1:2379"}, "", nil),
|
|
newTestMember(1, []string{"http://127.0.0.1:2379"}, "", nil),
|
|
|
},
|
|
},
|
|
|
[]*Member{},
|
|
[]*Member{},
|
|
|
},
|
|
},
|
|
|
{
|
|
{
|
|
|
// unmatched peer urls
|
|
// unmatched peer urls
|
|
|
- []Member{
|
|
|
|
|
|
|
+ []*Member{
|
|
|
newTestMember(1, []string{"http://127.0.0.1:2379"}, "", nil),
|
|
newTestMember(1, []string{"http://127.0.0.1:2379"}, "", nil),
|
|
|
},
|
|
},
|
|
|
[]*Member{
|
|
[]*Member{
|
|
|
- newTestMemberp(1, []string{"http://127.0.0.1:4001"}, "", nil),
|
|
|
|
|
|
|
+ newTestMember(1, []string{"http://127.0.0.1:4001"}, "", nil),
|
|
|
},
|
|
},
|
|
|
},
|
|
},
|
|
|
{
|
|
{
|
|
|
// unmatched peer urls
|
|
// unmatched peer urls
|
|
|
- []Member{
|
|
|
|
|
|
|
+ []*Member{
|
|
|
newTestMember(1, []string{"http://127.0.0.1:2379"}, "", nil),
|
|
newTestMember(1, []string{"http://127.0.0.1:2379"}, "", nil),
|
|
|
newTestMember(2, []string{"http://127.0.0.2:2379"}, "", nil),
|
|
newTestMember(2, []string{"http://127.0.0.2:2379"}, "", nil),
|
|
|
},
|
|
},
|
|
|
[]*Member{
|
|
[]*Member{
|
|
|
- newTestMemberp(1, []string{"http://127.0.0.1:2379"}, "", nil),
|
|
|
|
|
- newTestMemberp(2, []string{"http://127.0.0.2:4001"}, "", nil),
|
|
|
|
|
|
|
+ newTestMember(1, []string{"http://127.0.0.1:2379"}, "", nil),
|
|
|
|
|
+ newTestMember(2, []string{"http://127.0.0.2:4001"}, "", nil),
|
|
|
},
|
|
},
|
|
|
},
|
|
},
|
|
|
}
|
|
}
|
|
@@ -327,18 +326,18 @@ func TestClusterValidateAndAssignIDsBad(t *testing.T) {
|
|
|
|
|
|
|
|
func TestClusterValidateAndAssignIDs(t *testing.T) {
|
|
func TestClusterValidateAndAssignIDs(t *testing.T) {
|
|
|
tests := []struct {
|
|
tests := []struct {
|
|
|
- clmembs []Member
|
|
|
|
|
|
|
+ clmembs []*Member
|
|
|
membs []*Member
|
|
membs []*Member
|
|
|
wids []types.ID
|
|
wids []types.ID
|
|
|
}{
|
|
}{
|
|
|
{
|
|
{
|
|
|
- []Member{
|
|
|
|
|
|
|
+ []*Member{
|
|
|
newTestMember(1, []string{"http://127.0.0.1:2379"}, "", nil),
|
|
newTestMember(1, []string{"http://127.0.0.1:2379"}, "", nil),
|
|
|
newTestMember(2, []string{"http://127.0.0.2:2379"}, "", nil),
|
|
newTestMember(2, []string{"http://127.0.0.2:2379"}, "", nil),
|
|
|
},
|
|
},
|
|
|
[]*Member{
|
|
[]*Member{
|
|
|
- newTestMemberp(3, []string{"http://127.0.0.1:2379"}, "", nil),
|
|
|
|
|
- newTestMemberp(4, []string{"http://127.0.0.2:2379"}, "", nil),
|
|
|
|
|
|
|
+ newTestMember(3, []string{"http://127.0.0.1:2379"}, "", nil),
|
|
|
|
|
+ newTestMember(4, []string{"http://127.0.0.2:2379"}, "", nil),
|
|
|
},
|
|
},
|
|
|
[]types.ID{3, 4},
|
|
[]types.ID{3, 4},
|
|
|
},
|
|
},
|
|
@@ -364,7 +363,25 @@ func TestClusterValidateConfigurationChange(t *testing.T) {
|
|
|
cl.RemoveMember(4)
|
|
cl.RemoveMember(4)
|
|
|
|
|
|
|
|
attr := RaftAttributes{PeerURLs: []string{fmt.Sprintf("http://127.0.0.1:%d", 1)}}
|
|
attr := RaftAttributes{PeerURLs: []string{fmt.Sprintf("http://127.0.0.1:%d", 1)}}
|
|
|
- cxt, err := json.Marshal(&Member{ID: types.ID(5), RaftAttributes: attr})
|
|
|
|
|
|
|
+ ctx, err := json.Marshal(&Member{ID: types.ID(5), RaftAttributes: attr})
|
|
|
|
|
+ if err != nil {
|
|
|
|
|
+ t.Fatal(err)
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ attr = RaftAttributes{PeerURLs: []string{fmt.Sprintf("http://127.0.0.1:%d", 5)}}
|
|
|
|
|
+ ctx5, err := json.Marshal(&Member{ID: types.ID(5), RaftAttributes: attr})
|
|
|
|
|
+ if err != nil {
|
|
|
|
|
+ t.Fatal(err)
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ attr = RaftAttributes{PeerURLs: []string{fmt.Sprintf("http://127.0.0.1:%d", 3)}}
|
|
|
|
|
+ ctx2to3, err := json.Marshal(&Member{ID: types.ID(2), RaftAttributes: attr})
|
|
|
|
|
+ if err != nil {
|
|
|
|
|
+ t.Fatal(err)
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ attr = RaftAttributes{PeerURLs: []string{fmt.Sprintf("http://127.0.0.1:%d", 5)}}
|
|
|
|
|
+ ctx2to5, err := json.Marshal(&Member{ID: types.ID(2), RaftAttributes: attr})
|
|
|
if err != nil {
|
|
if err != nil {
|
|
|
t.Fatal(err)
|
|
t.Fatal(err)
|
|
|
}
|
|
}
|
|
@@ -405,7 +422,7 @@ func TestClusterValidateConfigurationChange(t *testing.T) {
|
|
|
raftpb.ConfChange{
|
|
raftpb.ConfChange{
|
|
|
Type: raftpb.ConfChangeAddNode,
|
|
Type: raftpb.ConfChangeAddNode,
|
|
|
NodeID: 5,
|
|
NodeID: 5,
|
|
|
- Context: cxt,
|
|
|
|
|
|
|
+ Context: ctx,
|
|
|
},
|
|
},
|
|
|
ErrPeerURLexists,
|
|
ErrPeerURLexists,
|
|
|
},
|
|
},
|
|
@@ -416,6 +433,39 @@ func TestClusterValidateConfigurationChange(t *testing.T) {
|
|
|
},
|
|
},
|
|
|
ErrIDNotFound,
|
|
ErrIDNotFound,
|
|
|
},
|
|
},
|
|
|
|
|
+ {
|
|
|
|
|
+ raftpb.ConfChange{
|
|
|
|
|
+ Type: raftpb.ConfChangeAddNode,
|
|
|
|
|
+ NodeID: 5,
|
|
|
|
|
+ Context: ctx5,
|
|
|
|
|
+ },
|
|
|
|
|
+ nil,
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ raftpb.ConfChange{
|
|
|
|
|
+ Type: raftpb.ConfChangeUpdateNode,
|
|
|
|
|
+ NodeID: 5,
|
|
|
|
|
+ Context: ctx,
|
|
|
|
|
+ },
|
|
|
|
|
+ ErrIDNotFound,
|
|
|
|
|
+ },
|
|
|
|
|
+ // try to change the peer url of 2 to the peer url of 3
|
|
|
|
|
+ {
|
|
|
|
|
+ raftpb.ConfChange{
|
|
|
|
|
+ Type: raftpb.ConfChangeUpdateNode,
|
|
|
|
|
+ NodeID: 2,
|
|
|
|
|
+ Context: ctx2to3,
|
|
|
|
|
+ },
|
|
|
|
|
+ ErrPeerURLexists,
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ raftpb.ConfChange{
|
|
|
|
|
+ Type: raftpb.ConfChangeUpdateNode,
|
|
|
|
|
+ NodeID: 2,
|
|
|
|
|
+ Context: ctx2to5,
|
|
|
|
|
+ },
|
|
|
|
|
+ nil,
|
|
|
|
|
+ },
|
|
|
}
|
|
}
|
|
|
for i, tt := range tests {
|
|
for i, tt := range tests {
|
|
|
err := cl.ValidateConfigurationChange(tt.cc)
|
|
err := cl.ValidateConfigurationChange(tt.cc)
|
|
@@ -426,7 +476,7 @@ func TestClusterValidateConfigurationChange(t *testing.T) {
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
func TestClusterGenID(t *testing.T) {
|
|
func TestClusterGenID(t *testing.T) {
|
|
|
- cs := newTestCluster([]Member{
|
|
|
|
|
|
|
+ cs := newTestCluster([]*Member{
|
|
|
newTestMember(1, nil, "", nil),
|
|
newTestMember(1, nil, "", nil),
|
|
|
newTestMember(2, nil, "", nil),
|
|
newTestMember(2, nil, "", nil),
|
|
|
})
|
|
})
|
|
@@ -438,7 +488,7 @@ func TestClusterGenID(t *testing.T) {
|
|
|
previd := cs.ID()
|
|
previd := cs.ID()
|
|
|
|
|
|
|
|
cs.SetStore(&storeRecorder{})
|
|
cs.SetStore(&storeRecorder{})
|
|
|
- cs.AddMember(newTestMemberp(3, nil, "", nil))
|
|
|
|
|
|
|
+ cs.AddMember(newTestMember(3, nil, "", nil))
|
|
|
cs.genID()
|
|
cs.genID()
|
|
|
if cs.ID() == previd {
|
|
if cs.ID() == previd {
|
|
|
t.Fatalf("cluster.ID = %v, want not %v", cs.ID(), previd)
|
|
t.Fatalf("cluster.ID = %v, want not %v", cs.ID(), previd)
|
|
@@ -451,22 +501,22 @@ func TestNodeToMemberBad(t *testing.T) {
|
|
|
{Key: "/1234/strange"},
|
|
{Key: "/1234/strange"},
|
|
|
}},
|
|
}},
|
|
|
{Key: "/1234", Nodes: []*store.NodeExtern{
|
|
{Key: "/1234", Nodes: []*store.NodeExtern{
|
|
|
- {Key: "/1234/dynamic", Value: stringp("garbage")},
|
|
|
|
|
|
|
+ {Key: "/1234/raftAttributes", Value: stringp("garbage")},
|
|
|
}},
|
|
}},
|
|
|
{Key: "/1234", Nodes: []*store.NodeExtern{
|
|
{Key: "/1234", Nodes: []*store.NodeExtern{
|
|
|
- {Key: "/1234/dynamic", Value: stringp(`{"peerURLs":null}`)},
|
|
|
|
|
|
|
+ {Key: "/1234/attributes", Value: stringp(`{"name":"node1","clientURLs":null}`)},
|
|
|
}},
|
|
}},
|
|
|
{Key: "/1234", Nodes: []*store.NodeExtern{
|
|
{Key: "/1234", Nodes: []*store.NodeExtern{
|
|
|
- {Key: "/1234/dynamic", Value: stringp(`{"peerURLs":null}`)},
|
|
|
|
|
|
|
+ {Key: "/1234/raftAttributes", Value: stringp(`{"peerURLs":null}`)},
|
|
|
{Key: "/1234/strange"},
|
|
{Key: "/1234/strange"},
|
|
|
}},
|
|
}},
|
|
|
{Key: "/1234", Nodes: []*store.NodeExtern{
|
|
{Key: "/1234", Nodes: []*store.NodeExtern{
|
|
|
- {Key: "/1234/dynamic", Value: stringp(`{"peerURLs":null}`)},
|
|
|
|
|
- {Key: "/1234/static", Value: stringp("garbage")},
|
|
|
|
|
|
|
+ {Key: "/1234/raftAttributes", Value: stringp(`{"peerURLs":null}`)},
|
|
|
|
|
+ {Key: "/1234/attributes", Value: stringp("garbage")},
|
|
|
}},
|
|
}},
|
|
|
{Key: "/1234", Nodes: []*store.NodeExtern{
|
|
{Key: "/1234", Nodes: []*store.NodeExtern{
|
|
|
- {Key: "/1234/dynamic", Value: stringp(`{"peerURLs":null}`)},
|
|
|
|
|
- {Key: "/1234/static", Value: stringp(`{"name":"node1","clientURLs":null}`)},
|
|
|
|
|
|
|
+ {Key: "/1234/raftAttributes", Value: stringp(`{"peerURLs":null}`)},
|
|
|
|
|
+ {Key: "/1234/attributes", Value: stringp(`{"name":"node1","clientURLs":null}`)},
|
|
|
{Key: "/1234/strange"},
|
|
{Key: "/1234/strange"},
|
|
|
}},
|
|
}},
|
|
|
}
|
|
}
|
|
@@ -481,7 +531,7 @@ func TestClusterAddMember(t *testing.T) {
|
|
|
st := &storeRecorder{}
|
|
st := &storeRecorder{}
|
|
|
c := newTestCluster(nil)
|
|
c := newTestCluster(nil)
|
|
|
c.SetStore(st)
|
|
c.SetStore(st)
|
|
|
- c.AddMember(newTestMemberp(1, nil, "node1", nil))
|
|
|
|
|
|
|
+ c.AddMember(newTestMember(1, nil, "node1", nil))
|
|
|
|
|
|
|
|
wactions := []action{
|
|
wactions := []action{
|
|
|
{
|
|
{
|
|
@@ -494,16 +544,6 @@ func TestClusterAddMember(t *testing.T) {
|
|
|
store.Permanent,
|
|
store.Permanent,
|
|
|
},
|
|
},
|
|
|
},
|
|
},
|
|
|
- {
|
|
|
|
|
- name: "Create",
|
|
|
|
|
- params: []interface{}{
|
|
|
|
|
- path.Join(storeMembersPrefix, "1", "attributes"),
|
|
|
|
|
- false,
|
|
|
|
|
- `{"name":"node1"}`,
|
|
|
|
|
- false,
|
|
|
|
|
- store.Permanent,
|
|
|
|
|
- },
|
|
|
|
|
- },
|
|
|
|
|
}
|
|
}
|
|
|
if g := st.Action(); !reflect.DeepEqual(g, wactions) {
|
|
if g := st.Action(); !reflect.DeepEqual(g, wactions) {
|
|
|
t.Errorf("actions = %v, want %v", g, wactions)
|
|
t.Errorf("actions = %v, want %v", g, wactions)
|
|
@@ -535,32 +575,32 @@ func TestClusterMembers(t *testing.T) {
|
|
|
func TestClusterString(t *testing.T) {
|
|
func TestClusterString(t *testing.T) {
|
|
|
cls := &Cluster{
|
|
cls := &Cluster{
|
|
|
members: map[types.ID]*Member{
|
|
members: map[types.ID]*Member{
|
|
|
- 1: newTestMemberp(
|
|
|
|
|
|
|
+ 1: newTestMember(
|
|
|
1,
|
|
1,
|
|
|
[]string{"http://1.1.1.1:1111", "http://0.0.0.0:0000"},
|
|
[]string{"http://1.1.1.1:1111", "http://0.0.0.0:0000"},
|
|
|
"abc",
|
|
"abc",
|
|
|
nil,
|
|
nil,
|
|
|
),
|
|
),
|
|
|
- 2: newTestMemberp(
|
|
|
|
|
|
|
+ 2: newTestMember(
|
|
|
2,
|
|
2,
|
|
|
[]string{"http://2.2.2.2:2222"},
|
|
[]string{"http://2.2.2.2:2222"},
|
|
|
"def",
|
|
"def",
|
|
|
nil,
|
|
nil,
|
|
|
),
|
|
),
|
|
|
- 3: newTestMemberp(
|
|
|
|
|
|
|
+ 3: newTestMember(
|
|
|
3,
|
|
3,
|
|
|
[]string{"http://3.3.3.3:1234", "http://127.0.0.1:7001"},
|
|
[]string{"http://3.3.3.3:1234", "http://127.0.0.1:7001"},
|
|
|
"ghi",
|
|
"ghi",
|
|
|
nil,
|
|
nil,
|
|
|
),
|
|
),
|
|
|
// no PeerURLs = not included
|
|
// no PeerURLs = not included
|
|
|
- 4: newTestMemberp(
|
|
|
|
|
|
|
+ 4: newTestMember(
|
|
|
4,
|
|
4,
|
|
|
[]string{},
|
|
[]string{},
|
|
|
"four",
|
|
"four",
|
|
|
nil,
|
|
nil,
|
|
|
),
|
|
),
|
|
|
- 5: newTestMemberp(
|
|
|
|
|
|
|
+ 5: newTestMember(
|
|
|
5,
|
|
5,
|
|
|
nil,
|
|
nil,
|
|
|
"five",
|
|
"five",
|
|
@@ -605,24 +645,10 @@ func TestNodeToMember(t *testing.T) {
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-func newTestCluster(membs []Member) *Cluster {
|
|
|
|
|
|
|
+func newTestCluster(membs []*Member) *Cluster {
|
|
|
c := &Cluster{members: make(map[types.ID]*Member), removed: make(map[types.ID]bool)}
|
|
c := &Cluster{members: make(map[types.ID]*Member), removed: make(map[types.ID]bool)}
|
|
|
- c.store = store.New()
|
|
|
|
|
- for i := range membs {
|
|
|
|
|
- c.AddMember(&membs[i])
|
|
|
|
|
|
|
+ for _, m := range membs {
|
|
|
|
|
+ c.members[m.ID] = m
|
|
|
}
|
|
}
|
|
|
return c
|
|
return c
|
|
|
}
|
|
}
|
|
|
-
|
|
|
|
|
-func newTestMember(id uint64, peerURLs []string, name string, clientURLs []string) Member {
|
|
|
|
|
- return Member{
|
|
|
|
|
- ID: types.ID(id),
|
|
|
|
|
- RaftAttributes: RaftAttributes{PeerURLs: peerURLs},
|
|
|
|
|
- Attributes: Attributes{Name: name, ClientURLs: clientURLs},
|
|
|
|
|
- }
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-func newTestMemberp(id uint64, peerURLs []string, name string, clientURLs []string) *Member {
|
|
|
|
|
- m := newTestMember(id, peerURLs, name, clientURLs)
|
|
|
|
|
- return &m
|
|
|
|
|
-}
|
|
|