|
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 org.easymock.ArgumentsMatcher; |
|
12 |
| import org.easymock.MockControl; |
|
13 |
| |
|
14 |
| public class LegacyMatcherProvider { |
|
15 |
| |
|
16 |
| private ArgumentsMatcher defaultMatcher; |
|
17 |
| |
|
18 |
| private boolean defaultMatcherSet; |
|
19 |
| |
|
20 |
| private Map<Method, ArgumentsMatcher> matchers = new HashMap<Method, ArgumentsMatcher>(); |
|
21 |
| |
|
22 |
14
| public ArgumentsMatcher getMatcher(Method method) {
|
|
23 |
14
| if (!matchers.containsKey(method)) {
|
|
24 |
5
| if (!defaultMatcherSet) {
|
|
25 |
1
| setDefaultMatcher(MockControl.EQUALS_MATCHER);
|
|
26 |
| } |
|
27 |
5
| matchers.put(method, defaultMatcher);
|
|
28 |
| } |
|
29 |
14
| return matchers.get(method);
|
|
30 |
| } |
|
31 |
| |
|
32 |
7
| public void setDefaultMatcher(ArgumentsMatcher matcher) {
|
|
33 |
7
| if (defaultMatcherSet) {
|
|
34 |
1
| throw new RuntimeExceptionWrapper(
|
|
35 |
| new IllegalStateException( |
|
36 |
| "default matcher can only be set once directly after creation of the MockControl")); |
|
37 |
| } |
|
38 |
6
| defaultMatcher = matcher;
|
|
39 |
6
| defaultMatcherSet = true;
|
|
40 |
| } |
|
41 |
| |
|
42 |
13
| public void setMatcher(Method method, ArgumentsMatcher matcher) {
|
|
43 |
13
| if (matchers.containsKey(method) && matchers.get(method) != matcher) {
|
|
44 |
2
| throw new RuntimeExceptionWrapper(new IllegalStateException(
|
|
45 |
| "for method " |
|
46 |
| + method.getName() |
|
47 |
| + "(" |
|
48 |
2
| + (method.getParameterTypes().length == 0 ? ""
|
|
49 |
| : "...") |
|
50 |
| + "), a matcher has already been set")); |
|
51 |
| } |
|
52 |
11
| matchers.put(method, matcher);
|
|
53 |
| } |
|
54 |
| } |
