Refactor bufpool methods
This commit is contained in:
@@ -9,7 +9,7 @@ func TestRetrieveBufferOfRequestedSize(t *testing.T) {
|
||||
bpm := NewBufferPoolManager()
|
||||
|
||||
size := 1024
|
||||
buffer := bpm.GetBuffer(uint(size))
|
||||
buffer := bpm.Get(uint(size))
|
||||
|
||||
if buffer == nil {
|
||||
t.Fatalf("Expected buffer, got nil")
|
||||
@@ -24,7 +24,7 @@ func TestRequestBufferSizeZero(t *testing.T) {
|
||||
bpm := NewBufferPoolManager()
|
||||
|
||||
size := 0
|
||||
buffer := bpm.GetBuffer(uint(size))
|
||||
buffer := bpm.Get(uint(size))
|
||||
|
||||
if buffer == nil {
|
||||
t.Fatalf("Expected buffer, got nil")
|
||||
@@ -46,7 +46,7 @@ func TestConcurrentAccessToBufferPool(t *testing.T) {
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
buffer := bpm.GetBuffer(uint(size))
|
||||
buffer := bpm.Get(uint(size))
|
||||
if buffer == nil {
|
||||
t.Errorf("Expected buffer, got nil")
|
||||
}
|
||||
@@ -63,7 +63,7 @@ func TestGetBufferLockUnlock(t *testing.T) {
|
||||
bpm := NewBufferPoolManager()
|
||||
|
||||
size := 1024
|
||||
buffer := bpm.GetBuffer(uint(size))
|
||||
buffer := bpm.Get(uint(size))
|
||||
|
||||
if buffer == nil {
|
||||
t.Fatalf("Expected buffer, got nil")
|
||||
@@ -78,7 +78,7 @@ func TestVerifyPoolCreationForNewSizes(t *testing.T) {
|
||||
bpm := NewBufferPoolManager()
|
||||
|
||||
size := 512
|
||||
buffer := bpm.GetBuffer(uint(size))
|
||||
buffer := bpm.Get(uint(size))
|
||||
|
||||
if buffer == nil {
|
||||
t.Fatalf("Expected buffer, got nil")
|
||||
@@ -93,7 +93,7 @@ func TestBufferPoolManagerGetBuffer(t *testing.T) {
|
||||
bpm := NewBufferPoolManager()
|
||||
|
||||
size := 1024
|
||||
buffer := bpm.GetBuffer(uint(size))
|
||||
buffer := bpm.Get(uint(size))
|
||||
|
||||
if buffer == nil {
|
||||
t.Fatalf("Expected buffer, got nil")
|
||||
@@ -109,7 +109,7 @@ func TestGetBufferWithMultipleSizes(t *testing.T) {
|
||||
|
||||
sizes := []int{512, 1024, 2048}
|
||||
for _, size := range sizes {
|
||||
buffer := bpm.GetBuffer(uint(size))
|
||||
buffer := bpm.Get(uint(size))
|
||||
|
||||
if buffer == nil {
|
||||
t.Fatalf("Expected buffer for size %d, got nil", size)
|
||||
@@ -124,9 +124,9 @@ func TestGetBufferWithMultipleSizes(t *testing.T) {
|
||||
func TestGetBufferIsAlwaysZero(t *testing.T) {
|
||||
bpm := NewBufferPoolManager()
|
||||
|
||||
var size uint = 1024*64
|
||||
var size uint = 1024 * 64
|
||||
for i := 0; i < 1000; i++ {
|
||||
buffer := bpm.GetBuffer(size)
|
||||
buffer := bpm.Get(size)
|
||||
|
||||
if buffer == nil {
|
||||
t.Fatalf("Expected buffer for size %d, got nil", size)
|
||||
@@ -142,6 +142,6 @@ func TestGetBufferIsAlwaysZero(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
bpm.PutBuffer(buffer)
|
||||
bpm.Put(buffer)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user