github.com/onsi/gomega@v1.32.0/gexec/prefixed_writer_test.go (about)

     1  package gexec_test
     2  
     3  import (
     4  	"bytes"
     5  
     6  	. "github.com/onsi/gomega/gexec"
     7  
     8  	. "github.com/onsi/ginkgo/v2"
     9  	. "github.com/onsi/gomega"
    10  )
    11  
    12  var _ = Describe("PrefixedWriter", func() {
    13  	var buffer *bytes.Buffer
    14  	var writer *PrefixedWriter
    15  	BeforeEach(func() {
    16  		buffer = &bytes.Buffer{}
    17  		writer = NewPrefixedWriter("[p]", buffer)
    18  	})
    19  
    20  	It("should emit the prefix on newlines", func() {
    21  		writer.Write([]byte("abc"))
    22  		writer.Write([]byte("def\n"))
    23  		writer.Write([]byte("hij\n"))
    24  		writer.Write([]byte("\n\n"))
    25  		writer.Write([]byte("klm\n\nnop"))
    26  		writer.Write([]byte(""))
    27  		writer.Write([]byte("qrs"))
    28  		writer.Write([]byte("\ntuv\nwx"))
    29  		writer.Write([]byte("yz\n\n"))
    30  
    31  		Expect(buffer.String()).Should(Equal(`[p]abcdef
    32  [p]hij
    33  [p]
    34  [p]
    35  [p]klm
    36  [p]
    37  [p]nopqrs
    38  [p]tuv
    39  [p]wxyz
    40  [p]
    41  `))
    42  	})
    43  })