Mockito ArgumentCaptor根据匹配器捕获

Question

In mockito we can capture arguments for a method that might be called many times like so:

verify(mocked, atLeastOnce())
    .myMethod(myCaptor.capture());

Then

myCaptor.getAllValues()

However then I would need to sift through all the captured values to find the one I am interested in for verification.

What I would like to do is something like this:

private class IsMySpecialArg extends ArgumentMatcher<Object> {
  public boolean matches(Object other) {
    // Matching condition
  }
}

...

verify(mocked, atLeastOnce())
        .myMethod(myCaptor.capture(argThat(new IsMySpecialArg()));

So that I can simply call myCaptor.getValue() and be assured it is referring to the argument I was actually interested in capturing. What is the best way to accomplish this is mockito , is it supported, or is there something fundamentally wrong with my testing strategy?

Answer 1

I use ArgumentMatchers pretty frequently when writing tests. Here's an example of what you can do when verifying the arguments that are passed to your mock objects:

private static class ExpectationArgumentMatcher extends ArgumentMatcher<String> {

    private final List<String> expectedArguments;

    public ExpectationArgumentMatcher(List<String> expectedArguments) {
        this.expectedArguments = expectedArguments;
    }

    public ExpectationArgumentMatcher() {
        this(new ArrayList<javax.swing.Action>());
    }


    @Override
    public boolean matches(Object argument) {
        if (argument instanceof String) {
            String actualArgument = (String) argument;
            return expectedArguments.contains(actualArgument);
        }
        return false;
    }
}

Obviously, you're expected arguments could just be a string, or something more sophisticated, but you're basically checking that your mock was called upon with an argument that matches your expected argument(s).