Fix pool to uint from int
This commit is contained in:
@@ -7,12 +7,12 @@ import (
|
||||
|
||||
func TestRetrieveBufferOfRequestedSize(t *testing.T) {
|
||||
bpm := &BufferPoolManager{
|
||||
pools: make(map[int]*sync.Pool),
|
||||
pools: make(map[uint]*sync.Pool),
|
||||
mu: sync.Mutex{},
|
||||
}
|
||||
|
||||
size := 1024
|
||||
buffer := bpm.GetBuffer(size)
|
||||
buffer := bpm.GetBuffer(uint(size))
|
||||
|
||||
if buffer == nil {
|
||||
t.Fatalf("Expected buffer, got nil")
|
||||
@@ -25,12 +25,12 @@ func TestRetrieveBufferOfRequestedSize(t *testing.T) {
|
||||
|
||||
func TestRequestBufferSizeZero(t *testing.T) {
|
||||
bpm := &BufferPoolManager{
|
||||
pools: make(map[int]*sync.Pool),
|
||||
pools: make(map[uint]*sync.Pool),
|
||||
mu: sync.Mutex{},
|
||||
}
|
||||
|
||||
size := 0
|
||||
buffer := bpm.GetBuffer(size)
|
||||
buffer := bpm.GetBuffer(uint(size))
|
||||
|
||||
if buffer == nil {
|
||||
t.Fatalf("Expected buffer, got nil")
|
||||
@@ -43,7 +43,7 @@ func TestRequestBufferSizeZero(t *testing.T) {
|
||||
|
||||
func TestConcurrentAccessToBufferPool(t *testing.T) {
|
||||
bpm := &BufferPoolManager{
|
||||
pools: make(map[int]*sync.Pool),
|
||||
pools: make(map[uint]*sync.Pool),
|
||||
mu: sync.Mutex{},
|
||||
}
|
||||
|
||||
@@ -55,7 +55,7 @@ func TestConcurrentAccessToBufferPool(t *testing.T) {
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
buffer := bpm.GetBuffer(size)
|
||||
buffer := bpm.GetBuffer(uint(size))
|
||||
if buffer == nil {
|
||||
t.Errorf("Expected buffer, got nil")
|
||||
}
|
||||
@@ -70,12 +70,12 @@ func TestConcurrentAccessToBufferPool(t *testing.T) {
|
||||
|
||||
func TestGetBufferLockUnlock(t *testing.T) {
|
||||
bpm := &BufferPoolManager{
|
||||
pools: make(map[int]*sync.Pool),
|
||||
pools: make(map[uint]*sync.Pool),
|
||||
mu: sync.Mutex{},
|
||||
}
|
||||
|
||||
size := 1024
|
||||
buffer := bpm.GetBuffer(size)
|
||||
buffer := bpm.GetBuffer(uint(size))
|
||||
|
||||
if buffer == nil {
|
||||
t.Fatalf("Expected buffer, got nil")
|
||||
@@ -88,12 +88,12 @@ func TestGetBufferLockUnlock(t *testing.T) {
|
||||
|
||||
func TestVerifyPoolCreationForNewSizes(t *testing.T) {
|
||||
bpm := &BufferPoolManager{
|
||||
pools: make(map[int]*sync.Pool),
|
||||
pools: make(map[uint]*sync.Pool),
|
||||
mu: sync.Mutex{},
|
||||
}
|
||||
|
||||
size := 512
|
||||
buffer := bpm.GetBuffer(size)
|
||||
buffer := bpm.GetBuffer(uint(size))
|
||||
|
||||
if buffer == nil {
|
||||
t.Fatalf("Expected buffer, got nil")
|
||||
@@ -106,12 +106,12 @@ func TestVerifyPoolCreationForNewSizes(t *testing.T) {
|
||||
|
||||
func TestBufferPoolManagerGetBuffer(t *testing.T) {
|
||||
bpm := &BufferPoolManager{
|
||||
pools: make(map[int]*sync.Pool),
|
||||
pools: make(map[uint]*sync.Pool),
|
||||
mu: sync.Mutex{},
|
||||
}
|
||||
|
||||
size := 1024
|
||||
buffer := bpm.GetBuffer(size)
|
||||
buffer := bpm.GetBuffer(uint(size))
|
||||
|
||||
if buffer == nil {
|
||||
t.Fatalf("Expected buffer, got nil")
|
||||
@@ -124,13 +124,13 @@ func TestBufferPoolManagerGetBuffer(t *testing.T) {
|
||||
|
||||
func TestGetBufferWithMultipleSizes(t *testing.T) {
|
||||
bpm := &BufferPoolManager{
|
||||
pools: make(map[int]*sync.Pool),
|
||||
pools: make(map[uint]*sync.Pool),
|
||||
mu: sync.Mutex{},
|
||||
}
|
||||
|
||||
sizes := []int{512, 1024, 2048}
|
||||
for _, size := range sizes {
|
||||
buffer := bpm.GetBuffer(size)
|
||||
buffer := bpm.GetBuffer(uint(size))
|
||||
|
||||
if buffer == nil {
|
||||
t.Fatalf("Expected buffer for size %d, got nil", size)
|
||||
|
||||
Reference in New Issue
Block a user