Extract appending to output from Write

Author: suntala

src/testing/testing.go

@@ -1054,34 +1054,6 @@ func (o *outputWriter) Write(p []byte) (int, error) {
 	o.c.mu.Lock()
 	defer o.c.mu.Unlock()
 
-	doWrite := func(s string) {
-		if o.c.done {
-			// This test has already finished. Try and log this message
-			// with our parent. If we don't have a parent, panic.
-			for parent := o.c.parent; parent != nil; parent = parent.parent {
-				parent.mu.Lock()
-				defer parent.mu.Unlock()
-				if !parent.done {
-					parent.output = append(parent.output, s...)
-					return
-				}
-			}
-			panic("Log in goroutine after " + o.c.name + " has completed: " + string(p))
-		} else {
-			if o.c.chatty != nil {
-				if o.c.bench {
-					// Benchmarks don't print === CONT, so we should skip the test
-					// printer and just print straight to stdout.
-					fmt.Print(s)
-				} else {
-					o.c.chatty.Printf(o.c.name, "%s", s)
-				}
-			} else {
-				o.c.output = append(o.c.output, s...)
-			}
-		}
-	}
-
 	str := string(o.b)
 	lines := strings.Split(str, "\n")
 	for i, line := range lines {
@@ -1091,7 +1063,7 @@ func (o *outputWriter) Write(p []byte) (int, error) {
 			if line != "" {
 				o.b = []byte(line)
 				if i > 0 {
-					doWrite("\n")
+					o.doWrite("\n", p)
 				}
 				break
 			}
@@ -1112,11 +1084,39 @@ func (o *outputWriter) Write(p []byte) (int, error) {
 
 		buf.WriteString(line)
 
-		doWrite(buf.String())
+		o.doWrite(buf.String(), p)
 	}
 	return len(p), nil
 }
 
+func (o *outputWriter) doWrite(s string, p []byte) {
+	if o.c.done {
+		// This test has already finished. Try and log this message
+		// with our parent. If we don't have a parent, panic.
+		for parent := o.c.parent; parent != nil; parent = parent.parent {
+			parent.mu.Lock()
+			defer parent.mu.Unlock()
+			if !parent.done {
+				parent.output = append(parent.output, s...)
+				return
+			}
+		}
+		panic("Log in goroutine after " + o.c.name + " has completed: " + string(p))
+	} else {
+		if o.c.chatty != nil {
+			if o.c.bench {
+				// Benchmarks don't print === CONT, so we should skip the test
+				// printer and just print straight to stdout.
+				fmt.Print(s)
+			} else {
+				o.c.chatty.Printf(o.c.name, "%s", s)
+			}
+		} else {
+			o.c.output = append(o.c.output, s...)
+		}
+	}
+}
+
 // Log formats its arguments using default formatting, analogous to 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