| 10 | ) |
| 11 | |
| 12 | func TestDiscardAfterClose(t *testing.T) { |
| 13 | t.Parallel() |
| 14 | exErr := xerrors.New("test") |
| 15 | fwc := &fakeWriteCloser{err: exErr} |
| 16 | uut := cliutil.DiscardAfterClose(fwc) |
| 17 | |
| 18 | n, err := uut.Write([]byte("one")) |
| 19 | require.Equal(t, 3, n) |
| 20 | require.NoError(t, err) |
| 21 | |
| 22 | n, err = uut.Write([]byte("two")) |
| 23 | require.Equal(t, 3, n) |
| 24 | require.NoError(t, err) |
| 25 | |
| 26 | err = uut.Close() |
| 27 | require.Equal(t, exErr, err) |
| 28 | |
| 29 | n, err = uut.Write([]byte("three")) |
| 30 | require.Equal(t, 5, n) |
| 31 | require.NoError(t, err) |
| 32 | |
| 33 | require.Len(t, fwc.writes, 2) |
| 34 | require.EqualValues(t, "one", fwc.writes[0]) |
| 35 | require.EqualValues(t, "two", fwc.writes[1]) |
| 36 | } |
| 37 | |
| 38 | type fakeWriteCloser struct { |
| 39 | writes [][]byte |