You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
660 B
Go
46 lines
660 B
Go
package threading
|
|
|
|
import (
|
|
"io"
|
|
"log"
|
|
"sync"
|
|
"sync/atomic"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestRoutineGroupRun(t *testing.T) {
|
|
var count int32
|
|
group := NewRoutineGroup()
|
|
for i := 0; i < 3; i++ {
|
|
group.Run(func() {
|
|
atomic.AddInt32(&count, 1)
|
|
})
|
|
}
|
|
|
|
group.Wait()
|
|
|
|
assert.Equal(t, int32(3), count)
|
|
}
|
|
|
|
func TestRoutingGroupRunSafe(t *testing.T) {
|
|
log.SetOutput(io.Discard)
|
|
|
|
var count int32
|
|
group := NewRoutineGroup()
|
|
var once sync.Once
|
|
for i := 0; i < 3; i++ {
|
|
group.RunSafe(func() {
|
|
once.Do(func() {
|
|
panic("")
|
|
})
|
|
atomic.AddInt32(&count, 1)
|
|
})
|
|
}
|
|
|
|
group.Wait()
|
|
|
|
assert.Equal(t, int32(2), count)
|
|
}
|