使用 PowerMockito 模拟 class 所有实例的方法

Question

无论调用此方法的实例是什么，我都试图模拟一个方法。据我所知，Mockito 不会发生这种情况，应该使用 PowerMockito 来完成，但我无法弄清楚应该如何完成。我有

public class B{
    public void veryAnnoyingMethod(int i, short s){}
}
public class A{
    public void veryImportantMethod(){
        B newB = new B();
        newB.veryAnnoyingMethod(i,s);
        ...
    }
}

我正在尝试测试 veryImportantMethod() 并想模拟 veryAnnoyingMethod()。我不想像这里解释的那样模拟构造函数：Mocking methods of local scope objects with Mockito 因为除了 veryAnnoyingMethod() 我需要 newB 功能齐全。

我想到了类似的东西：

B dummyB = PowerMockito.spy(new B());
PowerMockito.doReturn(null).when(dummyB.veryAnnoyingMethod(Mockito.anyInt(), Mockito.anyShort()));

但它抛出异常，我认为这不是我真正需要的。

更新

经过更多阅读，我似乎必须使用 PowerMockito.stub 或 PowerMockito.supress。所以我尝试了以下方法：

PowerMockito.stub(PowerMockito.method(B.class, "veryAnnoyingMethod", int.class, short.class)).toReturn(null);

或

PowerMockito.supress(PowerMockito.method(B.class, "veryAnnoyingMethod", int.class, short.class));

但是无论哪种方法仍然执行

Answer 1

好的，我终于让它工作了。使用 stub 和 suppress 中的任何一个都可以。测试看起来像这样：

@RunWith(PowerMockRunner.class)
@PrepareForTest(B.class)
class TestClassAWithPowerMockito{
    @Test
    public void testVeryImportantMethod() throws Exception {
        A a = new A();
        PowerMockito.suppress(PowerMockito.method(B.class, "veryAnnoyingMethod", int.class, short.class));
        a.veryImportantMethod();
        ...
    }
}

注意 @RunWith(PowerMockRunner.class) 和 @PrepareForTest(B.class) 我不得不单独使用 PowerMockito 进行测试 class 因为添加这 2 个符号使我的一些其他测试（不使用 PowerMockito，但仅使用 Mockito）失败。

希望这是清楚的，并帮助那些正在努力解决类似问题的人。

使用 PowerMockito 模拟 class 所有实例的方法

Mocking method for all instances of class with PowerMockito

java

unit-testing

mockito

powermockito