1 files changed, 71 insertions, 0 deletions
diff --git a/junit4/src/main/java/org/junit/internal/runners/statements/FailOnTimeout.java b/junit4/src/main/java/org/junit/internal/runners/statements/FailOnTimeout.java
new file mode 100644
index 0000000..bff7c72
--- /dev/null
+++ b/junit4/src/main/java/org/junit/internal/runners/statements/FailOnTimeout.java
@@ -0,0 +1,71 @@
+/**
+ * 
+ */
+package org.junit.internal.runners.statements;
+
+import org.junit.runners.model.Statement;
+
+public class FailOnTimeout extends Statement {
+	private final Statement fOriginalStatement;
+
+	private final long fTimeout;
+
+	public FailOnTimeout(Statement originalStatement, long timeout) {
+		fOriginalStatement= originalStatement;
+		fTimeout= timeout;
+	}
+
+	@Override
+	public void evaluate() throws Throwable {
+		StatementThread thread= evaluateStatement();
+		if (!thread.fFinished)
+			throwExceptionForUnfinishedThread(thread);
+	}
+
+	private StatementThread evaluateStatement() throws InterruptedException {
+		StatementThread thread= new StatementThread(fOriginalStatement);
+		thread.start();
+		thread.join(fTimeout);
+		thread.interrupt();
+		return thread;
+	}
+
+	private void throwExceptionForUnfinishedThread(StatementThread thread)
+			throws Throwable {
+		if (thread.fExceptionThrownByOriginalStatement != null)
+			throw thread.fExceptionThrownByOriginalStatement;
+		else
+			throwTimeoutException(thread);
+	}
+
+	private void throwTimeoutException(StatementThread thread) throws Exception {
+		Exception exception= new Exception(String.format(
+				"test timed out after %d milliseconds", fTimeout));
+		exception.setStackTrace(thread.getStackTrace());
+		throw exception;
+	}
+
+	private static class StatementThread extends Thread {
+		private final Statement fStatement;
+
+		private boolean fFinished= false;
+
+		private Throwable fExceptionThrownByOriginalStatement= null;
+
+		public StatementThread(Statement statement) {
+			fStatement= statement;
+		}
+
+		@Override
+		public void run() {
+			try {
+				fStatement.evaluate();
+				fFinished= true;
+			} catch (InterruptedException e) {
+				//don't log the InterruptedException
+			} catch (Throwable e) {
+				fExceptionThrownByOriginalStatement= e;
+			}
+		}
+	}
+}
+\ No newline at end of file