vitalif
/
etcd
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #1688 from xiang90/cluster_test 

Cluster test
release-2.0
Xiang Li 2014-11-11 11:36:08 -08:00
parent 4b2d6fc70b 67a0de4bbc
commit b87243d827
4 changed files with 74 additions and 81 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

111
etcdserver/cluster_test.go
View File

@ -31,13 +31,13 @@ import (
func TestClusterFromString(t *testing.T) {
tests := []struct {
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",
[]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(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),
},
},
@ -50,9 +50,8 @@ func TestClusterFromString(t *testing.T) {
if c.token != "abc" {
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,16 +79,16 @@ func TestClusterFromStringBad(t *testing.T) {
func TestClusterFromStore(t *testing.T) {
tests := []struct {
mems []Member
mems []*Member
}{
{
[]Member{newTestMember(1, nil, "node1", nil)},
[]*Member{newTestMember(1, nil, "node1", nil)},
},
{
[]Member{},
nil,
},
{
[]Member{
[]*Member{
newTestMember(1, nil, "node1", nil),
newTestMember(2, nil, "node2", nil),
},
@ -98,21 +97,20 @@ func TestClusterFromStore(t *testing.T) {
for i, tt := range tests {
hc := newTestCluster(nil)
for _, m := range tt.mems {
hc.AddMember(&m)
hc.AddMember(m)
}
c := NewClusterFromStore("abc", hc.store)
if 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) {
membs := []Member{
membs := []*Member{
newTestMember(1, nil, "node1", nil),
newTestMember(2, nil, "node2", nil),
}
@ -137,7 +135,7 @@ func TestClusterMember(t *testing.T) {
}
func TestClusterMemberByName(t *testing.T) {
membs := []Member{
membs := []*Member{
newTestMember(1, nil, "node1", nil),
newTestMember(2, nil, "node2", nil),
}
@ -162,7 +160,7 @@ func TestClusterMemberByName(t *testing.T) {
}
func TestClusterMemberIDs(t *testing.T) {
c := newTestCluster([]Member{
c := newTestCluster([]*Member{
newTestMember(1, nil, "", nil),
newTestMember(4, nil, "", nil),
newTestMember(100, nil, "", nil),
@ -176,12 +174,12 @@ func TestClusterMemberIDs(t *testing.T) {
func TestClusterPeerURLs(t *testing.T) {
tests := []struct {
mems []Member
mems []*Member
wurls []string
}{
// single peer with a single address
{
mems: []Member{
mems: []*Member{
newTestMember(1, []string{"http://192.0.2.1"}, "", nil),
},
wurls: []string{"http://192.0.2.1"},
@ -189,7 +187,7 @@ func TestClusterPeerURLs(t *testing.T) {
// single peer with a single address with a port
{
mems: []Member{
mems: []*Member{
newTestMember(1, []string{"http://192.0.2.1:8001"}, "", nil),
},
wurls: []string{"http://192.0.2.1:8001"},
@ -197,7 +195,7 @@ func TestClusterPeerURLs(t *testing.T) {
// several members explicitly unsorted
{
mems: []Member{
mems: []*Member{
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(1, []string{"http://192.0.2.1", "http://192.0.2.2"}, "", nil),
@ -207,13 +205,13 @@ func TestClusterPeerURLs(t *testing.T) {
// no members
{
mems: []Member{},
mems: []*Member{},
wurls: []string{},
},
// peer with no peer urls
{
mems: []Member{
mems: []*Member{
newTestMember(3, []string{}, "", nil),
},
wurls: []string{},
@ -231,12 +229,12 @@ func TestClusterPeerURLs(t *testing.T) {
func TestClusterClientURLs(t *testing.T) {
tests := []struct {
mems []Member
mems []*Member
wurls []string
}{
// single peer with a single address
{
mems: []Member{
mems: []*Member{
newTestMember(1, nil, "", []string{"http://192.0.2.1"}),
},
wurls: []string{"http://192.0.2.1"},
@ -244,7 +242,7 @@ func TestClusterClientURLs(t *testing.T) {
// single peer with a single address with a port
{
mems: []Member{
mems: []*Member{
newTestMember(1, nil, "", []string{"http://192.0.2.1:8001"}),
},
wurls: []string{"http://192.0.2.1:8001"},
@ -252,7 +250,7 @@ func TestClusterClientURLs(t *testing.T) {
// several members explicitly unsorted
{
mems: []Member{
mems: []*Member{
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(1, nil, "", []string{"http://192.0.2.1", "http://192.0.2.2"}),
@ -262,13 +260,13 @@ func TestClusterClientURLs(t *testing.T) {
// no members
{
mems: []Member{},
mems: []*Member{},
wurls: []string{},
},
// peer with no client urls
{
mems: []Member{
mems: []*Member{
newTestMember(3, nil, "", []string{}),
},
wurls: []string{},
@ -286,34 +284,34 @@ func TestClusterClientURLs(t *testing.T) {
func TestClusterValidateAndAssignIDsBad(t *testing.T) {
tests := []struct {
clmembs []Member
clmembs []*Member
membs []*Member
}{
{
// unmatched length
[]Member{
[]*Member{
newTestMember(1, []string{"http://127.0.0.1:2379"}, "", nil),
},
[]*Member{},
},
{
// unmatched peer urls
[]Member{
[]*Member{
newTestMember(1, []string{"http://127.0.0.1:2379"}, "", nil),
},
[]*Member{
newTestMemberp(1, []string{"http://127.0.0.1:4001"}, "", nil),
newTestMember(1, []string{"http://127.0.0.1:4001"}, "", nil),
},
},
{
// unmatched peer urls
[]Member{
[]*Member{
newTestMember(1, []string{"http://127.0.0.1:2379"}, "", nil),
newTestMember(2, []string{"http://127.0.0.2:2379"}, "", nil),
},
[]*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 +325,18 @@ func TestClusterValidateAndAssignIDsBad(t *testing.T) {
func TestClusterValidateAndAssignIDs(t *testing.T) {
tests := []struct {
clmembs []Member
clmembs []*Member
membs []*Member
wids []types.ID
}{
{
[]Member{
[]*Member{
newTestMember(1, []string{"http://127.0.0.1:2379"}, "", nil),
newTestMember(2, []string{"http://127.0.0.2:2379"}, "", nil),
},
[]*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},
},
@ -426,7 +424,7 @@ func TestClusterValidateConfigurationChange(t *testing.T) {
}
func TestClusterGenID(t *testing.T) {
cs := newTestCluster([]Member{
cs := newTestCluster([]*Member{
newTestMember(1, nil, "", nil),
newTestMember(2, nil, "", nil),
})
@ -438,7 +436,7 @@ func TestClusterGenID(t *testing.T) {
previd := cs.ID()
cs.SetStore(&storeRecorder{})
cs.AddMember(newTestMemberp(3, nil, "", nil))
cs.AddMember(newTestMember(3, nil, "", nil))
cs.genID()
if cs.ID() == previd {
t.Fatalf("cluster.ID = %v, want not %v", cs.ID(), previd)
@ -481,7 +479,7 @@ func TestClusterAddMember(t *testing.T) {
st := &storeRecorder{}
c := newTestCluster(nil)
c.SetStore(st)
c.AddMember(newTestMemberp(1, nil, "node1", nil))
c.AddMember(newTestMember(1, nil, "node1", nil))
wactions := []action{
{
@ -535,32 +533,32 @@ func TestClusterMembers(t *testing.T) {
func TestClusterString(t *testing.T) {
cls := &Cluster{
members: map[types.ID]*Member{
1: newTestMemberp(
1: newTestMember(
1,
[]string{"http://1.1.1.1:1111", "http://0.0.0.0:0000"},
"abc",
nil,
),
2: newTestMemberp(
2: newTestMember(
2,
[]string{"http://2.2.2.2:2222"},
"def",
nil,
),
3: newTestMemberp(
3: newTestMember(
3,
[]string{"http://3.3.3.3:1234", "http://127.0.0.1:7001"},
"ghi",
nil,
),
// no PeerURLs = not included
4: newTestMemberp(
4: newTestMember(
4,
[]string{},
"four",
nil,
),
5: newTestMemberp(
5: newTestMember(
5,
nil,
"five",
@ -605,24 +603,11 @@ 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.store = store.New()
for i := range membs {
c.AddMember(&membs[i])
c.AddMember(membs[i])
}
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
}

20
etcdserver/member_test.go
View File

@ -63,7 +63,7 @@ func TestMemberPick(t *testing.T) {
urls map[string]bool
}{
{
newTestMemberp(1, []string{"abc", "def", "ghi", "jkl", "mno", "pqr", "stu"}, "", nil),
newTestMember(1, []string{"abc", "def", "ghi", "jkl", "mno", "pqr", "stu"}, "", nil),
map[string]bool{
"abc": true,
"def": true,
@ -75,7 +75,7 @@ func TestMemberPick(t *testing.T) {
},
},
{
newTestMemberp(2, []string{"xyz"}, "", nil),
newTestMember(2, []string{"xyz"}, "", nil),
map[string]bool{"xyz": true},
},
}
@ -92,10 +92,10 @@ func TestMemberPick(t *testing.T) {
func TestMemberClone(t *testing.T) {
tests := []*Member{
newTestMemberp(1, nil, "abc", nil),
newTestMemberp(1, []string{"http://a"}, "abc", nil),
newTestMemberp(1, nil, "abc", []string{"http://b"}),
newTestMemberp(1, []string{"http://a"}, "abc", []string{"http://b"}),
newTestMember(1, nil, "abc", nil),
newTestMember(1, []string{"http://a"}, "abc", nil),
newTestMember(1, nil, "abc", []string{"http://b"}),
newTestMember(1, []string{"http://a"}, "abc", []string{"http://b"}),
}
for i, tt := range tests {
nm := tt.Clone()
@ -107,3 +107,11 @@ func TestMemberClone(t *testing.T) {
}
}
}
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},
}
}

6
etcdserver/sender_test.go
View File

@ -24,7 +24,7 @@ import (
)
func TestSendHubInitSenders(t *testing.T) {
membs := []Member{
membs := []*Member{
newTestMember(1, []string{"http://a"}, "", nil),
newTestMember(2, []string{"http://b"}, "", nil),
newTestMember(3, []string{"http://c"}, "", nil),
@ -48,7 +48,7 @@ func TestSendHubAdd(t *testing.T) {
cl := newTestCluster(nil)
ls := stats.NewLeaderStats("")
h := newSendHub(nil, cl, nil, ls)
m := newTestMemberp(1, []string{"http://a"}, "", nil)
m := newTestMember(1, []string{"http://a"}, "", nil)
h.Add(m)
if _, ok := ls.Followers["1"]; !ok {
@ -70,7 +70,7 @@ func TestSendHubAdd(t *testing.T) {
}
func TestSendHubRemove(t *testing.T) {
membs := []Member{
membs := []*Member{
newTestMember(1, []string{"http://a"}, "", nil),
}
cl := newTestCluster(membs)

18
etcdserver/server_test.go
View File

@ -407,7 +407,7 @@ func TestApplyRequest(t *testing.T) {
}
func TestApplyRequestOnAdminMemberAttributes(t *testing.T) {
cl := newTestCluster([]Member{{ID: 1}})
cl := newTestCluster([]*Member{{ID: 1}})
srv := &EtcdServer{
store: &storeRecorder{},
Cluster: cl,
@ -992,7 +992,7 @@ func TestRemoveMember(t *testing.T) {
Nodes: []uint64{1234, 2345, 3456},
},
}
cl := newTestCluster([]Member{{ID: 1234}})
cl := newTestCluster([]*Member{{ID: 1234}})
s := &EtcdServer{
node: n,
store: &storeRecorder{},
@ -1101,25 +1101,25 @@ func TestGetOtherPeerURLs(t *testing.T) {
}{
{
[]*Member{
newTestMemberp(1, []string{"http://10.0.0.1"}, "a", nil),
newTestMember(1, []string{"http://10.0.0.1"}, "a", nil),
},
"a",
[]string{},
},
{
[]*Member{
newTestMemberp(1, []string{"http://10.0.0.1"}, "a", nil),
newTestMemberp(2, []string{"http://10.0.0.2"}, "b", nil),
newTestMemberp(3, []string{"http://10.0.0.3"}, "c", nil),
newTestMember(1, []string{"http://10.0.0.1"}, "a", nil),
newTestMember(2, []string{"http://10.0.0.2"}, "b", nil),
newTestMember(3, []string{"http://10.0.0.3"}, "c", nil),
},
"a",
[]string{"http://10.0.0.2", "http://10.0.0.3"},
},
{
[]*Member{
newTestMemberp(1, []string{"http://10.0.0.1"}, "a", nil),
newTestMemberp(3, []string{"http://10.0.0.3"}, "c", nil),
newTestMemberp(2, []string{"http://10.0.0.2"}, "b", nil),
newTestMember(1, []string{"http://10.0.0.1"}, "a", nil),
newTestMember(3, []string{"http://10.0.0.3"}, "c", nil),
newTestMember(2, []string{"http://10.0.0.2"}, "b", nil),
},
"a",
[]string{"http://10.0.0.2", "http://10.0.0.3"},