Mockito更改包装类的默认值

时间:2014-05-13 21:20:21

标签: java mockito

我正在尝试模拟具有Integer属性的bean。我正在测试的方法检查该属性的初始值是否为null,如果它是设置它。根据mockito文档:

  

默认情况下,对于返回值的所有方法,mock返回null,空集合或适当的原始/原始包装值(例如:0,false,...表示int / Integer,boolean / Boolean,...)

所以我的属性被mockito设置为0,我的测试失败了。有没有办法覆盖这种默认行为?

修改

杰夫鲍曼给出了一个很好的答案,我想听从他的建议......但我不确定如何。以下是我尝试测试的一些方法:

public class ViewBeanBuilder {

    @Inject
    private ViewBean viewBean;

    public void buildViewBean() {
        ....
        for (Model model : getModels()) {
            if (viewBean.getAmount() == null || model.getAmount() < viewBean.getAmount()) {
                    viewBean.setAmount(model.getAmount());
            }            

        }
        ....
    }

}

我的问题是,如果我不嘲笑ViewBean,那么我会得到一个空指针异常。但是,当我模仿它时,我已经讨论过问题了。我采取了错误的做法吗?还有另一种方法吗?

我有一个想法是在我的viewBean周围放置一个getter:

public class ViewBeanBuilder {

    @Inject
    private ViewBean viewBean;

    public void buildViewBean() {
        ....
        for (Model model : getModels()) {
            if (getViewBean().getAmount() == null || model.getAmount() < getViewBean().getAmount()) {
                    viewBean.setAmount(model.getAmount());
            }            

        }
        ....
    }

    private ViewBean getViewBean() {
        return viewBean;
    }

}

然后在我的测试中我可以使用:

@InjectMocks
private ViewBeanBuilder builder = new ViewBeanBuilder();    

@Test
private void testBuilder() {
    ViewBean viewBean = new ViewBean();
    when(builder.getViewBean()).thenReturn(viewBean);

    builder.buildViewBean();
    ....
}

我明天可能会试试这个。但这是一种有效的方法吗?

修改

使用when也失败了......

org.mockito.exceptions.misusing.MissingMethodInvocationException: 
when() requires an argument which has to be a method call on a mock.
For example:
    when(mock.getArticles()).thenReturn(articles);

1 个答案:

答案 0 :(得分:3)

首先,不要模拟数据对象(包含重要状态的对象)。 Bean是数据对象。相反,创建一个真正的bean,将值放入其中,并在测试中使用该bean。

也就是说,Mockito模拟没有实现,并且bean方法和其他方法之间的行为不同:

YourBean bean = Mockito.mock(YourBean.class);  // A mock acting like YourBean.
bean.getIntegerValue();                        // Returns 0 by default.
bean.setIntegerValue(50);                      //
bean.getIntegerValue();                        // Still returns 0, not 50.

                                               // The call to setIntegerValue is
verify(bean).setIntegerValue(50);              // recorded; Mockito just doesn't
                                               // match the getter and setter.

要覆盖这些默认值stub using when and thenReturn statements。您可以随意使用多个,thenReturn可以接受您喜欢的任意数量的参数。最后一个值将无限期重复。<​​/ p>

YourBean bean = Mockito.mock(YourBean.class);  // A mock acting like YourBean.
when(bean.getIntegerValue())
    .thenReturn(1)                             // Calls to getIntegerValue()
    .thenReturn(3, 5)                          // in sequence will return
    .thenReturn(7);                            // (1, 3, 5, 7, 7, 7...).

这也适用于您的简单案例到存根null而不是0:

YourBean bean = Mockito.mock(YourBean.class);
when(bean.getIntegerValue()).thenReturn(null);

请注意,让getIntegerValue返回最近使用setIntegerValue设置的值的唯一方法是write Answers,这很棘手且冗长。这是模拟数据对象没有意义的部分原因:制作模拟bean的逻辑很复杂,使用真正的bean对测试来说简单而真实。