mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-05 13:17:36 +03:00
Code Issues Projects Releases Packages Wiki Activity

improve the version negotiation integration test

Browse source
This commit is contained in:
Marten Seemann 2018-08-27 09:07:11 +07:00
parent c481645276
commit 46f8c07daa
4 changed files with 14 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

11
integrationtests/self/handshake_test.go
View file

@ -12,6 +12,10 @@ import (
. "github.com/onsi/gomega"
)
type versioner interface {
GetVersion() protocol.VersionNumber
}
var _ = Describe("Handshake tests", func() {
var (
server quic.Listener
@ -67,19 +71,22 @@ var _ = Describe("Handshake tests", func() {
// but it supports a bunch of versions that the client doesn't speak
serverConfig.Versions = []protocol.VersionNumber{protocol.SupportedVersions[1], 7, 8, 9}
runServer()
_, err := quic.DialAddr(server.Addr().String(), &tls.Config{InsecureSkipVerify: true}, nil)
sess, err := quic.DialAddr(server.Addr().String(), &tls.Config{InsecureSkipVerify: true}, nil)
Expect(err).ToNot(HaveOccurred())
Expect(sess.(versioner).GetVersion()).To(Equal(protocol.SupportedVersions[1]))
})
It("when the client supports more versions than the server supports", func() {
// the server doesn't support the highest supported version, which is the first one the client will try
// but it supports a bunch of versions that the client doesn't speak
serverConfig.Versions = supportedVersions
runServer()
conf := &quic.Config{
Versions: []protocol.VersionNumber{7, 8, 9, protocol.SupportedVersions[1], 10},
}
_, err := quic.DialAddr(server.Addr().String(), &tls.Config{InsecureSkipVerify: true}, conf)
sess, err := quic.DialAddr(server.Addr().String(), &tls.Config{InsecureSkipVerify: true}, conf)
Expect(err).ToNot(HaveOccurred())
Expect(sess.(versioner).GetVersion()).To(Equal(protocol.SupportedVersions[1]))
})
})

2
internal/protocol/version.go
View file

@ -41,7 +41,7 @@ func IsValidVersion(v VersionNumber) bool {
// UsesTLS says if this QUIC version uses TLS 1.3 for the handshake
func (vn VersionNumber) UsesTLS() bool {
return vn == VersionTLS
return !vn.isGQUIC()
}
func (vn VersionNumber) String() string {

6
internal/wire/header_test.go
View file

@ -267,7 +267,7 @@ var _ = Describe("Header", func() {
PacketNumber: 2,
PacketNumberLen: protocol.PacketNumberLen2,
}
err := hdr.Write(buf, protocol.PerspectiveServer, protocol.VersionWhatever)
err := hdr.Write(buf, protocol.PerspectiveServer, versionPublicHeader)
Expect(err).To(MatchError("PublicHeader: wrong length for Connection ID: 7 (expected 8)"))
})
@ -277,7 +277,7 @@ var _ = Describe("Header", func() {
DestConnectionID: connID,
PacketNumber: 2,
}
err := hdr.Write(buf, protocol.PerspectiveServer, protocol.VersionWhatever)
err := hdr.Write(buf, protocol.PerspectiveServer, versionPublicHeader)
Expect(err).To(MatchError("PublicHeader: PacketNumberLen not set"))
})
@ -286,7 +286,7 @@ var _ = Describe("Header", func() {
PacketNumberLen: protocol.PacketNumberLen1,
PacketNumber: 1,
}
err := hdr.Write(buf, protocol.PerspectiveServer, protocol.VersionWhatever)
err := hdr.Write(buf, protocol.PerspectiveServer, versionPublicHeader)
Expect(err).ToNot(HaveOccurred())
Expect(buf.Bytes()).To(Equal([]byte{0x0, 0x1}))
})

2
packet_handler_map_test.go
View file

@ -24,7 +24,7 @@ var _ = Describe("Packet Handler Map", func() {
err := (&wire.Header{
DestConnectionID: connID,
PacketNumberLen: protocol.PacketNumberLen1,
}).Write(buf, protocol.PerspectiveServer, protocol.VersionWhatever)
}).Write(buf, protocol.PerspectiveServer, versionGQUICFrames)
Expect(err).ToNot(HaveOccurred())
return buf.Bytes()
}