01: /*
02: * Copyright 2005-2006 the original author or authors.
03: *
04: * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
05: * in compliance with the License. You may obtain a copy of the License at
06: *
07: * http://www.apache.org/licenses/LICENSE-2.0
08: *
09: * Unless required by applicable law or agreed to in writing, software distributed under the License
10: * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
11: * or implied. See the License for the specific language governing permissions and limitations under
12: * the License.
13: */
14:
15: package org.strecks.testng;
16:
17: import static org.easymock.classextension.EasyMock.createStrictMock;
18: import static org.easymock.classextension.EasyMock.replay;
19: import static org.easymock.classextension.EasyMock.reset;
20: import static org.easymock.classextension.EasyMock.verify;
21:
22: import java.util.ArrayList;
23: import java.util.List;
24:
25: public class AbstractMockTest {
26:
27: private List mocks = new ArrayList();
28:
29: @SuppressWarnings("unchecked")
30: protected <T extends Object> T newMock(Class<T> clazz) {
31: T object = createStrictMock(clazz);
32: mocks.add(object);
33: return object;
34: }
35:
36: protected void replayMocks() {
37: for (Object mock : mocks) {
38: replay(mock);
39: }
40: }
41:
42: protected void verifyMocks() {
43: for (Object mock : mocks) {
44: verify(mock);
45: }
46: }
47:
48: protected void resetMocks() {
49: for (Object mock : mocks) {
50: reset(mock);
51: }
52: }
53: }
|