Flush output streams in capture

Avoids possible test failures owing to unflushed buffers

Author: Dave Syer

spring-boot/src/test/java/org/springframework/boot/OutputCapture.java

@@ -72,8 +72,19 @@ protected void releaseOutput() {
 		this.copy = null;
 	}
 
+	public void flush() {
+		try {
+			this.captureOut.flush();
+			this.captureErr.flush();
+		}
+		catch (IOException e) {
+			// ignore
+		}
+	}
+
 	@Override
 	public String toString() {
+		flush();
 		return this.copy.toString();
 	}
 
@@ -104,12 +115,17 @@ public void write(byte[] b) throws IOException {
 		public void write(byte[] b, int off, int len) throws IOException {
 			this.copy.write(b, off, len);
 			this.original.write(b, off, len);
-			this.original.flush();
 		}
 
 		public PrintStream getOriginal() {
 			return this.original;
 		}
+
+		@Override
+		public void flush() throws IOException {
+			this.copy.flush();
+			this.original.flush();
+		}
 	}
 
 	/**