场景
示例,有如下DateFormat的formatCurrentTime()方法,代码如下:
public class DateFormat {
public static final SimpleDateFormat SIMPLE_DATE_FORMAT = new SimpleDateFormat("yyMMddHHmmssZ");
public static String formatCurrentTime() {
return SIMPLE_DATE_FORMAT.format(new Date());
}
}
我现在期望判断formatCurrentTime方法返回的值是否是SIMPLE_DATE_FORMAT对象格式化的值(当然了,这代码一看肯定是了,不要觉得这没意义,实际工作中真的有类似无聊而又什么什么的需求)。
为了满足这个要求,我只需要判断SIMPLE_DATE_FORMAT.format(new Date())是否等于DateFormat.formatCurrnetTime()方法调用的返回值就行了。但是很明显,formatCurrnetTime方法内部调用SIMPLE_DATE_FORMAT.format方法时传入的参数直接new了一个Date对象,我外部显式调用SIMPLE_DATE_FORMAT.format方法的时候又再创建的Date对象肯定不是这一个了,所以我需要mock java.util.Date类的无参构造方法,保证每次new的时候返回的是同一个对象。
测试代码如下:
实现
@RunWith(PowerMockRunner.class)
@PrepareForTest(DateFormat.class)
public class DateFormatTest {
@Before
public void setup() {
PowerMockito.mockStatic(Date.class);
}
@Test
public void testFormatCurrentTime() throws Exception {
Date date = new Date();
PowerMockito.whenNew(Date.class).withNoArguments().thenReturn(date);
Assert.assertEquals(DateFormat.SIMPLE_DATE_FORMAT.format(date), DateFormat.formatCurrentTime());
}
}
如上代码:
说明下对于java.util.Date类这种JDK的类的mock操作过程,这里使用Powermock,关于Powermock的配置可以看Mockito配合powermock工具mock构造函数这篇文章。
步骤如下:
1. 测试类上使用注解@RunWith(PowerMockRunner.class)
2. 注解@PrepareForTest声明调用系统类的类:上面自己定义的DateFormat类
3. 调用mockStatic方法mock Date类(调用PowerMockito的mock方法也可以,毕竟这里需求只是mock 构造方法,我使用mockStatic,是因为看了它的维基上的介绍,地址在这:https://github.com/powermock/powermock/wiki/Mock-System)。
最后根据这里的需要设置Date的无参构造方法的期望返回值,mock构造方法的api使用可以看Mockito配合powermock工具mock构造函数这篇文章了解下。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)