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
|
package utils
import (
"bufio"
"bytes"
"testing"
"github.com/stretchr/testify/require"
)
type nopCloser struct{}
func (nopCloser) Close() error { return nil }
func TestBufferedWriteCloserFlushBeforeClosing(t *testing.T) {
buf := &bytes.Buffer{}
w := bufio.NewWriter(buf)
wc := NewBufferedWriteCloser(w, &nopCloser{})
_, err := wc.Write([]byte("foobar"))
require.NoError(t, err)
require.Zero(t, buf.Len())
require.NoError(t, wc.Close())
require.Equal(t, "foobar", buf.String())
}
|
