mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-24 03:47:45 +08:00
43 lines
782 B
Go
43 lines
782 B
Go
package util
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"gx/ipfs/QmZy2y8t9zQH2a1b8q2ZSLKp17ATuJoCNxxyMFG5qFExpt/go-net/context"
|
|
)
|
|
|
|
func TestDoReturnsContextErr(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
ch := make(chan struct{})
|
|
err := ContextDo(ctx, func() error {
|
|
cancel()
|
|
ch <- struct{}{} // won't return
|
|
return nil
|
|
})
|
|
if err != ctx.Err() {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestDoReturnsFuncError(t *testing.T) {
|
|
ctx := context.Background()
|
|
expected := errors.New("expected to be returned by ContextDo")
|
|
err := ContextDo(ctx, func() error {
|
|
return expected
|
|
})
|
|
if err != expected {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestDoReturnsNil(t *testing.T) {
|
|
ctx := context.Background()
|
|
err := ContextDo(ctx, func() error {
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
t.Fail()
|
|
}
|
|
}
|