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

时间:2014-08-18 12:58:11

标签: java unit-testing mockito powermock

我陷入了一个非常奇怪的案子。 我有一些我需要测试的特定代码。  这是:

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);

3 个答案:

答案 0 :(得分:2)

您是否考虑使用variant来更改类的设计并使用合成而不是继承? 然后,您将能够模拟/侦察A类的实例并将其注入到B类的实例中。 在这种情况下,您将能够配置您需要的任何行为。

我真的不确定doCallRealMethod()会为你制作技巧,因为你可以选择模拟方法或调用真实方法,但不能同时使用。

答案 1 :(得分:0)

如果我理解正确,你想测试方法&#39; mainMethod&#39;来自B类。所以,你应该模拟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,已经知道其他方法会返回什么。

答案 2 :(得分:0)

我认为你不应该使用模拟库,而是恢复到旧的专用存根模式。

您必须创建具有以下功能的专用A:

  • 请注意已调用方法methodgenerifiedMethodmainMethod
  • 重置上述指标
  • 提供对上述指标的读取权限。

然后,您可以使用Junit或TestNG构建测试。恕我直言的主要问题是你可能必须使用自定义构建过程来加载A存根类而不是真正的A。