raft: fix node bench

Baoqiang-Zhang · Nov 27, 2014 · d244e3b · d244e3b
1 parent fe0bc4f
commit d244e3b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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"))
 	}