diff --git a/raft/node_bench_test.go b/raft/node_bench_test.go index a92d883c4..89a37d330 100644 --- a/raft/node_bench_test.go +++ b/raft/node_bench_test.go @@ -27,14 +27,16 @@ func BenchmarkOneNode(b *testing.B) { defer cancel() n := newNode() - r := newRaft(1, []uint64{1}, 10, 1, nil) + s := NewMemoryStorage() + r := newRaft(1, []uint64{1}, 10, 1, s) go n.run(r) defer n.Stop() n.Campaign(ctx) for i := 0; i < b.N; i++ { - <-n.Ready() + rd := <-n.Ready() + s.Append(rd.Entries) n.Advance() n.Propose(ctx, []byte("foo")) }