initial import
This commit is contained in:
28
core/threading/workergroup_test.go
Normal file
28
core/threading/workergroup_test.go
Normal file
@@ -0,0 +1,28 @@
|
||||
package threading
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"runtime"
|
||||
"sync"
|
||||
"testing"
|
||||
|
||||
"zero/core/lang"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestWorkerGroup(t *testing.T) {
|
||||
m := make(map[string]lang.PlaceholderType)
|
||||
var lock sync.Mutex
|
||||
var wg sync.WaitGroup
|
||||
wg.Add(runtime.NumCPU())
|
||||
group := NewWorkerGroup(func() {
|
||||
lock.Lock()
|
||||
m[fmt.Sprint(RoutineId())] = lang.Placeholder
|
||||
lock.Unlock()
|
||||
wg.Done()
|
||||
}, runtime.NumCPU())
|
||||
go group.Start()
|
||||
wg.Wait()
|
||||
assert.Equal(t, runtime.NumCPU(), len(m))
|
||||
}
|
||||
Reference in New Issue
Block a user