|
1 |
| |
|
2 |
| |
|
3 |
| |
|
4 |
| |
|
5 |
| package org.easymock.internal; |
|
6 |
| |
|
7 |
| import java.lang.reflect.Method; |
|
8 |
| import java.util.HashMap; |
|
9 |
| import java.util.Map; |
|
10 |
| |
|
11 |
| import junit.framework.AssertionFailedError; |
|
12 |
| |
|
13 |
| import org.easymock.ArgumentsMatcher; |
|
14 |
| import org.easymock.MockControl; |
|
15 |
| |
|
16 |
| public abstract class AbstractBehavior implements IBehavior { |
|
17 |
| private ArgumentsMatcher defaultMatcher; |
|
18 |
| |
|
19 |
| private boolean defaultMatcherSet; |
|
20 |
| |
|
21 |
| private Map<Method, ArgumentsMatcher> matchers = new HashMap<Method, ArgumentsMatcher>(); |
|
22 |
| |
|
23 |
| private Map<Method, Result> defaultResults = new HashMap<Method, Result>(); |
|
24 |
| |
|
25 |
138
| public void setDefaultMatcher(ArgumentsMatcher matcher) {
|
|
26 |
138
| if (defaultMatcherSet) {
|
|
27 |
2
| throw new RuntimeExceptionWrapper(
|
|
28 |
| new IllegalStateException( |
|
29 |
| "default matcher can only be set once directly after creation of the MockControl")); |
|
30 |
| } |
|
31 |
136
| defaultMatcher = matcher;
|
|
32 |
136
| defaultMatcherSet = true;
|
|
33 |
| } |
|
34 |
| |
|
35 |
13
| public void setMatcher(Method method, ArgumentsMatcher matcher) {
|
|
36 |
13
| if (matchers.containsKey(method) && matchers.get(method) != matcher) {
|
|
37 |
2
| throw new RuntimeExceptionWrapper(new IllegalStateException(
|
|
38 |
| "for method " |
|
39 |
| + method.getName() |
|
40 |
| + "(" |
|
41 |
2
| + (method.getParameterTypes().length == 0 ? ""
|
|
42 |
| : "...") |
|
43 |
| + "), a matcher has already been set")); |
|
44 |
| } |
|
45 |
11
| matchers.put(method, matcher);
|
|
46 |
| } |
|
47 |
| |
|
48 |
32
| public void setDefaultResult(Method method, Result result) {
|
|
49 |
32
| defaultResults.put(method, result);
|
|
50 |
| } |
|
51 |
| |
|
52 |
320
| public final Result addActual(MethodCall methodCall) {
|
|
53 |
320
| try {
|
|
54 |
320
| return doAddActual(methodCall);
|
|
55 |
| } catch (AssertionFailedErrorWrapper e) { |
|
56 |
19
| throw new AssertionFailedErrorWrapper(new AssertionFailedError(
|
|
57 |
| "\n Unexpected method call " + methodCall.toString(getMatcher(methodCall.getMethod())) + ":" |
|
58 |
| + e.getAssertionFailedError().getMessage())); |
|
59 |
| } |
|
60 |
| } |
|
61 |
| |
|
62 |
119
| public final void verify() {
|
|
63 |
119
| try {
|
|
64 |
119
| doVerify();
|
|
65 |
| } catch (AssertionFailedErrorWrapper e) { |
|
66 |
12
| throw new AssertionFailedErrorWrapper(new AssertionFailedError(
|
|
67 |
| "\n Expectation failure on verify:" |
|
68 |
| + e.getAssertionFailedError().getMessage())); |
|
69 |
| } |
|
70 |
| } |
|
71 |
| |
|
72 |
86
| protected Result getDefaultResult(Method method) {
|
|
73 |
86
| return defaultResults.get(method);
|
|
74 |
| } |
|
75 |
| |
|
76 |
214
| protected ArgumentsMatcher getMatcher(Method method) {
|
|
77 |
214
| if (!matchers.containsKey(method)) {
|
|
78 |
156
| if (!defaultMatcherSet) {
|
|
79 |
132
| setDefaultMatcher(MockControl.EQUALS_MATCHER);
|
|
80 |
| } |
|
81 |
156
| matchers.put(method, defaultMatcher);
|
|
82 |
| } |
|
83 |
214
| return matchers.get(method);
|
|
84 |
| } |
|
85 |
| |
|
86 |
| protected abstract void doVerify(); |
|
87 |
| |
|
88 |
| protected abstract Result doAddActual(MethodCall methodCall); |
|
89 |
| } |
