java mock静态方法_如何使用Powermock对静态方法进行mock

在平常工作过程中，总会设计些工具类，里面通常使用静态方法来实现。那么如何来对这些静态方法进行mock，以及如何验证静态方法被调用？下面简单介绍下如何使用Powermock来实现针对静态方法的mock。

首先我们设计一个静态类如下(Utility.java)：

public class Utility {

public static boolean listIsNullOrEmpty(List objectList) {

return objectList == null || objectList.isEmpty();

}

public static boolean listIsNotNullOrEmpty(List objectList) {

return objectList != null && !objectList.isEmpty();

}

}

被测试类如下(UtilityHelper.java)：

public class UtilityHelper {

public int sum(List dataLst) {

if (Utility.listIsNullOrEmpty(dataLst)) {

return 0;

}

int total = 0;

for (Integer data : dataLst) {

total += data;

}

return total;

}

public int product(List dataList) {

int total = 1;

if (Utility.listIsNotNullOrEmpty(dataList)) {

for (Integer data : dataList) {

total *=data;

}

}

return total;

}

}

在被测试类中分别定义了两个方法，分别调用了Utility类里面的两个静态方法，下面我们通过对这两个方法进行测试，来介绍下使用Powermock对静态方法进行mock的各种用法。

测试类如下(UtilityHelperTest.java)：

@RunWith(PowerMockRunner.class)

@PrepareForTest({Utility.class})

public class UtilityHelperTest {

private UtilityHelper utilityHelper;

private List dataList;

@Before

public void setUp() {

PowerMockito.mockStatic(Utility.class);

dataList = new ArrayList();

dataList.add(1);

dataList.add(2);

dataList.add(3);

utilityHelper = new UtilityHelper();

}

@Test

public void testSum_1() {

PowerMockito.when(Utility.listIsNullOrEmpty(Mockito.anyList())).thenReturn(true);

int sum = utilityHelper.sum(dataList);

Assert.assertEquals(0, sum);

PowerMockito.verifyStatic(Mockito.times(1));

Utility.listIsNullOrEmpty(Mockito.anyList());

}

@Test

public void testSum_2() {

PowerMockito.when(Utility.listIsNullOrEmpty(Mockito.anyList())).thenReturn(false);

int sum = utilityHelper.sum(dataList);

Assert.assertEquals(6, sum);

}

@Test

public void testProduct_1() {

int product = utilityHelper.product(dataList);

Assert.assertEquals(1, product);

}

@Test

public void testProduct_2() {

PowerMockito.spy(Utility.class);

int product = utilityHelper.product(dataList);

Assert.assertEquals(6, product);

}

@Test

public void testProduct_3() {

PowerMockito.when(Utility.listIsNotNullOrEmpty(Mockito.anyList())).thenCallRealMethod();

int product = utilityHelper.product(dataList);

Assert.assertEquals(6, product);

}

}

如果想要对某个类的静态方法进行mock，则必须在PrepareForTest后面加上相应的类名, 比如此例的Utility.class。

在对该类的某个方法进行mock之前，还必须先对整个类进行mock：

PowerMockito.mockStatic(Utility.class);

在testSum_1方法中，我们对listIsNullOrEmpty进行了mock：

PowerMockito.when(Utility.listIsNullOrEmpty(Mockito.anyList())).thenReturn(true);

可以看到虽然入参非空，但是由于返回值返回了true，使得调用sum方法返回的值是0。

另外，如果我们想要验证某静态方法是否被调用，或者被调用了几次，我们可以用如下方式验证：

PowerMockito.verifyStatic(Mockito.times(1));

Utility.listIsNullOrEmpty(Mockito.anyList());

先使用verifyStatic方法表明要验证静态方法，可以带参数，也可以不带参数，其参数可以使用Mockito的times方法或never方法来表示其调用次数。下面紧跟着的一行则表示要验证的是哪个已经mock的静态方法。

在test_sum2方法中，由于我们mock的返回值为false，所以调用sum方法返回的是实际值。

在test_product1中，我们可以看到并没有对product中调用的listIsNotNullOrEmpty进行mock，那么为什么返回值是 1 呢？

这个主要是因为我们在setup方法中对使用mockStatic方法对Utility.class进行了mock，那么此时该类中的所有方法实际上都已经被mock了，如果没有对某个方法进行具体mock返回值，则调用该方法时，会直接返回对应返回类型的默认值，并不会执行真正的方法。此例对于listIsNotNullOrEmpty方法来说，其返回类型为boolean型，其默认值为false，所以product方法返回 1 。

那么如果我们想对已经mock的类的某个方法调用真实的方法，而不是调用mock方法，那么该如何处理呢？此处我们介绍两种实现：

在test_product2中，我们看到相对test_product1来说，多了一行：

PowerMockito.spy(Utility.class);

加了上面一行后，虽然也没有对listIsNotNullOrEmpty进行mock，但此时返回值是真正的值，说明没有调用默认的mock方法。使用spy后，虽然已经对该类做了mockStatic处理，但此时该类中的所有方法仍然都会调用真实的方法，而不是默认的mock方法。这种用法主要适用于只想要对某个类的少量方法进行mock，其他方法仍然执行真正的方法，平常写时，可以紧跟在mockStatic方法后。

我们在看下test_product3中，该方法相对test_product1中，多了一行：

PowerMockito.when(Utility.listIsNotNullOrEmpty(Mockito.anyList())).thenCallRealMethod();

此行的含义就是调用到mock类的该方法执行真正的方法，而不是mock方法。这种方式就是需要对每个要执行的方法都要进行相应的mock处理。

上述两种方式，可以根据自己的需要进行选择使用哪一种。但是一定要记得，只要使用了mockStatic某类时，该类中的所有方法就已经都默认被mock了， 在调用该类的方法时，必须根据具体方法进行相应的处理。

另外，重要的事说三遍： 如果要mock静态方法，必须要在PrepareForTest后面加上该方法所在的类。