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 org.easymock.MockControl;
23
24 import java.util.HashSet;
25 import java.util.Iterator;
26 import java.util.Set;
27
28 import junit.framework.AssertionFailedError;
29
30 public class MockManager
31 {
32
33 private final Set<MockControl> mockControls = new HashSet<MockControl>();
34
35 public void add( final MockControl control )
36 {
37 mockControls.add( control );
38 }
39
40 public void clear()
41 {
42 mockControls.clear();
43 }
44
45 public void replayAll()
46 {
47 for ( final Iterator<MockControl> it = mockControls.iterator(); it.hasNext(); )
48 {
49 final MockControl control = it.next();
50
51 control.replay();
52 }
53 }
54
55 public void verifyAll()
56 {
57 for ( final Iterator<MockControl> it = mockControls.iterator(); it.hasNext(); )
58 {
59 final MockControl control = it.next();
60
61 try
62 {
63 control.verify();
64 }
65 catch ( final AssertionFailedError err )
66 {
67 final String message =
68 "MockControl: " + control + " of: " + control.getMock() + " failed.\n" + err.getMessage();
69 final AssertionFailedError e = new AssertionFailedError( message );
70 e.initCause( err );
71
72 throw e;
73 }
74 }
75 }
76
77 }