Java单元测试模拟一个以谓词作为参数的方法

时间:2015-06-29 15:32:05

标签: java-8 mockito predicate

我有两个班级:

ClassA {
   public String methodA(String accountId, Predicate<User> predicate) {
      // more code
   }; 
}

ClassB {
  methodB(){
    ClassA objectA = new ClassA();
    objectA.methodA("some id", PredicatesProvider.isUserValid());
    // more code ...
  }
}

class PredicatesProvider {
   static Predicate<User> isUserValid(){
   return (user) -> {
      return  user.isValid();
   }
}

在我的单元测试中,我需要模拟ClassA,所以我使用Mockito的模拟方法,如下所示:

ClassA mockObjectA = Mockito.mock(ClassA.class);
Mockito.when(mockObjectA).methodA("some id", PredicatesProvider.isUserValid()).thenReturn("something");

Mockito找不到签名赛。

The java.lang.AssertionError: expected:<PredicatesProvider$$Lambda$5/18242360@815b41f> but was:<PredicatesProvider$$Lambda$5/18242360@5542c4ed>

这是我想要实现的简化版本。我猜这是谓词的equals()函数的问题。知道如何模拟具有谓词参数的方法吗?

由于

1 个答案:

答案 0 :(得分:10)

我看到了4种可能的解决方案:

  1. 始终从isUserValid()方法返回完全相同的Predicate实例。由于谓词是无国籍的,这不是问题。

  2. 将Predicate实现为一个真正的类,实现equals()和hashCode()。但与第一种解决方案相比,这太过分了。

  3. 使用匹配器:

     Mockito.when(mockObjectA).methodA(Mockito.eq("some id"), Mockito.<Predicate<User>>anyObject()).thenReturn("something");
    
  4. 不要使用静态方法来创建谓词,而是使用可注入的工厂,您可以模拟并验证:

    PredicatesProvider mockPredicatesProvider = mock(PredicatesProvider.class);
    Predicate<User> expectedPredicate = (u -> true);
    when(mockPredicatesProvider.isUserValid()).thenReturn(expectedPredicate);
    when(mockObjectA).methodA("some id", expectedPredicate).thenReturn("something");