1 | |
package org.apache.maven.surefire.junit; |
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
|
8 | |
|
9 | |
|
10 | |
|
11 | |
|
12 | |
|
13 | |
|
14 | |
|
15 | |
|
16 | |
|
17 | |
|
18 | |
|
19 | |
|
20 | |
|
21 | |
|
22 | |
import org.apache.maven.surefire.report.RunListener; |
23 | |
import org.apache.maven.surefire.report.ReportEntry; |
24 | |
import org.apache.maven.surefire.report.SimpleReportEntry; |
25 | |
|
26 | |
import java.lang.reflect.InvocationHandler; |
27 | |
import java.lang.reflect.InvocationTargetException; |
28 | |
import java.lang.reflect.Method; |
29 | |
import java.util.HashSet; |
30 | |
import java.util.Set; |
31 | |
|
32 | |
public class TestListenerInvocationHandler |
33 | |
implements InvocationHandler |
34 | |
{ |
35 | |
|
36 | |
private static final String START_TEST = "startTest"; |
37 | |
|
38 | |
private static final String ADD_FAILURE = "addFailure"; |
39 | |
|
40 | |
private static final String ADD_ERROR = "addError"; |
41 | |
|
42 | |
private static final String END_TEST = "endTest"; |
43 | |
|
44 | 0 | private final Set failedTestsSet = new HashSet(); |
45 | |
|
46 | |
private RunListener reporter; |
47 | |
|
48 | 0 | private static final Class[] EMPTY_CLASS_ARRAY = new Class[]{ }; |
49 | |
|
50 | 0 | private static final String[] EMPTY_STRING_ARRAY = new String[]{ }; |
51 | |
|
52 | |
private static class FailedTest |
53 | |
{ |
54 | |
private Object testThatFailed; |
55 | |
|
56 | |
private Thread threadOnWhichTestFailed; |
57 | |
|
58 | |
FailedTest( Object testThatFailed, Thread threadOnWhichTestFailed ) |
59 | 0 | { |
60 | 0 | if ( testThatFailed == null ) |
61 | |
{ |
62 | 0 | throw new NullPointerException( "testThatFailed is null" ); |
63 | |
} |
64 | |
|
65 | 0 | if ( threadOnWhichTestFailed == null ) |
66 | |
{ |
67 | 0 | throw new NullPointerException( "threadOnWhichTestFailed is null" ); |
68 | |
} |
69 | |
|
70 | 0 | this.testThatFailed = testThatFailed; |
71 | |
|
72 | 0 | this.threadOnWhichTestFailed = threadOnWhichTestFailed; |
73 | 0 | } |
74 | |
|
75 | |
public boolean equals( Object obj ) |
76 | |
{ |
77 | 0 | boolean retVal = true; |
78 | |
|
79 | 0 | if ( obj == null || getClass() != obj.getClass() ) |
80 | |
{ |
81 | 0 | retVal = false; |
82 | |
} |
83 | |
else |
84 | |
{ |
85 | 0 | FailedTest ft = (FailedTest) obj; |
86 | |
|
87 | 0 | if ( ft.testThatFailed != testThatFailed ) |
88 | |
{ |
89 | 0 | retVal = false; |
90 | |
} |
91 | 0 | else if ( !ft.threadOnWhichTestFailed.equals( threadOnWhichTestFailed ) ) |
92 | |
{ |
93 | 0 | retVal = false; |
94 | |
} |
95 | |
} |
96 | |
|
97 | 0 | return retVal; |
98 | |
} |
99 | |
|
100 | |
public int hashCode() |
101 | |
{ |
102 | 0 | return threadOnWhichTestFailed.hashCode(); |
103 | |
} |
104 | |
} |
105 | |
|
106 | |
public TestListenerInvocationHandler( RunListener reporter ) |
107 | 0 | { |
108 | 0 | if ( reporter == null ) |
109 | |
{ |
110 | 0 | throw new NullPointerException( "reporter is null" ); |
111 | |
} |
112 | |
|
113 | 0 | this.reporter = reporter; |
114 | 0 | } |
115 | |
|
116 | |
public Object invoke( Object proxy, Method method, Object[] args ) |
117 | |
throws Throwable |
118 | |
{ |
119 | 0 | String methodName = method.getName(); |
120 | |
|
121 | 0 | if ( methodName.equals( START_TEST ) ) |
122 | |
{ |
123 | 0 | handleStartTest( args ); |
124 | |
} |
125 | 0 | else if ( methodName.equals( ADD_ERROR ) ) |
126 | |
{ |
127 | 0 | handleAddError( args ); |
128 | |
} |
129 | 0 | else if ( methodName.equals( ADD_FAILURE ) ) |
130 | |
{ |
131 | 0 | handleAddFailure( args ); |
132 | |
} |
133 | 0 | else if ( methodName.equals( END_TEST ) ) |
134 | |
{ |
135 | 0 | handleEndTest( args ); |
136 | |
} |
137 | |
|
138 | 0 | return null; |
139 | |
} |
140 | |
|
141 | |
|
142 | |
public void handleStartTest( Object[] args ) |
143 | |
{ |
144 | 0 | ReportEntry report = new SimpleReportEntry( args[0].getClass().getName(), args[0].toString() ); |
145 | |
|
146 | 0 | reporter.testStarting( report ); |
147 | 0 | } |
148 | |
|
149 | |
|
150 | |
private void handleAddError( Object[] args ) |
151 | |
throws IllegalAccessException, InvocationTargetException |
152 | |
{ |
153 | 0 | ReportEntry report = |
154 | |
new SimpleReportEntry( args[0].getClass().getName(), args[0].toString(), getStackTraceWriter( args ) ); |
155 | |
|
156 | 0 | reporter.testError( report ); |
157 | |
|
158 | 0 | failedTestsSet.add( new FailedTest( args[0], Thread.currentThread() ) ); |
159 | 0 | } |
160 | |
|
161 | |
private JUnitStackTraceWriter getStackTraceWriter( Object[] args ) |
162 | |
throws IllegalAccessException, InvocationTargetException |
163 | |
{ |
164 | |
String testName; |
165 | |
|
166 | |
try |
167 | |
{ |
168 | 0 | Method m = args[0].getClass().getMethod( "getName", EMPTY_CLASS_ARRAY ); |
169 | 0 | testName = (String) m.invoke( args[0], EMPTY_STRING_ARRAY ); |
170 | |
} |
171 | 0 | catch ( NoSuchMethodException e ) |
172 | |
{ |
173 | 0 | testName = "UNKNOWN"; |
174 | 0 | } |
175 | |
|
176 | 0 | return new JUnitStackTraceWriter( args[0].getClass().getName(), testName, (Throwable) args[1] ); |
177 | |
} |
178 | |
|
179 | |
private void handleAddFailure( Object[] args ) |
180 | |
throws IllegalAccessException, InvocationTargetException |
181 | |
{ |
182 | 0 | ReportEntry report = |
183 | |
new SimpleReportEntry( args[0].getClass().getName(), args[0].toString(), getStackTraceWriter( args ) ); |
184 | |
|
185 | 0 | reporter.testFailed( report ); |
186 | |
|
187 | 0 | failedTestsSet.add( new FailedTest( args[0], Thread.currentThread() ) ); |
188 | 0 | } |
189 | |
|
190 | |
private void handleEndTest( Object[] args ) |
191 | |
{ |
192 | 0 | boolean testHadFailed = failedTestsSet.remove( new FailedTest( args[0], Thread.currentThread() ) ); |
193 | |
|
194 | 0 | if ( !testHadFailed ) |
195 | |
{ |
196 | 0 | ReportEntry report = new SimpleReportEntry( args[0].getClass().getName(), args[0].toString() ); |
197 | |
|
198 | 0 | reporter.testSucceeded( report ); |
199 | |
} |
200 | 0 | } |
201 | |
} |