make wire an internal package

2017-08-29 23:51:56 +07:00
parent 1794636220
commit ed522ebbce
58 changed files with 32 additions and 31 deletions
--- a/internal/wire/goaway_frame_test.go
+++ b/internal/wire/goaway_frame_test.go
@@ -0,0 +1,140 @@
+package wire
+
+import (
+	"bytes"
+
+	"github.com/lucas-clemente/quic-go/protocol"
+	"github.com/lucas-clemente/quic-go/qerr"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+var _ = Describe("GoawayFrame", func() {
+	Context("when parsing", func() {
+		Context("in little endian", func() {
+			It("accepts sample frame", func() {
+				b := bytes.NewReader([]byte{0x3,
+					0x37, 0x13, 0x0, 0x0, // error code
+					0x34, 0x12, 0x0, 0x0, // last good stream id
+					0x3, 0x0, // reason phrase length
+					'f', 'o', 'o',
+				})
+				frame, err := ParseGoawayFrame(b, versionLittleEndian)
+				Expect(frame).To(Equal(&GoawayFrame{
+					ErrorCode:      0x1337,
+					LastGoodStream: 0x1234,
+					ReasonPhrase:   "foo",
+				}))
+				Expect(err).ToNot(HaveOccurred())
+				Expect(b.Len()).To(BeZero())
+			})
+
+			It("errors on EOFs", func() {
+				data := []byte{0x3,
+					0x1, 0x0, 0x0, 0x0, // error code
+					0x2, 0x0, 0x0, 0x0, // last good stream id
+					0x3, 0x0, // reason phrase length
+					'f', 'o', 'o',
+				}
+				_, err := ParseGoawayFrame(bytes.NewReader(data), versionLittleEndian)
+				Expect(err).NotTo(HaveOccurred())
+				for i := range data {
+					_, err := ParseGoawayFrame(bytes.NewReader(data[0:i]), versionLittleEndian)
+					Expect(err).To(HaveOccurred())
+				}
+			})
+		})
+
+		Context("in big endian", func() {
+			It("accepts sample frame", func() {
+				b := bytes.NewReader([]byte{0x3,
+					0x0, 0x0, 0x13, 0x37, // error code
+					0x0, 0x0, 0x12, 0x34, // last good stream id
+					0x0, 0x3, // reason phrase length
+					'f', 'o', 'o',
+				})
+				frame, err := ParseGoawayFrame(b, versionBigEndian)
+				Expect(frame).To(Equal(&GoawayFrame{
+					ErrorCode:      0x1337,
+					LastGoodStream: 0x1234,
+					ReasonPhrase:   "foo",
+				}))
+				Expect(err).ToNot(HaveOccurred())
+				Expect(b.Len()).To(BeZero())
+			})
+
+			It("errors on EOFs", func() {
+				data := []byte{0x3,
+					0x0, 0x0, 0x0, 0x1, // error code
+					0x0, 0x0, 0x0, 0x2, // last good stream id
+					0x0, 0x3, // reason phrase length
+					'f', 'o', 'o',
+				}
+				_, err := ParseGoawayFrame(bytes.NewReader(data), versionBigEndian)
+				Expect(err).NotTo(HaveOccurred())
+				for i := range data {
+					_, err := ParseGoawayFrame(bytes.NewReader(data[0:i]), versionBigEndian)
+					Expect(err).To(HaveOccurred())
+				}
+			})
+		})
+
+		It("rejects long reason phrases", func() {
+			b := bytes.NewReader([]byte{0x3,
+				0x1, 0x0, 0x0, 0x0, // error code
+				0x2, 0x0, 0x0, 0x0, // last good stream id
+				0xff, 0xff, // reason phrase length
+			})
+			_, err := ParseGoawayFrame(b, protocol.VersionWhatever)
+			Expect(err).To(MatchError(qerr.Error(qerr.InvalidGoawayData, "reason phrase too long")))
+		})
+	})
+
+	Context("when writing", func() {
+		Context("in little endian", func() {
+			It("writes a sample frame", func() {
+				b := &bytes.Buffer{}
+				frame := GoawayFrame{
+					ErrorCode:      0x1337,
+					LastGoodStream: 2,
+					ReasonPhrase:   "foo",
+				}
+				err := frame.Write(b, versionLittleEndian)
+				Expect(err).ToNot(HaveOccurred())
+				Expect(b.Bytes()).To(Equal([]byte{0x3,
+					0x37, 0x13, 0x0, 0x0, // error code
+					0x2, 0x0, 0x0, 0x0, // last good stream
+					0x3, 0x0, // reason phrase length
+					'f', 'o', 'o',
+				}))
+			})
+		})
+
+		Context("in big endian", func() {
+			It("writes a sample frame", func() {
+				b := &bytes.Buffer{}
+				frame := GoawayFrame{
+					ErrorCode:      0x1337,
+					LastGoodStream: 2,
+					ReasonPhrase:   "foo",
+				}
+				err := frame.Write(b, versionBigEndian)
+				Expect(err).ToNot(HaveOccurred())
+				Expect(b.Bytes()).To(Equal([]byte{0x3,
+					0x0, 0x0, 0x13, 0x37, // reason code
+					0x0, 0x0, 0x0, 0x2, // last good stream id
+					0x0, 0x3, // reason phrase length
+					'f', 'o', 'o',
+				}))
+			})
+		})
+
+		It("has the correct min length", func() {
+			frame := GoawayFrame{
+				ReasonPhrase: "foo",
+			}
+			Expect(frame.MinLength(0)).To(Equal(protocol.ByteCount(14)))
+		})
+	})
+})