Making sure async tests properly generate random data (#169)

loopholelabs · Mar 26, 2024 · 6e3ba6f · 6e3ba6f
1 parent 51a4026
commit 6e3ba6f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/helpers_test.go b/helpers_test.go
@@ -17,6 +17,8 @@
 package frisbee
 
 import (
+	"crypto/rand"
+	"github.com/stretchr/testify/require"
 	"testing"
 
 	"github.com/loopholelabs/frisbee-go/pkg/packet"
@@ -31,9 +33,10 @@ func throughputRunner(testSize, packetSize uint32, readerConn, writerConn Conn)
 	return func(b *testing.B) {
 		b.SetBytes(int64(testSize * packetSize))
 		b.ReportAllocs()
-		var err error
 
 		randomData := make([]byte, packetSize)
+		_, err := rand.Read(randomData)
+		require.NoError(b, err)
 
 		p := packet.Get()
 		p.Metadata.Id = 64