easymock教程-自定义参数匹配器

博客分类：

虽然easymock中提供了大量的方法来进行参数匹配，但是对于一些特殊场合比如参数是复杂对象而又不能简单的通过equals()方法来比较，这些现有的参数匹配器就无能为力了。easymock为此提供了IArgumentMatcher 接口来让我们实现自定义的参数匹配器。



    我们还是用例子来说话：

    public interface Service {
        public void execute(Request request);
    }     service类的execute()方法接收一个Request实例作为参数, Request是一个javabean:

   public static class Request {

        private boolean condition;

        private String  value1;

        private String  value2;

//ignore getter and setter method
    }     假设在我们的这个单独的测试案例中，我们有以下参数匹配逻辑: 如果condition为true，则只需要比较value1；如果condition为false，则只需要比较value2. 由于这个逻辑和默认的equals方法不一致，因此我们不能直接使用equals方法，只能实现自己的参数匹配器。

public class RequestMatcher implements IArgumentMatcher {

        private boolean condition;

        private String  expectedValue;

        private RequestMatcher(boolean condition, String expectedValue) {
            this.condition = condition;
            this.expectedValue = expectedValue;
        }

        @Override
        public void appendTo(StringBuffer buffer) {
            buffer.append("RequestMatcher expect(condition=");
            buffer.append(condition);
            buffer.append(" expectedValue=");
            buffer.append(expectedValue);
            buffer.append(")");
        }

        @Override
        public boolean matches(Object argument) {
            if (!(argument instanceof Request)) {
                return false;
            }

            Request request = (Request) argument;
            if (condition) {
                return expectedValue.equals(request.getValue1());
            } else {
                return expectedValue.equals(request.getValue2());
            }
        }

        public static Request requestEquals(boolean condition, String expectedValue) {
            EasyMock.reportMatcher(new RequestMatcher(condition, expectedValue));
            return null;
        }
    }     RequestMatcher 是我们定义的参数匹配器，matches()方法中是参数匹配逻辑的代码实现，appendTo()方法用于在匹配失败时打印错误信息，后面我们会演示这个方法的使用。然后是最重要的方法requestEquals()，在这里我们通过调用EasyMock.reportMatcher()告诉easymock我们要用的参数匹配器。



    在测试案例中，我们和以往一样，先创建了mock对象，然后准备request对象作为测试数据。不同的是，我们没有使用easymock提供的参数匹配方法，而是通过service.execute(RequestMatcher.requestEquals(expectedCondition, expectedValue)); 来调用EasyMock.reportMatcher()，以创建我们自定义的参数匹配器并为它传入了两个必备的参数expectedCondition和expectedValue。



    上面的测试案例可以顺利通过，我们的参数匹配器可以正常工作。然后我们来试试参数匹配不成功的情况



    @Test
    public void testConditionTrueFailure() {
        final boolean expectedCondition = true;
        final String expectedValue = "aaa";

        Service service = EasyMock.createMock("service", Service.class);
        Request request = prepareRequest(expectedCondition, "bbb", "ccc");
        service.execute(RequestMatcher.requestEquals(expectedCondition, expectedValue));
        EasyMock.expectLastCall();

        EasyMock.replay(service);
        service.execute(request);
        EasyMock.verify(service);
    }     注意在Request request = prepareRequest(expectedCondition, "bbb", "ccc")中，我们故意设置value为和期望的不同，当然这样测试案例就通不过了:



java.lang.AssertionError:
  Unexpected method call service.execute(net.sourcesky.study.easymock.tutorial.IArgumentMatcherTest$Request@10ef90c):
    service.execute(RequestMatcher expect(condition=true expectedValue=aaa)): expected: 1, actual: 0
at org.easymock.internal.MockInvocationHandler.invoke(MockInvocationHandler.java:45)
at org.easymock.internal.ObjectMethodsFilter.invoke(ObjectMethodsFilter.java:73)
at $Proxy4.execute(Unknown Source)
at net.sourcesky.study.easymock.tutorial.IArgumentMatcherTest.testConditionTrueFailure(IArgumentMatcherTest.java:72)



    注意"service.execute(RequestMatcher expect(condition=true expectedValue=aaa)): expected: 1, actual: 0"这行，其中的"RequestMatcher expect(condition=true expectedValue=aaa)"是我们在appendTo()方法中构建出来的错误信息。appendTo()方法只在这个时候才被调用，用于生成可读性强的错误信息以便我们在失败时检查，因此不要疏忽了这个方法的实现。