go.etcd.io/etcd@v3.3.27+incompatible/integration/v3_leadership_test.go (about) 1 // Copyright 2017 The etcd Authors 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 package integration 16 17 import ( 18 "context" 19 "testing" 20 "time" 21 22 "github.com/coreos/etcd/etcdserver/api/v3rpc/rpctypes" 23 pb "github.com/coreos/etcd/etcdserver/etcdserverpb" 24 "github.com/coreos/etcd/pkg/testutil" 25 ) 26 27 func TestMoveLeader(t *testing.T) { testMoveLeader(t, true) } 28 func TestMoveLeaderService(t *testing.T) { testMoveLeader(t, false) } 29 30 func testMoveLeader(t *testing.T, auto bool) { 31 defer testutil.AfterTest(t) 32 33 clus := NewClusterV3(t, &ClusterConfig{Size: 3}) 34 defer clus.Terminate(t) 35 36 oldLeadIdx := clus.WaitLeader(t) 37 oldLeadID := uint64(clus.Members[oldLeadIdx].s.ID()) 38 39 // ensure followers go through leader transition while learship transfer 40 idc := make(chan uint64) 41 for i := range clus.Members { 42 if oldLeadIdx != i { 43 go func(m *member) { 44 idc <- checkLeaderTransition(t, m, oldLeadID) 45 }(clus.Members[i]) 46 } 47 } 48 49 target := uint64(clus.Members[(oldLeadIdx+1)%3].s.ID()) 50 if auto { 51 err := clus.Members[oldLeadIdx].s.TransferLeadership() 52 if err != nil { 53 t.Fatal(err) 54 } 55 } else { 56 mvc := toGRPC(clus.Client(oldLeadIdx)).Maintenance 57 _, err := mvc.MoveLeader(context.TODO(), &pb.MoveLeaderRequest{TargetID: target}) 58 if err != nil { 59 t.Fatal(err) 60 } 61 } 62 63 // wait until leader transitions have happened 64 var newLeadIDs [2]uint64 65 for i := range newLeadIDs { 66 select { 67 case newLeadIDs[i] = <-idc: 68 case <-time.After(time.Second): 69 t.Fatal("timed out waiting for leader transition") 70 } 71 } 72 73 // remaining members must agree on the same leader 74 if newLeadIDs[0] != newLeadIDs[1] { 75 t.Fatalf("expected same new leader %d == %d", newLeadIDs[0], newLeadIDs[1]) 76 } 77 78 // new leader must be different than the old leader 79 if oldLeadID == newLeadIDs[0] { 80 t.Fatalf("expected old leader %d != new leader %d", oldLeadID, newLeadIDs[0]) 81 } 82 83 // if move-leader were used, new leader must match transferee 84 if !auto { 85 if newLeadIDs[0] != target { 86 t.Fatalf("expected new leader %d != target %d", newLeadIDs[0], target) 87 } 88 } 89 } 90 91 // TestMoveLeaderError ensures that request to non-leader fail. 92 func TestMoveLeaderError(t *testing.T) { 93 defer testutil.AfterTest(t) 94 95 clus := NewClusterV3(t, &ClusterConfig{Size: 3}) 96 defer clus.Terminate(t) 97 98 oldLeadIdx := clus.WaitLeader(t) 99 followerIdx := (oldLeadIdx + 1) % 3 100 101 target := uint64(clus.Members[(oldLeadIdx+2)%3].s.ID()) 102 103 mvc := toGRPC(clus.Client(followerIdx)).Maintenance 104 _, err := mvc.MoveLeader(context.TODO(), &pb.MoveLeaderRequest{TargetID: target}) 105 if !eqErrGRPC(err, rpctypes.ErrGRPCNotLeader) { 106 t.Errorf("err = %v, want %v", err, rpctypes.ErrGRPCNotLeader) 107 } 108 }