使用 PowerMockito 模拟界面

Question

我需要在 hbase apis 中模拟一个方法。请在下面找到方法

 public static Connection createConnection() throws IOException {
    return createConnection(HBaseConfiguration.create(), null, null);
 }

Connection接口的源代码请看下面的链接

http://grepcode.com/file/repo1.maven.org/maven2/org.apache.hbase/hbase-client/1.1.1/org/apache/hadoop/hbase/client/Connection.java

我试过如下

Connection mockconnection = PowerMockito.mock(Connection.class);
PowerMockito.when(ConnectionFactory.createConnection()).thenReturn(mockconnection);

这是正确的模拟形式吗，因为它不能正常工作

Answer 1

要模拟 static 方法，您需要：

1. 在类或方法级别添加@PrepareForTest。

例子：

@PrepareForTest(Static.class) // Static.class contains static methods

1. Call PowerMockito.mockStatic(class) 模拟静态类（使用PowerMockito.spy(class) 模拟特定方法）：

例子：

PowerMockito.mockStatic(Static.class);

1. 只需使用Mockito.when() 设置您的期望：

例子：

Mockito.when(Static.firstStaticMethod(param)).thenReturn(value);

---

所以在你的情况下，它会是这样的：

@RunWith(PowerMockRunner.class)
public class ConnectionFactoryTest {

    @Test
    @PrepareForTest(ConnectionFactory.class)
    public void testConnection() throws IOException {
        Connection mockconnection = PowerMockito.mock(Connection.class);
        PowerMockito.mockStatic(ConnectionFactory.class);
        PowerMockito.when(ConnectionFactory.createConnection()).thenReturn(mockconnection);

        // Do something here
    }
}

更多关于how to mock a static method的详情