Add unordered benchmark for RecvDataFromRemote

internal/multiplex/session_test.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"github.com/cbeuw/connutil"
 	"github.com/stretchr/testify/assert"
+	"io/ioutil"
 	"math/rand"
 	"strconv"
 	"sync"
@@ -415,7 +416,7 @@ func TestSession_timeoutAfter(t *testing.T) {
 	}
 }
 
-func BenchmarkRecvDataFromRemote_Ordered(b *testing.B) {
+func BenchmarkRecvDataFromRemote(b *testing.B) {
 	testPayload := make([]byte, testPayloadLen)
 	rand.Read(testPayload)
 	f := &Frame{
@@ -439,11 +440,17 @@ func BenchmarkRecvDataFromRemote_Ordered(b *testing.B) {
 	for name, ep := range table {
 		ep := ep
 		b.Run(name, func(b *testing.B) {
-			seshConfig := seshConfigs["ordered"]
+			for seshType, seshConfig := range seshConfigs {
+				b.Run(seshType, func(b *testing.B) {
 					obfuscator, _ := MakeObfuscator(ep, sessionKey)
 					seshConfig.Obfuscator = obfuscator
 					sesh := MakeSession(0, seshConfig)
 
+					go func() {
+						stream, _ := sesh.Accept()
+						stream.(*Stream).WriteTo(ioutil.Discard)
+					}()
+
 					binaryFrames := [maxIter][]byte{}
 					for i := 0; i < maxIter; i++ {
 						obfsBuf := make([]byte, obfsBufLen)
@@ -459,6 +466,8 @@ func BenchmarkRecvDataFromRemote_Ordered(b *testing.B) {
 					}
 				})
 			}
+		})
+	}
 }
 
 func BenchmarkMultiStreamWrite(b *testing.B) {