From c7c26f31c43450d405474903f4f5765c61c8a28e Mon Sep 17 00:00:00 2001
From: kikimo <fzuwwl@gmail.com>
Date: Fri, 2 Jun 2023 09:21:50 +0800
Subject: [PATCH] some minor fixes for raft test

Signed-off-by: kikimo <fzuwwl@gmail.com>
---
 raft_test.go | 14 +++++---------
 1 file changed, 5 insertions(+), 9 deletions(-)

diff --git a/raft_test.go b/raft_test.go
index 9f8f4a45..64337d37 100644
--- a/raft_test.go
+++ b/raft_test.go
@@ -461,13 +461,9 @@ func TestLearnerCanVote(t *testing.T) {
 	n2.Step(pb.Message{From: 1, To: 2, Term: 2, Type: pb.MsgVote, LogTerm: 11, Index: 11})
 
 	msgs := n2.readMessages()
-	if len(msgs) != 1 {
-		t.Fatalf("expected exactly one message, not %+v", msgs)
-	}
-	msg := msgs[0]
-	if msg.Type != pb.MsgVoteResp && !msg.Reject {
-		t.Fatal("expected learner to not reject vote")
-	}
+	require.Len(t, msgs, 1)
+	require.Equal(t, msgs[0].Type, pb.MsgVoteResp)
+	require.False(t, msgs[0].Reject, "expected learner to not reject vote")
 }
 
 func TestLeaderCycle(t *testing.T) {
@@ -835,7 +831,7 @@ func TestCommitWithoutNewTermEntry(t *testing.T) {
 	tt := newNetwork(nil, nil, nil, nil, nil)
 	tt.send(pb.Message{From: 1, To: 1, Type: pb.MsgHup})
 
-	// 0 cannot reach 2,3,4
+	// 0 cannot reach 3,4,5
 	tt.cut(1, 3)
 	tt.cut(1, 4)
 	tt.cut(1, 5)
@@ -1725,7 +1721,7 @@ func TestAllServerStepdown(t *testing.T) {
 				wlead = None
 			}
 			if sm.lead != wlead {
-				t.Errorf("#%d, sm.lead = %d, want %d", i, sm.lead, None)
+				t.Errorf("#%d, sm.lead = %d, want %d", i, sm.lead, wlead)
 			}
 		}
 	}