distributed/talent-plan-tinykv
1
0
Fork 0
mirror of https://github.com/talent-plan/tinykv.git synced 2024-12-26 12:50:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

ut: add TestRestoreFromSnapWithOverlapingPeersMsg2C (#408)

Browse Source
This commit is contained in:
vince.wu 2022-07-13 14:30:03 +08:00 committed by GitHub
parent e7de6af1d4
commit 6ecddc7f00
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 29 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
raft/raft_test.go
View File

@ -1094,6 +1094,35 @@ func TestRestoreFromSnapMsg2C(t *testing.T) {
} }
} }
func TestRestoreFromSnapWithOverlapingPeersMsg2C(t *testing.T) {
s := pb.Snapshot{
Metadata: &pb.SnapshotMetadata{
Index: 11, // magic number
Term: 11, // magic number
ConfState: &pb.ConfState{Nodes: []uint64{2, 3, 4}},
},
}
m := pb.Message{MsgType: pb.MessageType_MsgSnapshot, From: 1, Term: 2, Snapshot: &s}
sm := newTestRaft(2, []uint64{1, 2, 3}, 10, 1, NewMemoryStorage())
sm.Step(m)
if sm.Lead != uint64(1) {
t.Errorf("sm.Lead = %d, want 1", sm.Lead)
}
nodes := s.Metadata.ConfState.Nodes
if len(nodes) != len(sm.Prs) {
t.Errorf("len(sm.Prs) = %d, want %d", len(sm.Prs), len(nodes))
}
for _, p := range nodes {
if _, ok := sm.Prs[p]; !ok {
t.Errorf("missing peer %d", p)
}
}
}
func TestSlowNodeRestore2C(t *testing.T) { func TestSlowNodeRestore2C(t *testing.T) {
nt := newNetwork(nil, nil, nil) nt := newNetwork(nil, nil, nil)
nt.send(pb.Message{From: 1, To: 1, MsgType: pb.MessageType_MsgHup}) nt.send(pb.Message{From: 1, To: 1, MsgType: pb.MessageType_MsgHup})