1 package org.apache.maven.plugin.assembly.testutils;
2
3 /*
4 * Licensed to the Apache Software Foundation (ASF) under one
5 * or more contributor license agreements. See the NOTICE file
6 * distributed with this work for additional information
7 * regarding copyright ownership. The ASF licenses this file
8 * to you under the Apache License, Version 2.0 (the
9 * "License"); you may not use this file except in compliance
10 * with the License. You may obtain a copy of the License at
11 *
12 * http://www.apache.org/licenses/LICENSE-2.0
13 *
14 * Unless required by applicable law or agreed to in writing,
15 * software distributed under the License is distributed on an
16 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
17 * KIND, either express or implied. See the License for the
18 * specific language governing permissions and limitations
19 * under the License.
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 }