362f447d9f
* cmd/statreceiver: lua-scriptable stat receiver Change-Id: I3ce0fe3f1ef4b1f4f27eed90bac0e91cfecf22d7 * some updates Change-Id: I7c3485adcda1278fce01ae077b4761b3ddb9fb7a * more comments Change-Id: I0bb22993cd934c3d40fc1da80d07e49e686b80dd * linter fixes Change-Id: Ied014304ecb9aadcf00a6b66ad28f856a428d150 * catch errors Change-Id: I6e1920f1fd941e66199b30bc427285c19769fc70 * review feedback Change-Id: I9d4051851eab18970c5f5ddcf4ff265508e541d3 * errorgroup improvements Change-Id: I4699dda3022f0485fbb50c9dafe692d3921734ff * too tricky the previous thing was better for memory with lots of errors at a time but https://play.golang.org/p/RweTMRjoSCt is too much of a foot gun Change-Id: I23f0b3d77dd4288fcc20b3756a7110359576bf44
61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package utils_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zeebo/errs"
|
|
|
|
"storj.io/storj/pkg/utils"
|
|
)
|
|
|
|
func TestCollectSingleError(t *testing.T) {
|
|
errchan := make(chan error)
|
|
defer close(errchan)
|
|
|
|
go func() {
|
|
errchan <- errs.New("error")
|
|
}()
|
|
|
|
err := utils.CollectErrors(errchan, 1*time.Second)
|
|
assert.Error(t, err)
|
|
assert.Equal(t, err.Error(), "error")
|
|
}
|
|
|
|
func TestCollecMultipleError(t *testing.T) {
|
|
errchan := make(chan error)
|
|
defer close(errchan)
|
|
|
|
go func() {
|
|
errchan <- errs.New("error1")
|
|
errchan <- errs.New("error2")
|
|
errchan <- errs.New("error3")
|
|
}()
|
|
|
|
err := utils.CollectErrors(errchan, 1*time.Second)
|
|
assert.Error(t, err)
|
|
assert.Equal(t, err.Error(), "error1\nerror2\nerror3")
|
|
}
|
|
|
|
func TestErrorGroup(t *testing.T) {
|
|
var errlist utils.ErrorGroup
|
|
errlist.Add(nil, nil, nil)
|
|
assert.NoError(t, errlist.Finish())
|
|
assert.Equal(t, len(errlist), 0)
|
|
e1 := errs.New("err1")
|
|
errlist.Add(nil, nil, e1, nil)
|
|
assert.Equal(t, errlist.Finish(), e1)
|
|
assert.Equal(t, len(errlist), 1)
|
|
e2, e3 := errs.New("err2"), errs.New("err3")
|
|
errlist.Add(e2, e3)
|
|
assert.Error(t, errlist.Finish())
|
|
assert.Equal(t, len(errlist), 3)
|
|
assert.Equal(t, errlist[0], e1)
|
|
assert.Equal(t, errlist[1], e2)
|
|
assert.Equal(t, errlist[2], e3)
|
|
}
|