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

add more tests for Dial in the h2quic client

Browse Source
This commit is contained in:
Marten Seemann
2017-05-12 21:56:04 +08:00
parent 9fad63ff50
commit 33a9b5aae4
2 changed files with 42 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
h2quic/client_test.go
View File

@@ -5,7 +5,6 @@ import (
"compress/gzip"
"crypto/tls"
"errors"
"net"
"net/http"
"golang.org/x/net/http2"
@@ -52,16 +51,45 @@ var _ = Describe("Client", func() {
})
It("dials", func() {
udpConn, err := net.ListenUDP("udp", &net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 0})
client = NewClient(quicTransport, nil, "localhost")
session.streamToOpen = &mockStream{id: 3}
client.dialAddr = func(hostname string, conf *quic.Config) (quic.Session, error) {
return session, nil
}
err := client.Dial()
Expect(err).ToNot(HaveOccurred())
client = NewClient(quicTransport, nil, udpConn.LocalAddr().String())
go client.Dial()
data := make([]byte, 100)
_, err = udpConn.Read(data)
hdr, err := quic.ParsePublicHeader(bytes.NewReader(data), protocol.PerspectiveClient)
Expect(err).ToNot(HaveOccurred())
Expect(hdr.VersionFlag).To(BeTrue())
Expect(hdr.ConnectionID).ToNot(BeNil())
Expect(client.session).To(Equal(session))
})
It("errors when dialing fails", func() {
testErr := errors.New("handshake error")
client = NewClient(quicTransport, nil, "localhost")
client.dialAddr = func(hostname string, conf *quic.Config) (quic.Session, error) {
return nil, testErr
}
err := client.Dial()
Expect(err).To(MatchError(testErr))
})
It("errors if the header stream has the wrong stream ID", func() {
client = NewClient(quicTransport, nil, "localhost")
session.streamToOpen = &mockStream{id: 2}
client.dialAddr = func(hostname string, conf *quic.Config) (quic.Session, error) {
return session, nil
}
err := client.Dial()
Expect(err).To(MatchError("h2quic Client BUG: StreamID of Header Stream is not 3"))
})
It("errors if it can't open a stream", func() {
testErr := errors.New("you shall not pass")
client = NewClient(quicTransport, nil, "localhost")
session.streamOpenErr = testErr
client.dialAddr = func(hostname string, conf *quic.Config) (quic.Session, error) {
return session, nil
}
err := client.Dial()
Expect(err).To(MatchError(testErr))
})
Context("Doing requests", func() {