error aggregation

1 files changed, 12 insertions, 8 deletions

diff --git a/error_buffer_test.go b/error_buffer_test.go
index 2bf98915..09ea4f03 100644
--- a/error_buffer_test.go
+++ b/error_buffer_test.go
@@ -7,6 +7,7 @@ import (
 
 func TestErrBuffer_Write_Len(t *testing.T) {
 	buf := newErrBuffer()
+	defer buf.Close()
 
 	buf.Write([]byte("hello"))
 	assert.Equal(t, 5, buf.Len())
@@ -15,44 +16,47 @@ func TestErrBuffer_Write_Len(t *testing.T) {
 
 func TestErrBuffer_Write_Event(t *testing.T) {
 	buf := newErrBuffer()
+	defer buf.Close()
 
 	tr := make(chan interface{})
 	buf.Listen(func(event int, ctx interface{}) {
 		assert.Equal(t, EventStderrOutput, event)
-		assert.Equal(t, []byte("hello\n\n"), ctx)
+		assert.Equal(t, []byte("hello\n"), ctx)
 		close(tr)
 	})
 
-	buf.Write([]byte("hello\n\n"))
+	buf.Write([]byte("hello\n"))
 
 	<-tr
 
 	// messages are read
-	assert.Equal(t, 0, buf.Len())
-	assert.Equal(t, "", buf.String())
+	assert.Equal(t, 6, buf.Len())
+	assert.Equal(t, "hello\n", buf.String())
 }
 
 func TestErrBuffer_Write_Event_Separated(t *testing.T) {
 	buf := newErrBuffer()
+	defer buf.Close()
 
 	tr := make(chan interface{})
 	buf.Listen(func(event int, ctx interface{}) {
 		assert.Equal(t, EventStderrOutput, event)
-		assert.Equal(t, []byte("hello\n\n"), ctx)
+		assert.Equal(t, []byte("hello\nending"), ctx)
 		close(tr)
 	})
 
 	buf.Write([]byte("hel"))
-	buf.Write([]byte("lo\n\n"))
+	buf.Write([]byte("lo\n"))
 	buf.Write([]byte("ending"))
 
 	<-tr
-	assert.Equal(t, 6, buf.Len())
-	assert.Equal(t, "ending", buf.String())
+	assert.Equal(t, 12, buf.Len())
+	assert.Equal(t, "hello\nending", buf.String())
 }
 
 func TestErrBuffer_Write_Remaining(t *testing.T) {
 	buf := newErrBuffer()
+	defer buf.Close()
 
 	buf.Write([]byte("hel"))