vendor/github.com/onsi/gomega/gexec/prefixed_writer_test.go

   1 package gexec_test
   2
   3 import (
   4         "bytes"
   5
   6         . "github.com/onsi/gomega/gexec"
   7
   8         . "github.com/onsi/ginkgo"
   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                 Ω(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 })