Merge pull request #22 from spiral/feature/stderr

Feature/stderr

1 files changed, 54 insertions, 3 deletions

diff --git a/error_buffer_test.go b/error_buffer_test.go
index afbc80e2..09ea4f03 100644
--- a/error_buffer_test.go
+++ b/error_buffer_test.go
@@ -1,14 +1,65 @@
 package roadrunner
 
 import (
-	"bytes"
 	"github.com/stretchr/testify/assert"
 	"testing"
 )
 
 func TestErrBuffer_Write_Len(t *testing.T) {
-	buf := &errBuffer{buffer: new(bytes.Buffer)}
+	buf := newErrBuffer()
+	defer buf.Close()
+
 	buf.Write([]byte("hello"))
 	assert.Equal(t, 5, buf.Len())
-	assert.Equal(t, buf.String(), "hello")
+	assert.Equal(t, "hello", buf.String())
+}
+
+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"), ctx)
+		close(tr)
+	})
+
+	buf.Write([]byte("hello\n"))
+
+	<-tr
+
+	// messages are read
+	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\nending"), ctx)
+		close(tr)
+	})
+
+	buf.Write([]byte("hel"))
+	buf.Write([]byte("lo\n"))
+	buf.Write([]byte("ending"))
+
+	<-tr
+	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"))
+
+	assert.Equal(t, 3, buf.Len())
+	assert.Equal(t, "hel", buf.String())
 }