1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
package ioutils
import (
"bytes"
"context"
"testing"
"time"
)
type blockingReader struct{}
func (r blockingReader) Read(p []byte) (int, error) {
time.Sleep(time.Second)
return 0, nil
}
func TestCopyCtx(t *testing.T) {
ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond*5)
defer cancel()
dst := new(bytes.Buffer)
finished := make(chan struct{})
go func() {
CopyCtx(ctx, dst, blockingReader{})
close(finished)
}()
select {
case <-finished:
case <-time.After(time.Millisecond * 100):
t.Fatal("CopyCtx did not return after context was cancelled")
}
}
|
