diff options
Diffstat (limited to 'src/testing/testing.go')
| -rw-r--r-- | src/testing/testing.go | 49 |
1 files changed, 44 insertions, 5 deletions
diff --git a/src/testing/testing.go b/src/testing/testing.go index bf95f1cfbb..6cce1a451f 100644 --- a/src/testing/testing.go +++ b/src/testing/testing.go @@ -562,11 +562,14 @@ func (f *chattyFlag) Get() any { return f.on } -const marker = byte(0x16) // ^V for framing +const ( + markFraming byte = 'V' &^ '@' // ^V: framing + markEscape byte = '[' &^ '@' // ^[: escape +) func (f *chattyFlag) prefix() string { if f.json { - return string(marker) + return string(markFraming) } return "" } @@ -588,7 +591,7 @@ func newChattyPrinter(w io.Writer) *chattyPrinter { // that as not in json mode (because it's not chatty at all). func (p *chattyPrinter) prefix() string { if p != nil && p.json { - return string(marker) + return string(markFraming) } return "" } @@ -869,8 +872,8 @@ func (w indenter) Write(b []byte) (n int, err error) { // An indent of 4 spaces will neatly align the dashes with the status // indicator of the parent. line := b[:end] - if line[0] == marker { - w.c.output = append(w.c.output, marker) + if line[0] == markFraming { + w.c.output = append(w.c.output, markFraming) line = line[1:] } w.c.output = append(w.c.output, indent...) @@ -1166,6 +1169,9 @@ func (o *outputWriter) Write(p []byte) (int, error) { // writeLine generates the output for a given line. func (o *outputWriter) writeLine(b []byte) { if !o.c.done && (o.c.chatty != nil) { + // Escape the framing marker. + b = escapeMarkers(b) + if o.c.bench { // Benchmarks don't print === CONT, so we should skip the test // printer and just print straight to stdout. @@ -1179,6 +1185,39 @@ func (o *outputWriter) writeLine(b []byte) { o.c.output = append(o.c.output, b...) } +func escapeMarkers(b []byte) []byte { + j := nextMark(b) + if j < 0 { + // Allocation-free fast path. + return b + } + + c := make([]byte, 0, len(b)+10) + i := 0 + for i < len(b) && j >= i { + if j > i { + c = append(c, b[i:j]...) + } + c = append(c, markEscape, b[j]) + i = j + 1 + j = i + nextMark(b[i:]) + } + if i < len(b) { + c = append(c, b[i:]...) + } + return c +} + +func nextMark(b []byte) int { + for i, b := range b { + switch b { + case markFraming, markEscape: + return i + } + } + return -1 +} + // Log formats its arguments using default formatting, analogous to [fmt.Println], // and records the text in the error log. For tests, the text will be printed only if // the test fails or the -test.v flag is set. For benchmarks, the text is always |
