Refactor bufpool methods

This commit is contained in:
PhatPhuckDave
2024-07-23 14:55:48 +02:00
parent a9827f10ce
commit 2759185dbb
4 changed files with 15 additions and 15 deletions

View File

@@ -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)
}
}