58 lines
1.5 KiB
Go
58 lines
1.5 KiB
Go
package mwc_test
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"git.wntrmute.dev/kyle/goutils/assert"
|
|
"git.wntrmute.dev/kyle/goutils/mwc"
|
|
"git.wntrmute.dev/kyle/goutils/testio"
|
|
)
|
|
|
|
func TestMWC(t *testing.T) {
|
|
buf1 := testio.NewBufCloser(nil)
|
|
buf2 := testio.NewBufCloser(nil)
|
|
|
|
mwc := mwc.MultiWriteCloser(buf1, buf2)
|
|
|
|
_, err := mwc.Write([]byte("hello, world"))
|
|
assert.NoErrorT(t, err)
|
|
|
|
assert.BoolT(t, bytes.Equal(buf1.Bytes(), buf2.Bytes()), "write failed")
|
|
assert.BoolT(t, bytes.Equal(buf1.Bytes(), []byte("hello, world")), "write failed")
|
|
|
|
err = mwc.Close()
|
|
assert.NoErrorT(t, err)
|
|
}
|
|
|
|
func TestMWCShort(t *testing.T) {
|
|
buf1 := testio.NewBufCloser(nil)
|
|
buf2 := testio.NewBufCloser(nil)
|
|
buf3 := testio.NewBrokenWriter(5)
|
|
buf4 := testio.NewSilentBrokenWriter(5)
|
|
|
|
multiWriter := mwc.MultiWriteCloser(buf1, buf2, buf3)
|
|
defer multiWriter.Close()
|
|
|
|
_, err := multiWriter.Write([]byte("hello, world"))
|
|
assert.ErrorT(t, err, "expected a short write error", "but no error occurred")
|
|
multiWriter.Close()
|
|
|
|
multiWriter = mwc.MultiWriteCloser(buf1, buf2, buf4)
|
|
_, err = multiWriter.Write([]byte("hello, world"))
|
|
assert.ErrorT(t, err, "expected a short write error", "but no error occurred")
|
|
}
|
|
|
|
func TestMWCClose(t *testing.T) {
|
|
buf1 := testio.NewBufCloser(nil)
|
|
buf2 := testio.NewBufCloser(nil)
|
|
buf3 := testio.NewBrokenCloser(nil)
|
|
|
|
mwc := mwc.MultiWriteCloser(buf1, buf2, buf3)
|
|
_, err := mwc.Write([]byte("hello, world"))
|
|
assert.NoErrorT(t, err)
|
|
|
|
err = mwc.Close()
|
|
assert.ErrorT(t, err, "expected broken closer to fail")
|
|
}
|