Mockito中模拟静态方法-Toy模板网

这篇具有很好参考价值的文章主要介绍了Mockito中模拟静态方法。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

Mockito中模拟静态方法

背景

在项目实际开发中，编写单元测试用例时，需要对静态方法进行模拟，本次文章就简单整理下如何使用Mockito来模拟静态方法。

添加依赖

<dependency>
  <groupId>org.mockito</groupId>
  <artifactId>mockito-inline</artifactId>
  <version>3.8.0</version>
</dependency>

测试类

测试类中有两个方法，一个带参数，一个不带参数

static class StaticUtils {
  static String getName() {
    return "yuan";
  }

  static int add(int x, int y) {
    return x + y;
  }
}

静态方法模拟

在Mockito中可以使用Mockito.mockStatic( Class classToMock )方法来模拟对静态方法的调用，它会返回一个MockedStatic类型的模拟对象，该对象继承了AutoCloseable接口，需要手动释放资源。下面是具体的使用：

无参方法

@Test
public void testStaticWithNoArgs() {
  // 无参方法模拟
  try (final MockedStatic<StaticUtils> staticUtilsMockedStatic = Mockito.mockStatic(StaticUtils.class)) {
    staticUtilsMockedStatic.when(StaticUtils::getName).thenReturn("wu");
    Assert.assertEquals(StaticUtils.getName(), "wu");
  }
}

有参方法

@Test
public void testStaticWithArgs() {
  // 有参方法模拟
  try (MockedStatic<StaticUtils> staticUtilsMockedStatic = Mockito.mockStatic(StaticUtils.class)) {
    staticUtilsMockedStatic.when(() -> StaticUtils.add(Mockito.anyInt(), Mockito.anyInt())).thenReturn(55);
    Assert.assertEquals(StaticUtils.add(1,3), 52);
  }
}