blob: bff7c723510785cb8e1243c22877481b082c1017 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
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;
}
}
}
}
|
