fix syncx/barrier test case (#123)

This commit is contained in:
codingfanlt
2020-10-13 19:29:20 +08:00
committed by GitHub
parent 7d4a548d29
commit 6f49639f80

View File

@@ -1,6 +1,7 @@
package syncx
import (
"sync"
"testing"
"github.com/stretchr/testify/assert"
@@ -10,11 +11,15 @@ func TestBarrier_Guard(t *testing.T) {
const total = 10000
var barrier Barrier
var count int
var wg sync.WaitGroup
wg.Add(total)
for i := 0; i < total; i++ {
barrier.Guard(func() {
go barrier.Guard(func() {
count++
wg.Done()
})
}
wg.Wait()
assert.Equal(t, total, count)
}
@@ -22,10 +27,14 @@ func TestBarrierPtr_Guard(t *testing.T) {
const total = 10000
barrier := new(Barrier)
var count int
wg := new(sync.WaitGroup)
wg.Add(total)
for i := 0; i < total; i++ {
barrier.Guard(func() {
go barrier.Guard(func() {
count++
wg.Done()
})
}
wg.Wait()
assert.Equal(t, total, count)
}