使用Powermock工具mock系统类(java.util.Date的构造方法)

2023-05-16

场景

示例,有如下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(使用前将#替换为@)

使用Powermock工具mock系统类(java.util.Date的构造方法) 的相关文章

随机推荐