aboutsummaryrefslogtreecommitdiff
path: root/src/testing/testing.go
diff options
context:
space:
mode:
Diffstat (limited to 'src/testing/testing.go')
-rw-r--r--src/testing/testing.go49
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