1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-06-13 03:10:44 +00:00
v2fly/common/task/task_test.go
2018-05-27 13:02:29 +02:00

44 lines
862 B
Go

package task_test
import (
"context"
"errors"
"testing"
"time"
. "v2ray.com/core/common/task"
. "v2ray.com/ext/assert"
)
func TestExecuteParallel(t *testing.T) {
assert := With(t)
err := Run(Parallel(func() error {
time.Sleep(time.Millisecond * 200)
return errors.New("test")
}, func() error {
time.Sleep(time.Millisecond * 500)
return errors.New("test2")
}))()
assert(err.Error(), Equals, "test")
}
func TestExecuteParallelContextCancel(t *testing.T) {
assert := With(t)
ctx, cancel := context.WithCancel(context.Background())
err := Run(WithContext(ctx), Parallel(func() error {
time.Sleep(time.Millisecond * 2000)
return errors.New("test")
}, func() error {
time.Sleep(time.Millisecond * 5000)
return errors.New("test2")
}, func() error {
cancel()
return nil
}))()
assert(err.Error(), HasSubstring, "canceled")
}