diff options
Diffstat (limited to 'streamio/stream_test.go')
-rw-r--r-- | streamio/stream_test.go | 53 |
1 files changed, 0 insertions, 53 deletions
diff --git a/streamio/stream_test.go b/streamio/stream_test.go index ee45f7196..2e879ab55 100644 --- a/streamio/stream_test.go +++ b/streamio/stream_test.go @@ -61,29 +61,6 @@ func TestReadSizes(t *testing.T) { }) } -func TestWriterTo(t *testing.T) { - testData := "Hello this is the test data that will be received. It goes on for a while bla bla bla." - testCases := []struct { - desc string - r io.Reader - }{ - {desc: "base", r: strings.NewReader(testData)}, - {desc: "dataerr", r: iotest.DataErrReader(strings.NewReader(testData))}, - {desc: "onebyte", r: iotest.OneByteReader(strings.NewReader(testData))}, - {desc: "dataerr(onebyte)", r: iotest.DataErrReader(iotest.OneByteReader(strings.NewReader(testData)))}, - } - - for _, tc := range testCases { - result := &bytes.Buffer{} - reader := NewReader(receiverFromReader(tc.r)) - n, err := reader.(io.WriterTo).WriteTo(result) - - require.NoError(t, err, tc.desc) - require.Equal(t, int64(len(testData)), n, tc.desc) - require.Equal(t, testData, result.String(), tc.desc) - } -} - func receiverFromReader(r io.Reader) func() ([]byte, error) { return func() ([]byte, error) { data := make([]byte, 10) @@ -151,33 +128,3 @@ func (ts *testSender) send(p []byte) error { ts.sends = append(ts.sends, buf) return nil } - -func TestReadFrom(t *testing.T) { - defer func(oldBufferSize int) { - WriteBufferSize = oldBufferSize - }(WriteBufferSize) - WriteBufferSize = 5 - - testData := "Hello this is the test data that will be received. It goes on for a while bla bla bla." - testCases := []struct { - desc string - r io.Reader - }{ - {desc: "base", r: strings.NewReader(testData)}, - {desc: "dataerr", r: iotest.DataErrReader(strings.NewReader(testData))}, - {desc: "onebyte", r: iotest.OneByteReader(strings.NewReader(testData))}, - {desc: "dataerr(onebyte)", r: iotest.DataErrReader(iotest.OneByteReader(strings.NewReader(testData)))}, - } - - for _, tc := range testCases { - ts := &testSender{} - n, err := NewWriter(ts.send).(io.ReaderFrom).ReadFrom(tc.r) - - require.NoError(t, err, tc.desc) - require.Equal(t, int64(len(testData)), n, tc.desc) - require.Equal(t, testData, string(bytes.Join(ts.sends, nil)), tc.desc) - for _, send := range ts.sends { - require.True(t, len(send) <= WriteBufferSize, "send calls may not exceed WriteBufferSize") - } - } -} |