1 package org.apache.maven.plugin.assembly.testutils;
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 import java.util.HashSet;
23 import java.util.Set;
24
25 import junit.framework.AssertionFailedError;
26
27 import org.easymock.MockControl;
28
29 public class MockManager
30 {
31
32 private final Set<MockControl> mockControls = new HashSet<MockControl>();
33
34 public void add( final MockControl control )
35 {
36 mockControls.add( control );
37 }
38
39 public void clear()
40 {
41 mockControls.clear();
42 }
43
44 public void replayAll()
45 {
46 for (final MockControl control : mockControls) {
47 control.replay();
48 }
49 }
50
51 public void verifyAll()
52 {
53 for (final MockControl control : mockControls) {
54 try {
55 control.verify();
56 } catch (final AssertionFailedError err) {
57 final String message =
58 "MockControl: " + control + " of: " + control.getMock() + " failed.\n" + err.getMessage();
59 final AssertionFailedError e = new AssertionFailedError(message);
60 e.initCause(err);
61
62 throw e;
63 }
64 }
65 }
66
67 }