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

add a function to write short headers

Browse Source
This commit is contained in:
Marten Seemann
2022-12-26 10:13:12 +13:00
parent 87f51d33dc
commit 124e597cf9
4 changed files with 45 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
internal/wire/short_header_test.go
View File

@@ -75,6 +75,20 @@ var _ = Describe("Short Header", func() {
Expect(ShortHeaderLen(protocol.ParseConnectionID([]byte{}), protocol.PacketNumberLen1)).To(BeEquivalentTo(2))
})
Context("writing", func() {
It("writes a short header packet", func() {
b := &bytes.Buffer{}
connID := protocol.ParseConnectionID([]byte{1, 2, 3, 4})
Expect(WriteShortHeader(b, connID, 1337, 4, protocol.KeyPhaseOne)).To(Succeed())
l, pn, pnLen, kp, err := ParseShortHeader(b.Bytes(), 4)
Expect(err).ToNot(HaveOccurred())
Expect(pn).To(Equal(protocol.PacketNumber(1337)))
Expect(pnLen).To(Equal(protocol.PacketNumberLen4))
Expect(kp).To(Equal(protocol.KeyPhaseOne))
Expect(l).To(Equal(b.Len()))
})
})
Context("logging", func() {
var (
buf *bytes.Buffer