Mockito自定义verify参数Matcher

在TDD开发中，也许我们会遇见对一些重要的无返回值的行为测试，比如在用户的积分DB中增加用户的积分，这个行为对于我们的业务具有重要的价值，所以我们也希望能测试覆盖这部分业务价值。这个时候我们就得使用mockito带来的verify断言，但verify的参数断言主要有eq，或者any常见的方式。有时我们也希望能够断言对象的一部分属性，比如上文的积分数值，对于不同的场景增加的用户积分可能不同。

回到Mockito的参数Matcher，Mockito给我们提供了ArgumentMatcher，以供我们来扩展Matcher。下面假设一个增加用户积分的场景：

 public class Game {        PRivate String type;        private int rate;        public Game(String type, int rate) {            this.type = type;            this.rate = rate;        }        public String getType() {            return type;        }        public int getRate() {            return rate;        }    }    public class GameDao {        public void addRate(Game game) {            //TODO: insert to db        }    }

我们希望能够对verify GameDao调用了addRate，并且是积分rate为特定值。

所以我们可以扩展Mockito的ArgumentMatcher：

public class PartyMatcher<T> extends ArgumentMatcher<T> {    private Object value;    private Function<T, Object> function;    public PartyMatcher(Function<T, Object> getProperty, Object value) {        this.value = value;        this.function = getProperty;    }    public static <F> PartyMatcher<F> partyMatcher(Function<F, Object> getProperty, Object value) {        return new PartyMatcher<F>(getProperty, value);    }    @Override    public boolean matches(Object o) {        return function.apply((T) o).equals(value);    }}

所以我们的测试可以如下：

    @Test    public void should_run_customer_mockito_matcher() throws Exception {        final GameDao gameDao = mock(GameDao.class);        gameDao.addRate(new Game("签到", 7));        verify(gameDao).addRate(argThat(new PartyMatcher<Game>(new Function<Game, Object>() {            @Override            public Object apply(Game game) {                return game.getRate();            }        }, 7)));        verify(gameDao).addRate(argThat(new PartyMatcher<Game>(new Function<Game, Object>() {            @Override            public Object apply(Game game) {                return game.getType();            }        }, "签到")));    }

Mockito给我们提供了很多关于Matcher扩展的方法，本文只是ArgumentMatcher的实例。