148 lines
2.9 KiB
Go
148 lines
2.9 KiB
Go
package pdu
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
)
|
|
|
|
func TestRetrieveBufferOfRequestedSize(t *testing.T) {
|
|
bpm := NewBufferPoolManager()
|
|
|
|
size := 1024
|
|
buffer := bpm.Get(uint(size))
|
|
|
|
if buffer == nil {
|
|
t.Fatalf("Expected buffer, got nil")
|
|
}
|
|
|
|
if len(*buffer) != size {
|
|
t.Errorf("Expected buffer size %d, got %d", size, len(*buffer))
|
|
}
|
|
}
|
|
|
|
func TestRequestBufferSizeZero(t *testing.T) {
|
|
bpm := NewBufferPoolManager()
|
|
|
|
size := 0
|
|
buffer := bpm.Get(uint(size))
|
|
|
|
if buffer == nil {
|
|
t.Fatalf("Expected buffer, got nil")
|
|
}
|
|
|
|
if len(*buffer) != size {
|
|
t.Errorf("Expected buffer size %d, got %d", size, len(*buffer))
|
|
}
|
|
}
|
|
|
|
func TestConcurrentAccessToBufferPool(t *testing.T) {
|
|
bpm := NewBufferPoolManager()
|
|
|
|
size := 1024
|
|
var wg sync.WaitGroup
|
|
numRoutines := 100
|
|
|
|
for i := 0; i < numRoutines; i++ {
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
buffer := bpm.Get(uint(size))
|
|
if buffer == nil {
|
|
t.Errorf("Expected buffer, got nil")
|
|
}
|
|
if len(*buffer) != size {
|
|
t.Errorf("Expected buffer size %d, got %d", size, len(*buffer))
|
|
}
|
|
}()
|
|
}
|
|
|
|
wg.Wait()
|
|
}
|
|
|
|
func TestGetBufferLockUnlock(t *testing.T) {
|
|
bpm := NewBufferPoolManager()
|
|
|
|
size := 1024
|
|
buffer := bpm.Get(uint(size))
|
|
|
|
if buffer == nil {
|
|
t.Fatalf("Expected buffer, got nil")
|
|
}
|
|
|
|
if len(*buffer) != size {
|
|
t.Errorf("Expected buffer size %d, got %d", size, len(*buffer))
|
|
}
|
|
}
|
|
|
|
func TestVerifyPoolCreationForNewSizes(t *testing.T) {
|
|
bpm := NewBufferPoolManager()
|
|
|
|
size := 512
|
|
buffer := bpm.Get(uint(size))
|
|
|
|
if buffer == nil {
|
|
t.Fatalf("Expected buffer, got nil")
|
|
}
|
|
|
|
if len(*buffer) != size {
|
|
t.Errorf("Expected buffer size %d, got %d", size, len(*buffer))
|
|
}
|
|
}
|
|
|
|
func TestBufferPoolManagerGetBuffer(t *testing.T) {
|
|
bpm := NewBufferPoolManager()
|
|
|
|
size := 1024
|
|
buffer := bpm.Get(uint(size))
|
|
|
|
if buffer == nil {
|
|
t.Fatalf("Expected buffer, got nil")
|
|
}
|
|
|
|
if len(*buffer) != size {
|
|
t.Errorf("Expected buffer size %d, got %d", size, len(*buffer))
|
|
}
|
|
}
|
|
|
|
func TestGetBufferWithMultipleSizes(t *testing.T) {
|
|
bpm := NewBufferPoolManager()
|
|
|
|
sizes := []int{512, 1024, 2048}
|
|
for _, size := range sizes {
|
|
buffer := bpm.Get(uint(size))
|
|
|
|
if buffer == nil {
|
|
t.Fatalf("Expected buffer for size %d, got nil", size)
|
|
}
|
|
|
|
if len(*buffer) != size {
|
|
t.Errorf("Expected buffer size %d, got %d", size, len(*buffer))
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestGetBufferIsAlwaysZero(t *testing.T) {
|
|
bpm := NewBufferPoolManager()
|
|
|
|
var size uint = 1024 * 64
|
|
for i := 0; i < 1000; i++ {
|
|
buffer := bpm.Get(size)
|
|
|
|
if buffer == nil {
|
|
t.Fatalf("Expected buffer for size %d, got nil", size)
|
|
}
|
|
|
|
if uint(len(*buffer)) != size {
|
|
t.Errorf("Expected buffer size %d, got %d", size, len(*buffer))
|
|
}
|
|
|
|
for _, b := range *buffer {
|
|
if b != 0 {
|
|
t.Errorf("Expected buffer to be zero, got %d", b)
|
|
}
|
|
}
|
|
|
|
bpm.Put(buffer)
|
|
}
|
|
}
|