From c703ccab638a8f2b892e6ce215bd1c1dd2f17588 Mon Sep 17 00:00:00 2001 From: "swingbach@gmail.com" Date: Wed, 18 May 2016 14:20:02 +0800 Subject: [PATCH] raft: add more assertions for dueling candidates test case --- raft/raft_test.go | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/raft/raft_test.go b/raft/raft_test.go index 1a58f957d..f100a246f 100644 --- a/raft/raft_test.go +++ b/raft/raft_test.go @@ -476,6 +476,16 @@ func TestDuelingCandidates(t *testing.T) { nt.send(pb.Message{From: 1, To: 1, Type: pb.MsgHup}) nt.send(pb.Message{From: 3, To: 3, Type: pb.MsgHup}) + sm := nt.peers[1].(*raft) + if sm.state != StateLeader { + t.Errorf("state = %s, want %s", sm.state, StateLeader) + } + + sm = nt.peers[3].(*raft) + if sm.state != StateCandidate { + t.Errorf("state = %s, want %s", sm.state, StateCandidate) + } + nt.recover() nt.send(pb.Message{From: 3, To: 3, Type: pb.MsgHup})