|
1 | 1 | package org.jooby.test; |
2 | 2 |
|
| 3 | +import com.google.common.collect.ArrayListMultimap; |
| 4 | +import com.google.common.collect.Multimap; |
| 5 | +import com.google.common.primitives.Primitives; |
3 | 6 | import static java.util.Objects.requireNonNull; |
| 7 | +import org.easymock.Capture; |
| 8 | +import org.easymock.EasyMock; |
4 | 9 | import static org.easymock.EasyMock.createMock; |
5 | 10 | import static org.easymock.EasyMock.createStrictMock; |
| 11 | +import org.jooby.funzy.Try; |
| 12 | +import org.powermock.api.easymock.PowerMock; |
6 | 13 |
|
7 | 14 | import java.lang.reflect.Modifier; |
8 | 15 | import java.util.ArrayList; |
|
14 | 21 | import java.util.Map; |
15 | 22 | import java.util.Set; |
16 | 23 |
|
17 | | -import org.easymock.Capture; |
18 | | -import org.easymock.EasyMock; |
19 | | -import org.powermock.api.easymock.PowerMock; |
20 | | - |
21 | | -import com.google.common.base.Throwables; |
22 | | -import com.google.common.collect.ArrayListMultimap; |
23 | | -import com.google.common.collect.Multimap; |
24 | | -import com.google.common.primitives.Primitives; |
25 | | - |
26 | 24 | /** |
27 | 25 | * Utility test class for mocks. Internal use only. |
28 | 26 | * |
@@ -213,27 +211,16 @@ public MockUnit expect(final Block block) { |
213 | 211 | public MockUnit run(final Block... blocks) throws Exception { |
214 | 212 |
|
215 | 213 | for (Block block : this.blocks) { |
216 | | - try { |
217 | | - block.run(this); |
218 | | - } catch (Exception | AssertionError ex) { |
219 | | - throw ex; |
220 | | - } catch (Throwable ex) { |
221 | | - Throwables.propagate(ex); |
222 | | - } |
| 214 | + Try.run(() -> block.run(this)) |
| 215 | + .throwException(); |
223 | 216 | } |
224 | 217 |
|
225 | 218 | mockClasses.forEach(PowerMock::replay); |
226 | 219 | partialMocks.forEach(PowerMock::replay); |
227 | 220 | mocks.forEach(EasyMock::replay); |
228 | 221 |
|
229 | 222 | for (Block main : blocks) { |
230 | | - try { |
231 | | - main.run(this); |
232 | | - } catch (Exception | AssertionError ex) { |
233 | | - throw ex; |
234 | | - } catch (Throwable ex) { |
235 | | - Throwables.propagate(ex); |
236 | | - } |
| 223 | + Try.run(() -> main.run(this)).throwException(); |
237 | 224 | } |
238 | 225 |
|
239 | 226 | mocks.forEach(EasyMock::verify); |
|
0 commit comments