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

append to a byte slice instead of a bytes.Buffer when serializing frames

Browse Source
This commit is contained in:
Marten Seemann
2022-08-28 23:05:07 +03:00
parent 65dd82ad90
commit 3ca1001951
50 changed files with 443 additions and 425 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
internal/wire/new_connection_id_frame_test.go
View File

@@ -77,15 +77,15 @@ var _ = Describe("NEW_CONNECTION_ID frame", func() {
ConnectionID: protocol.ConnectionID{1, 2, 3, 4, 5, 6},
StatelessResetToken: token,
}
b := &bytes.Buffer{}
Expect(frame.Write(b, protocol.Version1)).To(Succeed())
b, err := frame.Write(nil, protocol.Version1)
Expect(err).ToNot(HaveOccurred())
expected := []byte{0x18}
expected = append(expected, encodeVarInt(0x1337)...)
expected = append(expected, encodeVarInt(0x42)...)
expected = append(expected, 6)
expected = append(expected, []byte{1, 2, 3, 4, 5, 6}...)
expected = append(expected, token[:]...)
Expect(b.Bytes()).To(Equal(expected))
Expect(b).To(Equal(expected))
})
It("has the correct length", func() {
@@ -96,9 +96,9 @@ var _ = Describe("NEW_CONNECTION_ID frame", func() {
ConnectionID: protocol.ConnectionID{1, 2, 3, 4, 5, 6, 7, 8},
StatelessResetToken: token,
}
b := &bytes.Buffer{}
Expect(frame.Write(b, protocol.Version1)).To(Succeed())
Expect(frame.Length(protocol.Version1)).To(BeEquivalentTo(b.Len()))
b, err := frame.Write(nil, protocol.Version1)
Expect(err).ToNot(HaveOccurred())
Expect(b).To(HaveLen(int(frame.Length(protocol.Version1))))
})
})
})