fix benchmarks

1 files changed, 17 insertions, 18 deletions

diff --git a/satp/packet_test.go b/satp/packet_test.go
index 056727f..5db7828 100644
--- a/satp/packet_test.go
+++ b/satp/packet_test.go
@@ -33,8 +33,6 @@ package satp
 import (
 	"bytes"
 	"math/rand"
-	// "reflect"
-	"io/ioutil"
 	"testing"
 )
 
@@ -354,19 +352,24 @@ func TestEncryptedPacketReadWrite(t *testing.T) {
 // Benchmarking
 //
 
+type devZero struct {
+}
+
+func (dz *devZero) Read(data []byte) (int, error) {
+	return len(data), nil
+}
+
+func (dz *devZero) Write(data []byte) (int, error) {
+	return len(data), nil
+}
+
 func BenchmarkPlainPacketReadFrom(b *testing.B) {
-	ins := []*bytes.Buffer{}
-	for i := 0; i < NUM_RANDOM_DATASETS; i++ {
-		_, payload := generateRandomTestDataPlainPacket()
-		ins = append(ins, bytes.NewBuffer(payload))
-	}
+	in := &devZero{}
 	pkt := NewPlainPacket()
 
 	b.ResetTimer()
 
 	for i := 0; i < b.N; i++ {
-		in := ins[i%len(ins)]
-		in.Reset()
 		if _, err := pkt.ReadFrom(in); err != nil {
 			b.Fatal("unexpected error:", err)
 		}
@@ -383,30 +386,25 @@ func BenchmarkPlainPacketWriteTo(b *testing.B) {
 		}
 		ins = append(ins, in)
 	}
+	out := &devZero{}
 
 	b.ResetTimer()
 
 	for i := 0; i < b.N; i++ {
 		in := ins[i%len(ins)]
-		if _, err := in.WriteTo(ioutil.Discard); err != nil {
+		if _, err := in.WriteTo(out); err != nil {
 			b.Fatal("unexpected error:", err)
 		}
 	}
 }
 
 func BenchmarkEncryptedPacketReadFrom(b *testing.B) {
-	ins := []*bytes.Buffer{}
-	for i := 0; i < NUM_RANDOM_DATASETS; i++ {
-		_, data := generateRandomTestDataEncryptedPacket()
-		ins = append(ins, bytes.NewBuffer(data))
-	}
+	in := &devZero{}
 	pkt := NewEncryptedPacket()
 
 	b.ResetTimer()
 
 	for i := 0; i < b.N; i++ {
-		in := ins[i%len(ins)]
-		in.Reset()
 		if _, err := pkt.ReadFrom(in); err != nil {
 			b.Fatal("unexpected error:", err)
 		}
@@ -423,12 +421,13 @@ func BenchmarkEncryptedPacketWriteTo(b *testing.B) {
 		}
 		ins = append(ins, in)
 	}
+	out := &devZero{}
 
 	b.ResetTimer()
 
 	for i := 0; i < b.N; i++ {
 		in := ins[i%len(ins)]
-		if _, err := in.WriteTo(ioutil.Discard); err != nil {
+		if _, err := in.WriteTo(out); err != nil {
 			b.Fatal("unexpected error:", err)
 		}
 	}