[RICPLT-1832] - Rnib module changes + unit tests. After code review

[ric-plt/nodeb-rnib.git] / common / rNibPool_test.go

diff --git a/common/rNibPool_test.go b/common/rNibPool_test.go
index e7eb662..c8f0c00 100644 (file)
--- a/common/rNibPool_test.go
+++ b/common/rNibPool_test.go
@@ -68,21 +68,23 @@ func validateMaxLimit(size int, iterations int, t *testing.T) {
        )
        group := sync.WaitGroup{}
        for i := 0; i < iterations; i++ {
-               go getPutInstance(group)
+               go func() {
+                       group.Add(1)
+                       getPutInstance()
+                       group.Done()
+               }()
        }
        time.Sleep(time.Second)
        group.Wait()
        assert.Equal(t, int32(size), max)
 }
 
-func getPutInstance(group sync.WaitGroup) {
-       group.Add(1)
+func getPutInstance() {
        inst := poolGlob.Get().(instance)
        inst.up()
        time.Sleep(time.Millisecond*10)
        inst.down()
        poolGlob.Put(inst)
-       group.Done()
 }
 
 func TestNewPool(t *testing.T){