Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

clientv3/integration: fix cluster tests

Browse Source 
Fixes TestMemberAddForLearner and TestMemberPromoteForLearner.
This commit is contained in:
Jingyi Hu 2019-04-18 11:07:34 -07:00
parent ac057951cc
commit b23c8f3e8f
1 changed files with 33 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
clientv3/integration/cluster_test.go
View File

@ -202,6 +202,16 @@ func TestMemberAddForLearner(t *testing.T) {
if !resp.Member.IsLearner {
t.Errorf("Added a member as learner, got resp.Member.IsLearner = %v", resp.Member.IsLearner)
}
numberOfLearners := 0
for _, m := range resp.Members {
if m.IsLearner {
numberOfLearners++
}
}
if numberOfLearners != 1 {
t.Errorf("Added 1 learner node to cluster, got %d", numberOfLearners)
}
}
func TestMemberPromoteForLearner(t *testing.T) {
@ -214,35 +224,38 @@ func TestMemberPromoteForLearner(t *testing.T) {
capi := clus.RandClient()
urls := []string{"http://127.0.0.1:1234"}
isLearner := true
resp, err := capi.MemberAddAsLearner(context.Background(), urls)
memberAddResp, err := capi.MemberAddAsLearner(context.Background(), urls)
if err != nil {
t.Fatalf("failed to add member %v", err)
}
if !resp.Member.IsLearner {
t.Errorf("Added a member as learner, got resp.Member.IsLearner = %v", resp.Member.IsLearner)
if !memberAddResp.Member.IsLearner {
t.Fatalf("Added a member as learner, got resp.Member.IsLearner = %v", memberAddResp.Member.IsLearner)
}
learnerID := memberAddResp.Member.ID
numberOfLearners := 0
for _, m := range memberAddResp.Members {
if m.IsLearner {
numberOfLearners++
}
}
if numberOfLearners != 1 {
t.Fatalf("Added 1 learner node to cluster, got %d", numberOfLearners)
}
learners, err := clus.GetLearnerMembers()
memberPromoteResp, err := capi.MemberPromote(context.Background(), learnerID)
if err != nil {
t.Fatalf("failed to get the learner members in cluster: %v", err)
}
if len(learners) != 1 {
t.Errorf("Added 1 learner node to cluster, got %d", len(learners))
}
_, err = capi.MemberPromote(context.Background(), resp.Member.ID)
if err != nil {
t.Fatalf("failed to promote member error: %v", err)
t.Fatalf("failed to promote member: %v", err)
}
learners, err = clus.GetLearnerMembers()
if err != nil {
t.Fatalf("failed to get the number of learners in cluster: %v", err)
numberOfLearners = 0
for _, m := range memberPromoteResp.Members {
if m.IsLearner {
numberOfLearners++
}
}
if len(learners) != 0 {
t.Errorf("learner promoted, expect 0 learner, got %d", len(learners))
if numberOfLearners != 0 {
t.Errorf("learner promoted, expect 0 learner, got %d", numberOfLearners)
}
}