1 files changed, 43 insertions, 0 deletions
diff --git a/vendor/github.com/onsi/gomega/gexec/prefixed_writer_test.go b/vendor/github.com/onsi/gomega/gexec/prefixed_writer_test.go
new file mode 100644
index 0000000..8657d0c
--- /dev/null
+++ b/vendor/github.com/onsi/gomega/gexec/prefixed_writer_test.go
@@ -0,0 +1,43 @@
+package gexec_test
+
+import (
+	"bytes"
+
+	. "github.com/onsi/gomega/gexec"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+var _ = Describe("PrefixedWriter", func() {
+	var buffer *bytes.Buffer
+	var writer *PrefixedWriter
+	BeforeEach(func() {
+		buffer = &bytes.Buffer{}
+		writer = NewPrefixedWriter("[p]", buffer)
+	})
+
+	It("should emit the prefix on newlines", func() {
+		writer.Write([]byte("abc"))
+		writer.Write([]byte("def\n"))
+		writer.Write([]byte("hij\n"))
+		writer.Write([]byte("\n\n"))
+		writer.Write([]byte("klm\n\nnop"))
+		writer.Write([]byte(""))
+		writer.Write([]byte("qrs"))
+		writer.Write([]byte("\ntuv\nwx"))
+		writer.Write([]byte("yz\n\n"))
+
+		Ω(buffer.String()).Should(Equal(`[p]abcdef
+[p]hij
+[p]
+[p]
+[p]klm
+[p]
+[p]nopqrs
+[p]tuv
+[p]wxyz
+[p]
+`))
+	})
+})