-
Notifications
You must be signed in to change notification settings - Fork 38
/
Copy pathreusable_read_closer_test.go
60 lines (50 loc) · 1.25 KB
/
reusable_read_closer_test.go
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package reader
import (
"bytes"
"io"
"strings"
"sync"
"testing"
"github.com/stretchr/testify/require"
)
func TestReusableReader(t *testing.T) {
testcases := []interface{}{
strings.NewReader("test"),
bytes.NewBuffer([]byte("test")),
bytes.NewBufferString("test"),
bytes.NewReader([]byte("test")),
[]byte("test"),
"test",
}
for _, v := range testcases {
t.Run("sequential reuse", func(t *testing.T) {
reusableReader, err := NewReusableReadCloser(v)
require.Nil(t, err)
for i := 0; i < 100; i++ {
n, err := io.Copy(io.Discard, reusableReader)
require.Nil(t, err)
require.Positive(t, n)
bin, err := io.ReadAll(reusableReader)
require.Nil(t, err)
require.Len(t, bin, 4)
}
})
// todo: readers shouldn't be used concurrently, so here we just try to catch pontential read concurring with resets panics
t.Run("concurrent-reset-with-read", func(t *testing.T) {
reusableReader, err := NewReusableReadCloser(v)
require.Nil(t, err)
var wg sync.WaitGroup
for i := 0; i < 100; i++ {
wg.Add(1)
go func() {
defer wg.Done()
_, err := io.Copy(io.Discard, reusableReader)
require.Nil(t, err)
_, err = io.ReadAll(reusableReader)
require.Nil(t, err)
}()
}
wg.Wait()
})
}
}