嘲弄保护方法

时间:2011-11-29 14:15:29

标签: java mockito powermock powermockito

我想模拟一个继承的受保护方法。我无法直接从java代码调用此方法,因为它从另一个包中的类继承。我找不到在when(...)

中指定此方法来存根的方法
package a;

public class A() {
    protected int m() {}
}

package b;

public class B extends a.A {
    // this class currently does not override m method from a.A
    public asd() {}
}

// test
package b;

class BTest {
    @Test
    public void testClass() {
        B instance = PowerMockito.spy(new B());
        PowerMockito.when(instance, <specify a method m>).thenReturn(123);
        //PowerMockito.when(instance.m()).thenReturn(123); -- obviously does not work
    }
}

我查看了PowerMockito.when覆盖,这似乎只是私有方法!

如何指定受保护的方法?

1 个答案:

答案 0 :(得分:25)

Nutshell:不能总是使用when来存储间谍;使用doReturn

假设静态导入spydoReturnPowerMockito}:

@RunWith(PowerMockRunner.class)
@PrepareForTest(B.class)
public class BTest {
    @Test public void testClass() throws Exception {
        B b = spy(new B());
        doReturn(42).when(b, "m");
        b.asd();
    }
}

您还可以@PrepareForTest(A.class)并在doReturn上设置when(a, "m")。哪个更有意义取决于实际测试。

相关问题