使用Mockito测试委托方法

Question

在下面这样的类中，围绕doActionOne()和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
    }
}

这样的测试似乎需要某种部分模拟或间谍，但我无法纠正这一点。

尽管这种方法不起作用，但测试应该如下所示。

@Test
public void testDoActionOne(){
    ClassUnderTest cut = Mockito.mock(ClassUnderTest.class);
    //call the real method
    when(cut.doActionOne("A1", 1)).thenCallRealMethod();
    //test delegate called with correct params
    verify(cut, times(1)).doAction("A1", 1, true); //fails wanted but not invoked
}

不确定我是否需要这样的东西：

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

Answer 1

我知道两种方法。一种方法是创建一个匿名内部类，在其中覆盖不想测试的方法。这不涉及任何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);

Answer 2

它是这样的：

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