mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-05 05:07:36 +03:00
Code Issues Projects Releases Packages Wiki 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())
})
}) })