java - PowerMock：模拟静态方法（+在某些特定方法中返回原始值）

Question

我使用 PowerMock 1.4.7 和 JUnit 4.8.2

我只需要模拟一些静态方法，我希望其他方法（来自同一个类）只是返回原始值。当我模拟mockStatic并且不调用when().doReturn()所有静态方法时，返回它们的默认值——比如返回 Object 时为 null 或返回 boolean 时为 false ......等等。所以我尝试 thenCallRealMethod在每个静态方法上显式使用来返回默认实现（意味着没有模拟/没有假货），但我不知道如何在每个可能的参数变化上调用它（=我想要每个可能的输入调用原始方法）。我只知道如何模拟具体的论点变化。

Answer 1

您可以在静态类上使用间谍并仅模拟特定方法：

@RunWith(PowerMockRunner.class)
@PrepareForTest(MyStaticTest.MyStaticClass.class)
public class MyStaticTest {

public static class MyStaticClass {
    public static String getA(String a) {
        return a;
    }
    public static String getB(String b) {
        return b;
    }
}

@Test
public void should_partial_mock_static_class() throws Exception {
    //given
    PowerMockito.spy(MyStaticClass.class);
    given(MyStaticClass.getB(Mockito.anyString())).willReturn("B");
    //then
    assertEquals("A", MyStaticClass.getA("A"));
    assertEquals("B", MyStaticClass.getA("B"));
    assertEquals("C", MyStaticClass.getA("C"));
    assertEquals("B", MyStaticClass.getB("A"));
    assertEquals("B", MyStaticClass.getB("B"));
    assertEquals("B", MyStaticClass.getB("C"));
}

}

Answer 2

您还可以使用存根 API：

stub(method(MyStaticClass.class, "getB")).toReturn("B");

编辑：

要从这些包中使用stub和method静态导入方法：

1. org.powermock.api.support.membermodification.MemberModifier
2. org.powermock.api.support.membermodification.MemberMatcher

有关更多信息，请参阅文档

Answer 3

我设法使用spy并doReturn实现了它。

class MyStatic {
    static String foo() { return "foo"; }
    static String foobar() { return foo() + "bar"; }
}

@Test
public void thisShouldSpyStaticMethods() {
    // arrange
    spy(MyStatic.class);
    doReturn("mocked foo").when(MyStatic.class);
    MyStatic.foo();

    // act
    final String result = MyStatic.foobar();

    // assert
    assertThat(result).isEqualTo("mocked foobar");
}

然后调用要模拟doReturn的方法看起来很奇怪（至少对我来说），但似乎可以解决问题。

使用spywithwhen(MyStatic.foo()).thenReturn("mocked foo")对我不起作用。

PowerMockito 关于模拟静态方法的文档。

Answer 4

基于这个问题PowerMockito mock single static method and return object

PowerMockito.mockStatic(MyStaticClass.class);

单独不模拟所有方法（至少在最近版本的 PowerMockito 中），仅允许稍后模拟单个方法。