具有多个参数的Mockito参数匹配器

时间:2016-11-02 04:19:13

标签: java mockito

我有一个方法课:

class URAction {
      public List<URules> getUrules(Cond cond, Cat cat) {
             ...
      } 
}

我想创建它的模拟:

@Mock
URAction uraMock;

@Test
public void testSth() {
    Cond cond1;
    Cat cat1;
    List<URule> uRules;

    // pseudo code
    // when uraMock's getUrules is called with cond1 and cat1
    // then return uRules
}

问题是我可以只为一个参数使模拟返回uRules:

 when(uraMock.getUrules(argThat(
        new ArgumentMatcher<Cond> () {

            @Override
            public boolean matches(Object argument) {
                Cond cond = ((Cond) argument);
                 if (cond.getConditionKey().equals(cond1.getConditionKey()) 
                     return true;
                 return false;
            }

        }
    ))).thenReturn(uRules);

不确定如何传递第二个参数,即上面调用时的Cat。

非常感谢任何帮助。

由于

1 个答案:

答案 0 :(得分:3)

你能尝试为第二个参数匹配器添加另一个argThat(argumentMatcher)吗?

另外,我发现最好不要将匿名类定义为方法,而不是像你所做的那样内联。然后你也可以将它用于verify()。

您的方法应如下所示

ArgumentMatcher<Cond> matcherOne(Cond cond1){
   return new ArgumentMatcher<Cond> () {
        @Override
        public boolean matches(Object argument) {
            Cond cond = ((Cond) argument);
            if (cond.getConditionKey().equals(cond1.getConditionKey()) 
               return true;
            return false;
        }
   }
}

ArgumentMatcher<OtherParam> matcherTwo(OtherParam otherParam){
   return new ArgumentMatcher<OtherParam> () {
        @Override
        public boolean matches(Object argument) {
            OtherParam otherParam = ((OtherParam) argument);
            if (<some logic>) 
               return true;
            return false;
        }
   }
}

然后你可以这样调用你的方法,

when(uraMock.getUrules(argThat(matcherOne(cond1)), argThat(matcherTwo(otherParam)))).thenReturn(uRules);

然后,正如我可以调用here,检查你的when方法是否真的被称为

verify(uraMock).getUrules(argThat(matcherOne(cond1)), argThat(matcherTwo(otherParam)));

如果你不关心另一个参数,你可以这样做,

when(uraMock.getUrules(argThat(matcherOne(cond1)), argThat(any()))).thenReturn(uRules);

有关详细信息,请参阅:verify

希望很清楚..祝你好运!