From 920b155f17ed716037942ad685b6971c60f26d41 Mon Sep 17 00:00:00 2001
From: "Dylan.Wen" <hhkbp2@gmail.com>
Date: Mon, 26 Dec 2016 14:31:59 +0800
Subject: [PATCH] raft: fix pre-vote tests

---
 raft/raft_test.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/raft/raft_test.go b/raft/raft_test.go
index 371bc24ce..83108a852 100644
--- a/raft/raft_test.go
+++ b/raft/raft_test.go
@@ -510,7 +510,7 @@ func testVoteFromAnyState(t *testing.T, vt pb.MessageType) {
 		// If this was a real vote, we reset our state and term.
 		if vt == pb.MsgVote {
 			if r.state != StateFollower {
-				t.Errorf("%s,%s: state %s, want %s", vt, StateFollower, r.state, st)
+				t.Errorf("%s,%s: state %s, want %s", vt, st, r.state, StateFollower)
 			}
 			if r.Term != newTerm {
 				t.Errorf("%s,%s: term %d, want %d", vt, st, r.Term, newTerm)
@@ -1474,6 +1474,8 @@ func TestAllServerStepdown(t *testing.T) {
 		switch tt.state {
 		case StateFollower:
 			sm.becomeFollower(1, None)
+		case StatePreCandidate:
+			sm.becomePreCandidate()
 		case StateCandidate:
 			sm.becomeCandidate()
 		case StateLeader: