在与被测试类(CUT)相同的类中使用Mockito到Stub方法

时间:2014-01-07 15:00:35

标签: java mockito

我正在尝试使用Mockito测试一些遗留代码,并且该方法的类型为void。

我已经在其他类中省略了很多方法调用,这很好用。 但是,我还需要能够在同一个类中删除对其他方法的某些调用。

目前这不起作用。

e.g。我的班级如下:

public class Test {


    public Test(dummy dummy) {

    }

    public void checkTask(Task task, List <String> dependencyOnLastSuccessList) throws TaskException {
        callToOtherClass.method1 // This works fine, I can stub it using mockito

        updateAndReschedule(Long id, String message) // call to method in same class, I cannot stub it
    }

    public void updateAndReschedule(Long id, String message) {
        //method logic.....
    }
}

这是我的testClass,显示我现在的情况:

@Test
public void testMyMethod() {
    Test testRef = new Test(taskJob);
    Test spy = spy (testRef);

    // when a particular method is called, return a specific  object
    when(callToOtherClass.method1).thenReturn(ObjectABC);

    //doNothing when my local method is called
    doNothing().when(spy).updateAndReschedule(1, "test");       
    //make method call
    spy.checkTask(ts, taskDependencies);

}

2 个答案:

答案 0 :(得分:26)

您应该按如下方式实例化testRef:

Test testRef = new Test(taskJob) {

    public void updateAndReschedule(Long id, String message) {
        //do nothing
    }

};

不需要间谍。

答案 1 :(得分:3)

在我看来间谍对象而不是模拟。 Spy是作为现有真实对象的代理创建的模拟;一些方法可以被存根,而未被存在的方法被转发到被覆盖的对象

间谍比选择方法的匿名实现更优雅

看一下例子:

Mockito: Trying to spy on method is calling the original method

关于mockito的好文章你可以阅读

http://refcardz.dzone.com/refcardz/mockito