/** * 演示如何mock一个类的方法 * @sina weibo:[email protected] */ public class DateUtil { private int type; public static final String getCurrentDateStr() { SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); return sdf.format(DateUtil.now()); } public static final String getCurrentDateStrByFormatType(int type) { if (type == 1) { SimpleDateFormat sdf = new SimpleDateFormat("yyyy/MM/dd HH:mm:ss"); return sdf.format(DateUtil.now()); } else { return DateUtil.getCurrentDateStr(); } } public static final Date now() { return new Date(); } public int getType() { return type; } public void setType(int type) { this.type = type; } }
/** * 演示如何mock一个类的方法 */ public class DateUtilTest { /** * Mock某个类方法 */ @Test public void testGetCurrentDateStr() { //DateUtil.class,要Mock的类 new Expectations(DateUtil.class) { { //要Mock的方法now,其他方法DateUtil.class DateUtil.now(); //期望方法返回的结果 result = mockDate(); } }; Assert.assertEquals("2010-07-22 15:52:55", DateUtil.getCurrentDateStr()); } /** * Mock 某个类方法根据不同参数返回不同值 */ @Test public void testGetCurrentDateStrByFormatType() { new Expectations(DateUtil.class) { { DateUtil.getCurrentDateStrByFormatType(anyInt); result = new Delegate() { public String getCurrentDateStrByFormatType(int type) { if (type == 1) { return "2010/07/22 15:52:55"; } else { return "2010-07-22 15:52:55"; } } }; } }; Assert.assertEquals("2010-07-22 15:52:55", DateUtil.getCurrentDateStrByFormatType(2)); } public static Date mockDate() { Calendar c = Calendar.getInstance(); c.set(2010, 6, 22, 15, 52, 55); return c.getTime(); } }
Expectations:一个Expectations块是给定测试方法中将会涉及到的mock项中,预期将要被调用的方法或构造函数。一个Expectations可以包含多个预期的要执行方法(mock),但不必包含所有预期会被调用的方法。在Expectations中;除了可以指定预期的方法外,还可以指定方法的参数的精确值或约束行为(满足某个断言);同时Expectations中还可以指定该方法预期的返回值(如果有)或预期抛出的异常。Expectations(.class){}这种方式只会模拟区域中包含的方法,这个类的其它方法将按照正常的业务逻辑运行,上面的例子,定义了一个mock类DateUtil,同时在Expectation中定义了预期会被调用的方法now,以及now方法的返回值,这种方式还有种等价实现方式,使用@Mocked标签
@Test public void testGetCurrentDateStr(@Mocked(methods="now")DateUtil dateUtil) { //DateUtil.class,要Mock的类 new Expectations() { { //声明要Mock的方法(注:其它方法按照正常的业务逻辑运行) DateUtil.now(); //期望方法返回的结果 result = mockDate(); } }; Assert.assertEquals("2010-07-22 15:52:55", DateUtil.getCurrentDateStr()); }
NonStrictExpectations:Expectations块里声明的mock方法,是一定要被执行的,如果没有被执行,会认为整个测试case不通过;NonStrictExpectations就没有这个限制,看例子:
@Test public void testGetCurrentDateStr(@Mocked(methods="now")DateUtil dateUtil) { //DateUtil.class,要Mock的类 new NonStrictExpectations() { { //声明要Mock的方法(注:其它方法按照正常的业务逻辑运行) DateUtil.now(); //期望方法返回的结果 result = mockDate(); DateUtil.getType(); result = 1; } }; Assert.assertEquals("2010-07-22 15:52:55", DateUtil.getCurrentDateStr()); }
DateUtil.getType()在后面的断言没用被调用,但也不会出错,但是如果把NonStrictExpectations换成Expectations,就会出错,在Expectations情况必须把
DateUtil.getType(); result = 1;
给删除掉,上述就是二者的区别