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

add a buffer pool test for wrong-sized puts

Browse Source
This commit is contained in:
Lucas Clemente
2016-08-06 17:45:32 +02:00
parent 3e42d1da0f
commit 0696cd51bf
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
buffer_pool.go
View File

@@ -14,7 +14,6 @@ func getPacketBuffer() []byte {
func putPacketBuffer(buf []byte) { func putPacketBuffer(buf []byte) {
if cap(buf) != int(protocol.MaxPacketSize) { if cap(buf) != int(protocol.MaxPacketSize) {
println(buf)
panic("putPacketBuffer called with packet of wrong size!") panic("putPacketBuffer called with packet of wrong size!")
} }
bufferPool.Put(buf[:0]) bufferPool.Put(buf[:0])

6
buffer_pool_test.go
View File

@@ -23,4 +23,10 @@ var _ = Describe("Buffer Pool", func() {
Expect(buf).To(HaveCap(int(protocol.MaxPacketSize))) Expect(buf).To(HaveCap(int(protocol.MaxPacketSize)))
} }
}) })
It("panics if wrong-sized buffers are passed", func() {
Expect(func() {
putPacketBuffer([]byte{0})
}).To(Panic())
})
}) })