From f3323c21f9e53ee64d5a54f437055798af721fc4 Mon Sep 17 00:00:00 2001
From: Marten Seemann <martenseemann@gmail.com>
Date: Mon, 5 Feb 2018 15:29:47 +0800
Subject: [PATCH] add integration tests using many unidirectional streams

---
 integrationtests/self/stream_test.go     |   2 +-
 integrationtests/self/uni_stream_test.go | 132 +++++++++++++++++++++++
 2 files changed, 133 insertions(+), 1 deletion(-)
 create mode 100644 integrationtests/self/uni_stream_test.go

diff --git a/integrationtests/self/stream_test.go b/integrationtests/self/stream_test.go
index 58852d021..a7b3b565d 100644
--- a/integrationtests/self/stream_test.go
+++ b/integrationtests/self/stream_test.go
@@ -15,7 +15,7 @@ import (
 	. "github.com/onsi/gomega"
 )
 
-var _ = Describe("Stream tests", func() {
+var _ = Describe("Bidirectional streams", func() {
 	const numStreams = 300
 
 	var (
diff --git a/integrationtests/self/uni_stream_test.go b/integrationtests/self/uni_stream_test.go
new file mode 100644
index 000000000..562ab716f
--- /dev/null
+++ b/integrationtests/self/uni_stream_test.go
@@ -0,0 +1,132 @@
+package self_test
+
+import (
+	"fmt"
+	"io/ioutil"
+	"net"
+	"sync"
+
+	quic "github.com/lucas-clemente/quic-go"
+	"github.com/lucas-clemente/quic-go/integrationtests/tools/testserver"
+	"github.com/lucas-clemente/quic-go/internal/protocol"
+	"github.com/lucas-clemente/quic-go/internal/testdata"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+var _ = Describe("Unidirectional Streams", func() {
+	const numStreams = 500
+
+	var (
+		server     quic.Listener
+		serverAddr string
+		qconf      *quic.Config
+	)
+
+	BeforeEach(func() {
+		var err error
+		qconf = &quic.Config{Versions: []protocol.VersionNumber{protocol.VersionTLS}}
+		server, err = quic.ListenAddr("localhost:0", testdata.GetTLSConfig(), qconf)
+		Expect(err).ToNot(HaveOccurred())
+		serverAddr = fmt.Sprintf("quic.clemente.io:%d", server.Addr().(*net.UDPAddr).Port)
+	})
+
+	AfterEach(func() {
+		server.Close()
+	})
+
+	dataForStream := func(id protocol.StreamID) []byte {
+		return testserver.GeneratePRData(10 * int(id))
+	}
+
+	runSendingPeer := func(sess quic.Session) {
+		for i := 0; i < numStreams; i++ {
+			str, err := sess.OpenUniStreamSync()
+			Expect(err).ToNot(HaveOccurred())
+			go func() {
+				defer GinkgoRecover()
+				_, err := str.Write(dataForStream(str.StreamID()))
+				Expect(err).ToNot(HaveOccurred())
+				Expect(str.Close()).To(Succeed())
+			}()
+		}
+	}
+
+	runReceivingPeer := func(sess quic.Session) {
+		var wg sync.WaitGroup
+		wg.Add(numStreams)
+		for i := 0; i < numStreams; i++ {
+			str, err := sess.AcceptUniStream()
+			Expect(err).ToNot(HaveOccurred())
+			go func() {
+				defer GinkgoRecover()
+				defer wg.Done()
+				data, err := ioutil.ReadAll(str)
+				Expect(err).ToNot(HaveOccurred())
+				Expect(data).To(Equal(dataForStream(str.StreamID())))
+			}()
+		}
+		wg.Wait()
+	}
+
+	It(fmt.Sprintf("client opening %d streams to a server", numStreams), func() {
+		var sess quic.Session
+		go func() {
+			defer GinkgoRecover()
+			var err error
+			sess, err = server.Accept()
+			Expect(err).ToNot(HaveOccurred())
+			runReceivingPeer(sess)
+			sess.Close(nil)
+		}()
+
+		client, err := quic.DialAddr(serverAddr, nil, qconf)
+		Expect(err).ToNot(HaveOccurred())
+		runSendingPeer(client)
+		<-client.Context().Done()
+	})
+
+	It(fmt.Sprintf("server opening %d streams to a client", numStreams), func() {
+		go func() {
+			defer GinkgoRecover()
+			sess, err := server.Accept()
+			Expect(err).ToNot(HaveOccurred())
+			runSendingPeer(sess)
+		}()
+
+		client, err := quic.DialAddr(serverAddr, nil, qconf)
+		Expect(err).ToNot(HaveOccurred())
+		runReceivingPeer(client)
+	})
+
+	It(fmt.Sprintf("client and server opening %d streams each and sending data to the peer", numStreams), func() {
+		done1 := make(chan struct{})
+		go func() {
+			defer GinkgoRecover()
+			sess, err := server.Accept()
+			Expect(err).ToNot(HaveOccurred())
+			done := make(chan struct{})
+			go func() {
+				defer GinkgoRecover()
+				runReceivingPeer(sess)
+				close(done)
+			}()
+			runSendingPeer(sess)
+			<-done
+			close(done1)
+		}()
+
+		client, err := quic.DialAddr(serverAddr, nil, qconf)
+		Expect(err).ToNot(HaveOccurred())
+		done2 := make(chan struct{})
+		go func() {
+			defer GinkgoRecover()
+			runSendingPeer(client)
+			close(done2)
+		}()
+		runReceivingPeer(client)
+		<-done1
+		<-done2
+	})
+})