SwingConsolePaneBenchmark: do some sanity tests

Let's double check the output that was sent to the console,
verifying that everything is there.

For convenience, we use JUnit for our assertion statements,
even though the routine in question is not a unit test per se.

Author: ctrueden

pom.xml

@@ -104,6 +104,12 @@
 			<artifactId>jdatepicker</artifactId>
 			<version>1.3.2</version>
 		</dependency>
+
+		<!-- Test scope dependencies -->
+		<dependency>
+			<groupId>junit</groupId>
+			<artifactId>junit</artifactId>
+		</dependency>
 	</dependencies>
 
 	<build>

src/test/java/org/scijava/ui/swing/console/SwingConsolePaneBenchmark.java

@@ -30,11 +30,20 @@
 
 package org.scijava.ui.swing.console;
 
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+
+import java.util.Arrays;
 import java.util.concurrent.Future;
 
+import javax.swing.JTextPane;
+import javax.swing.text.BadLocationException;
+import javax.swing.text.Document;
+
 import org.scijava.Context;
 import org.scijava.thread.ThreadService;
 import org.scijava.ui.UIService;
+import org.scijava.ui.swing.sdi.SwingSDIUI;
 
 /**
  * A behavioral test and benchmark of {@link SwingConsolePane}.
@@ -60,6 +69,7 @@ public static void main(final String[] args) throws Exception {
 			{ ": {ERR} iteration #", ": {OUT} iteration #" };
 		final String outLabel = streamLabels[1];
 		final String errLabel = streamLabels[0];
+		final int numStreams = streamLabels.length;
 
 		final int initialDelay = 500;
 
@@ -95,6 +105,52 @@ public void run() {
 		final long end = System.currentTimeMillis();
 		System.out.println();
 		System.out.println("Benchmark took " + (end - start) + " ms");
+
+		// Finally, check for completeness of output.
+		// NB: We do this **also on the EDT** so that all output has flushed.
+		final String completenessMessage = "Checking for completeness of output...";
+		System.out.println();
+		System.out.println(completenessMessage);
+		threadService.queue(new Runnable() {
+
+			@Override
+			public void run() {
+				System.out.println();
+				final SwingSDIUI ui =
+					(SwingSDIUI) context.service(UIService.class).getVisibleUIs().get(0);
+				final JTextPane textPane = ui.getConsolePane().getTextPane();
+				final Document doc = textPane.getDocument();
+				try {
+					final String text = doc.getText(0, doc.getLength());
+					final String[] lines = text.split("\n");
+					Arrays.sort(lines);
+
+					int lineIndex = 0;
+					assertEquals("", lines[lineIndex++]);
+					assertEquals("", lines[lineIndex++]);
+					for (int t = 0; t < numThreads; t++) {
+						for (int s = 0; s < numStreams; s++) {
+							for (int i = 0; i < numOperations; i++) {
+								final String expected = str(t, streamLabels[s], i);
+								final String actual = lines[lineIndex++];
+								assertEquals(expected, actual);
+							}
+						}
+					}
+					assertTrue(lines[lineIndex++].startsWith("Benchmark took "));
+					assertEquals(completenessMessage, lines[lineIndex++]);
+					assertEquals("Goodbye cruel world!", lines[lineIndex++]);
+					assertEquals("Hello world!", lines[lineIndex++]);
+					assertEquals(lineIndex, lines.length);
+
+					System.out.println("Success! All output accounted for!");
+				}
+				catch (final BadLocationException exc) {
+					exc.printStackTrace();
+				}
+			}
+		});
+
 	}
 
 	// - Helper methods --