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 })
44
View as plain text