如何测试调用父类的受保护(不需要的)方法的方法?

Question

我陷入了一个非常奇怪的情况。我有一些特定的代码，我需要测试。

public class A {
    /*
     * The real method of real class is so big that I just don't want to test it.
     * That's why I use throwing an exception.
     */
    protected void method(Integer result) {
        throw new RuntimeException("Oops!");
    }
    protected <T> T generifiedMethod(String s, T type) {
        throw new RuntimeException("Oops!");
    }
    protected void mainMethod(Integer value) {
        throw new RuntimeException("Oops!");
    }
}

我也有一个子类:

public class B extends A {
    @Override
    protected void mainMethod(Integer value) {
        if (value == 100500) {
            Integer result = super.generifiedMethod("abc", 100);
            super.method(result);
        }
        super.mainMethod(value);
    }
}

我需要用测试覆盖子类。

我尝试了很多与PowerMockito的组合，但是它们都不能验证父类的受保护方法的调用。此外，我有一个限制，只使用Mockito, PowerMockito和TestNG。

下面是我的测试代码(其中一个变体):

@Test
public void should_invoke_parent_logic_methods_of_A_class() throws Exception {
    /* Given */
    A aSpy = PowerMockito.spy(new A());
    PowerMockito.doReturn(250).when(aSpy, "generifiedMethod", "abc", 100);
    PowerMockito.doNothing().when(aSpy, "method", 250);
    PowerMockito.suppress(method(A.class, "mainMethod", Integer.class));
    /* When */
    aSpy.mainMethod(100500);
    /* Then */
    /**
     * Here I need to verify invocation of all methods of class A (generifiedMethod(), method(),
     * and mainMethod()). But I don't need them to be invoked because their logic is unwanted
     * to be tested in case of tests for class B.
     */
}

如果你能给我一些关于b类测试的建议，我将不胜感激。

更新

如果我在Then部分添加以下代码

Mockito.verify(aSpy, times(3)).mainMethod(100500);
Mockito.verify(aSpy, times(1)).generifiedMethod("abc", 100);
Mockito.verify(aSpy, times(1)).method(250);

它给了我以下错误信息:

Wanted but not invoked:
a.generifiedMethod("abc", 100);

Answer 1

您是否考虑过使用变体来改变类的设计并使用组合而不是继承?然后，您将能够模拟/监视类A的实例并将其注入类B的实例。在这种情况下，您将能够配置您需要的任何行为。

我真的不确定doCallRealMethod()会让你的把戏，因为你可以选择模拟方法或调用真正的方法，但不能同时。

Answer 2

如果我理解正确的话，您想要测试类B中的方法'mainMethod'。因此，您应该模拟类B中的对象，而不是类a中的对象。

/* Given */
B b = Mockito.mock(B.class);
//use this to skip the method execution
Mockito.doNothing().when(b).generifiedMethod(Mockito.anyString(), Mockito.any());
Mockito.doNothing().when(b).method(Mockito.anyInt());
//or use this to return whatever you want when the methods are called
Mockito.doReturn(new Object()).when(b).method(Mockito.anyInt());
Mockito.doReturn(new Object()).when(b).generifiedMethod(Mockito.anyString(), Mockito.any());

然后你可以从B调用mainMethod，已经知道其他方法将返回什么。