errorz

nsqio · Aug 7, 2015 · ef3662e · ef3662e
1 parent cea5995
commit ef3662e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/nsqd/diskqueue_test.go b/nsqd/diskqueue_test.go
@@ -306,14 +306,17 @@ func benchmarkDiskQueuePut(size int64, b *testing.B) {
 		panic(err)
 	}
 	defer os.RemoveAll(tmpDir)
-	dq := newDiskQueue(dqName, tmpDir, 1024768*100, 0, 1<<10, 2500, 2*time.Second, l)
+	dq := newDiskQueue(dqName, tmpDir, 1024768*100, 0, 1<<20, 2500, 2*time.Second, l)
 	defer dq.Close()
 	b.SetBytes(size)
 	data := make([]byte, size)
 	b.StartTimer()
 
 	for i := 0; i < b.N; i++ {
-		dq.Put(data)
+		err := dq.Put(data)
+		if err != nil {
+			panic(err)
+		}
 	}
 }