在平常工作过程中，总会设计些工具类，里面通常使用静态方法来实现。那么如何来对这些静态方法进行mock，以及如何验证静态方法被调用？下面简单介绍下如何使用Powermock来实现针对静态方法的mock。
首先我们设计一个静态类如下(Utility.java)：
public class Utility {
public static <T> boolean listIsNullOrEmpty(List<T> objectList) {
return objectList == null || objectList.isEmpty();
}
public static <T> boolean listIsNotNullOrEmpty(List<T> objectList) {
return objectList != null && !objectList.isEmpty();
}
}
被测试类如下(UtilityHelper.java)：
public class UtilityHelper {
public int sum(List<Integer> dataLst) {
if (Utility.listIsNullOrEmpty(dataLst)) {
return 0;
}
int total = 0;
for (Integer data : dataLst) {
total += data;
}
return total;
}
public int product(List<Integer> 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<Integer> dataList;
@Before
public void setUp() {
PowerMockito.mockStatic(Utility.class);
dataList = new ArrayList<Integer>();
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后面加上该方法所在的类。
网友评论