better tests for PublicHeader

This commit is contained in:
Marten Seemann
2016-05-06 21:15:57 +07:00
parent a1f6289587
commit 965addac79
2 changed files with 20 additions and 5 deletions

View File

@@ -26,6 +26,14 @@ var _ = Describe("Public Header", func() {
b := bytes.NewReader([]byte{0x00, 0x01})
_, err := ParsePublicHeader(b)
Expect(err).To(HaveOccurred())
Expect(err).To(Equal(errReceivedTruncatedConnectionID))
})
It("rejects 0 as a connection ID", func() {
b := bytes.NewReader([]byte{0x09, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x51, 0x30, 0x33, 0x30, 0x01})
_, err := ParsePublicHeader(b)
Expect(err).To(HaveOccurred())
Expect(err).To(Equal(errInvalidConnectionID))
})
It("accepts 1-byte packet numbers", func() {
@@ -115,6 +123,7 @@ var _ = Describe("Public Header", func() {
}
err := publicHeader.WritePublicHeader(b)
Expect(err).To(HaveOccurred())
Expect(err).To(Equal(errResetAndVersionFlagSet))
})
})
})