gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

send 5 kB of stream data in the handshake drop test

Browse Source
This commit is contained in:
Marten Seemann
2020-11-17 13:26:51 +07:00
parent 192bc8dc2a
commit 47c3cd8bb4
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
integrationtests/self/handshake_drop_test.go
View File

@@ -4,6 +4,7 @@ import (
"context" "context"
"crypto/tls" "crypto/tls"
"fmt" "fmt"
"io/ioutil"
mrand "math/rand" mrand "math/rand"
"net" "net"
"sync/atomic" "sync/atomic"
@@ -31,6 +32,7 @@ var _ = Describe("Handshake drop tests", func() {
ln quic.Listener ln quic.Listener
) )
data := GeneratePRData(5000)
const timeout = 2 * time.Minute const timeout = 2 * time.Minute
startListenerAndProxy := func(dropCallback quicproxy.DropCallback, doRetry bool, longCertChain bool, version protocol.VersionNumber) { startListenerAndProxy := func(dropCallback quicproxy.DropCallback, doRetry bool, longCertChain bool, version protocol.VersionNumber) {
@@ -77,10 +79,9 @@ var _ = Describe("Handshake drop tests", func() {
defer sess.CloseWithError(0, "") defer sess.CloseWithError(0, "")
str, err := sess.AcceptStream(context.Background()) str, err := sess.AcceptStream(context.Background())
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
b := make([]byte, 6) b, err := ioutil.ReadAll(gbytes.TimeoutReader(str, timeout))
_, err = gbytes.TimeoutReader(str, 10*time.Second).Read(b)
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
Expect(string(b)).To(Equal("foobar")) Expect(b).To(Equal(data))
serverSessionChan <- sess serverSessionChan <- sess
}() }()
sess, err := quic.DialAddr( sess, err := quic.DialAddr(
@@ -95,8 +96,9 @@ var _ = Describe("Handshake drop tests", func() {
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
str, err := sess.OpenStream() str, err := sess.OpenStream()
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
_, err = str.Write([]byte("foobar")) _, err = str.Write(data)
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
Expect(str.Close()).To(Succeed())
var serverSession quic.Session var serverSession quic.Session
Eventually(serverSessionChan, timeout).Should(Receive(&serverSession)) Eventually(serverSessionChan, timeout).Should(Receive(&serverSession))
@@ -115,8 +117,9 @@ var _ = Describe("Handshake drop tests", func() {
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
str, err := sess.OpenStream() str, err := sess.OpenStream()
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
_, err = str.Write([]byte("foobar")) _, err = str.Write(data)
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
Expect(str.Close()).To(Succeed())
serverSessionChan <- sess serverSessionChan <- sess
}() }()
sess, err := quic.DialAddr( sess, err := quic.DialAddr(
@@ -131,10 +134,9 @@ var _ = Describe("Handshake drop tests", func() {
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
str, err := sess.AcceptStream(context.Background()) str, err := sess.AcceptStream(context.Background())
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
b := make([]byte, 6) b, err := ioutil.ReadAll(gbytes.TimeoutReader(str, timeout))
_, err = gbytes.TimeoutReader(str, 10*time.Second).Read(b)
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
Expect(string(b)).To(Equal("foobar")) Expect(b).To(Equal(data))
var serverSession quic.Session var serverSession quic.Session
Eventually(serverSessionChan, timeout).Should(Receive(&serverSession)) Eventually(serverSessionChan, timeout).Should(Receive(&serverSession))