博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
Mockito为什么不能mock静态方法
阅读量:7110 次
发布时间:2019-06-28

本文共 4122 字,大约阅读时间需要 13 分钟。

因为Mockito使用继承的方式实现mock的,用CGLIB生成mock对象代替真实的对象进行执行,为了mock实例的方法,你可以在subclass中覆盖它,而static方法是不能被子类覆盖的,所以Mockito不能mock静态方法。

但PowerMock可以mock静态方法,因为它直接在bytecode上工作。

PowerMock是一个JUnit扩展,它利用了EasyMock和Mockito模拟静态method的方法对Java中的静态method进行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); } }
  1. 如果想要对某个类的静态方法进行mock,则必须在PrepareForTest后面加上相应的类名, 比如此例的Utility.class。

  2. 在对该类的某个方法进行mock之前,还必须先对整个类进行mock:

PowerMockito.mockStatic(Utility.class);
  1. 在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的静态方法。

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

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

  1. 在test_product2中,我们看到相对test_product1来说,多了一行:
PowerMockito.spy(Utility.class);

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

  1. 我们在看下test_product3中,该方法相对test_product1中,多了一行:
PowerMockito.when(Utility.listIsNotNullOrEmpty(Mockito.anyList())).thenCallRealMethod();

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

  上述两种方式,可以根据自己的需要进行选择使用哪一种。但是一定要记得,只要使用了mockStatic某类时,该类中的所有方法就已经都默认被mock了, 在调用该类的方法时,必须根据具体方法进行相应的处理。
  另外,重要的事说三遍: 如果要mock静态方法,必须要在PrepareForTest后面加上该方法所在的类

转载地址:http://tjlhl.baihongyu.com/

你可能感兴趣的文章
教你如何修改github上的项目语言类型
查看>>
spring mvc中的几类拦截器对比
查看>>
PAI在线深度学习开发产品DSW发布
查看>>
Branckets快捷键
查看>>
赶紧登陆百家号看看,有没有被降级
查看>>
ATAC-Seq 数据分析(上)
查看>>
WPS Office 2019 上架微软商城,全新可定制 UI
查看>>
服务器宕机原因
查看>>
「mysql优化专题」视图应用竟然还可以这么优化?不得不收藏(8)
查看>>
Facebook开源看图问答模型Pythia:拿下VQA比赛冠军就靠它
查看>>
Android DownloadManager下载状态查询(2)
查看>>
结合AI与IoT技术,日本研发出可英语会话的小鸟学习型机器人
查看>>
OpenSSL密码算法库: MD5示例小程序
查看>>
[Android] 字体使用dp单位避免设置系统字体大小对排版的影响
查看>>
HDU 2037 今年暑假不AC(贪心,区间更新,板子题)
查看>>
Java系列之EJB
查看>>
EF架构~有时使用SQL更方便
查看>>
Jenkins内置环境变量的使用
查看>>
《微信生活白皮书》发布微信用户数据
查看>>
VSTO之旅系列(三):自定义Excel UI
查看>>