使用 Mockito 模拟静态方法

Question

我写了一个工厂来生产 java.sql.Connection 对象：

 public class MySQLDatabaseConnectionFactory implements DatabaseConnectionFactory {

    @Override public Connection getConnection() {
        try {
            return DriverManager.getConnection(...);
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }
    }
}

我想验证传递给 DriverManager.getConnection 的参数，但我不知道如何模拟静态方法。我正在为我的测试用例使用 JUnit 4 和 Mockito。是否有模拟/验证此特定用例的好方法？

原文由 Naftuli Kay 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

在 Mockito 之上使用 PowerMockito 。

示例代码：

 @RunWith(PowerMockRunner.class)
@PrepareForTest(DriverManager.class)
public class Mocker {

    @Test
    public void shouldVerifyParameters() throws Exception {

        //given
        PowerMockito.mockStatic(DriverManager.class);
        BDDMockito.given(DriverManager.getConnection(...)).willReturn(...);

        //when
        sut.execute(); // System Under Test (sut)

        //then
        PowerMockito.verifyStatic();
        DriverManager.getConnection(...);

    }

更多信息：

• 为什么 Mockito 不模拟静态方法？

原文由 MariuszS 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

自 Mockito 3.4.0 以来，可以在 Mockito 中模拟静态方法。有关详细信息，请参阅：

https://github.com/mockito/mockito/releases/tag/v3.4.0

https://github.com/mockito/mockito/issues/1013

https://javadoc.io/doc/org.mockito/mockito-core/latest/org/mockito/Mockito.html#static_mocks

 assertEquals("foo", Foo.method());
try (MockedStatic mocked = mockStatic(Foo.class)) {
 mocked.when(Foo::method).thenReturn("bar");
 assertEquals("bar", Foo.method());
 mocked.verify(Foo::method);
}
assertEquals("foo", Foo.method());

在你的情况下，是这样的：

   @Test
  public void testStaticMockWithVerification() throws SQLException {
    try (MockedStatic<DriverManager> dummy = Mockito.mockStatic(DriverManager.class)) {
      DatabaseConnectionFactory factory = new MySQLDatabaseConnectionFactory();
      dummy.when(() -> DriverManager.getConnection("arg1", "arg2", "arg3"))
        .thenReturn(new Connection() {/*...*/});

      factory.getConnection();

      dummy.verify(() -> DriverManager.getConnection(eq("arg1"), eq("arg2"), eq("arg3")));
    }
  }

注意：模拟静态方法需要 mockito-inline 依赖而不是 mockito-core。

对于 JUnit5，还要添加以下内容：

 <dependency>
  <groupId>org.mockito</groupId>
  <artifactId>mockito-junit-jupiter</artifactId>
  <version>${mockito.version}</version>
  <scope>test</scope>
</dependency>

原文由 leokom 发布，翻译遵循 CC BY-SA 4.0 许可协议