使用Java8 lambda在模拟类上匹配参数

Question

我单位测试的类要依赖性，并在该依赖项上调用功能。该功能将复杂的对象作为其参数，并产生结果。现在，我想模拟依赖关系，并根据传递的参数返回某些内容。下面给出了简化的工作版本。

我可以在when方法中使用Java 8 Lambda表达式消除ArgHasNext类吗？类似于下面注释的代码（不编译）。

class ArgHasNext implements ArgumentMatcher<Arg> {
    public boolean matches(Arg arg) {
        return arg.hasNext();
    }
    @Override
    public boolean matches(Object o) {
        return matches((Arg)o);
    }
}
@RunWith(MockitoJUnitRunner.class)
public class ArgumentMatcherTest {
    @Mock
    private Dependency dep = mock(Dependency.class);
    @Test
    public void test() {
        when(dep.func(argThat(new ArgHasNext()))).thenReturn(true);
        // when(dep.func(argThat((Arg a) -> a.hasNext()))).thenReturn(true);
        // when(dep.func(argThat((Arg a) -> !a.hasNext()))).thenReturn(false);
        Sut sut = new Sut(dep);
        assertEquals(sut.method(new Arg(true)), "True");
        assertEquals(sut.method(new Arg(false)), "False");
    }
}
class Arg {
    private boolean hasNext;
    public Arg(boolean hasNext) {
        this.hasNext = hasNext;
    }
    public boolean hasNext() {
        return this.hasNext;
    }
}
class Sut {
    private Dependency dep;
    public Sut(Dependency dep) {
        this.dep = dep;
    }
    public String method(Arg arg) {
        if (dep.func(arg)) {
            return "True";
        }
        else {
            return "False";
        }
    }
}
class Dependency {
    public boolean func(Arg arg) {
        if (arg.hasNext()) {
            return true;
        }
        return false;
    }
}

我正在使用Mockito-Core版本2.0.54-Beta。

编辑好吧，也许我过度简化了示例。在实际情况下，依赖关系func方法返回在测试方法返回之前在SUT中处理的分类查询结果。根据Arg，我希望依赖关系func首次返回第1页结果，第2页结果是第二次结果。只有在我可以根据传递给函数调用的参数获得模拟返回不同值的模拟返回值时，我才能执行此操作。在Mockito中使用lambdas？

Answer 1

现在，我想模拟依赖性，并根据传递的参数返回某些东西。

您首先不应该这样做。任何机会让您的模拟返回定义良好的常数值。

原因是您应该尽可能简单地保持测试代码，以降低测试失败的风险，因为测试代码是错误的。

---

解决问题的解决方案可能是Mockitos Answer接口：

doAnswer(new Answer<YourReturnType>(){ 
    public YourReturnType answer(InvocationOnMock invocation) {
       YourParameterType parameter = (YourParameterType)invocation.getArguments()[0];
       // calculate your return value
       return yourCalculatedReturnValue;
    }
}).when(yourMock).theMethod(any(YourParameterType.class));

---

要清楚，我的模型正在返回一个恒定值。该模拟被多次称为，我希望它第二次返回其他值。 - mvdd

您应该在问题中写这篇文章。该解决方案听起来很简单：

doReturn(VALUE_FOR_FIRST_CALL).
   thenReturn(VALUE_FOR_SECOND_CALL).
   thenReturn(VALUE_FOR_ANY_FURTHER_CALL).
   when(mock).theMethod();

，或者如果您想减少健谈：

doReturn(VALUE_FOR_FIRST_CALL, VALUE_FOR_SECOND_CALL, VALUE_FOR_ANY_FURTHER_CALL).
   when(mock).theMethod();

Answer 2

我认为你要掉下错误的兔子洞。含义：模拟规范是模拟规格。他们不应该是否。

我的意思是：我建议您编写类似：

的测试

@Test
public void testForTrue() {
    when(dep.func(any()).thenReturn(true);
    Sut sut = new Sut(dep);
    assertEquals(sut.method(new Arg(true)), "True");
}
@Test
public void testForFalse() {
    when(dep.func(any()).thenReturn(false);
    Sut sut = new Sut(dep);
    assertEquals(sut.method(new Arg(false)), "False");
}

如果有的话，我然后重构：

private void testFor(boolean depReturnValue, expectedResult) {
    when(dep.func(any()).thenReturn(depReturnValue);
    Sut sut = new Sut(dep);
    assertEquals(sut.method(new Arg(depReturnValue)), expectedResult);
}

并从上面概述的@Test方法调用该方法。

换句话说：不要花费大量时间和精力来具有"智能"嘲笑规格。而是专注于编写简单的直截了当测试。

何时可以编写测试而无需对传递给模拟呼叫的参数进行复杂的分析 - 然后选择该解决方案。

Answer 3

回答我自己的问题。我太专注于使模拟返回的某些不同之处，这取决于所传递的参数，因为模拟方法被多次调用，带有不同的复杂参数对象。

解决方案只是在随后的调用中返回不同的结果，使用Mockito。

很简单。

@RunWith(MockitoJUnitRunner.class)
public class ArgumentMatcherTest {
    @Mock
    private Dependency dep = mock(Dependency.class);
    @Test
    public void test() {
        // return true on first invocation, false on second.
        when(dep.func(anyObject())).thenReturn(true).thenReturn(false);
        Sut sut = new Sut(dep);
        assertEquals(sut.method(new Arg(true)), "True");
        assertEquals(sut.method(new Arg(false)), "False");
    }
}