使用Mockito测试委托方法

时间:2015-11-18 09:52:41

标签: java unit-testing mockito

在下面的类中,DYLD_PRINT_STATISTICSdoActionOne()所需的唯一测试是确保他们使用正确的参数委托给doActionTwo()

由于委托doAction()方法需要大量设置,因此任何解决方案都应该阻止调用真实方法。

doAction(String a, int b, boolean c)

这样的测试似乎需要某种部分模拟或间谍,但是我无法正确判断。

测试看起来应该如下所示,尽管这种方法不起作用。

public class ClassUnderTest {

    public void doActionOne(String a, int b) {
        doAction(a, b, true);
    }

    public void doActionTwo(String a, int b) {
        doAction(a, b, false);
    }

    public void doAction(String a, int b, boolean c) {
        //already tested
    }
}

不确定我是否需要以下内容:

http://docs.mockito.googlecode.com/hg/1.9.5/org/mockito/AdditionalAnswers.html#delegatesTo(java.lang.Object)

2 个答案:

答案 0 :(得分:2)

我知道有两种方法可以做到这一点。一种方法是创建一个匿名内部类,您可以在其中覆盖您不想测试的方法。这不涉及任何Mockito魔法。另一种方法是使用Mockito Spy对象,这是实例的代理,允许您为某些方法指定存根行为,并让其他方法落入实际实例。

方法1,使用匿名内部类:

public class MyTest {

    private String a;
    private String b;
    private boolean c;

    private ClassUnderTest instance = new ClassUnderTest() {
        @Override
        public void doAction(String a, int b, boolean c) {
            MyTest.this.a = a; 
            MyTest.this.b = b;
            MyTest.this.c = c;
        }
    }

    public void test() {
        // SETUP
        String expectedA = "test value A";
        String expectedB = "test value B";
        boolean expectedC = true;

        // CALL
        instance.doActionOne(expectedA, expectedB);

        // VERIFY
        assertEquals(expectedA, a);
        assertEquals(expectedB, b);
        assertEquals(expectedC , c);
    }

方法2,使用Mockito间谍对象:

@RunWith(MockitoJUnitRunner.class)
public class MyTest {

    @Spy
    private ClassUnderTest instance;

    public void test() {
        // SETUP
        String expectedA = "test value A";
        String expectedB = "test value B";
        boolean expectedC = true;
        doNothing().when(instance).doAction(expectedA , expectedB, expectedC);

        // CALL
        instance.doActionOne(expectedA, expectedB);

        // VERIFY
        verify(instance, times(1)).doAction(expectedA , expectedB, expectedC);
    }

因为间谍是Mockito控制的代理,你也可以验证是否在间谍上调用了方法,这就是你需要的。您还可以为doAction指定一个存根返回值(如果它不是一个void方法):

        // SETUP
        doReturn("stubbed value").when(instance).doAction(expectedA , expectedB, expectedC);

答案 1 :(得分:0)

它是这样的:

when(object.getFieldValue(attribute)).thenReturn(NESTED_ID);
verify(object, times(1)).getFieldValue(eq(attribute));