强大的Mockito测试框架
编程技术 / houtizong 发布于 3年前 57
一.自动生成Mock类
在需要Mock的属性上标记@Mock注解，然后@RunWith中配置Mockito的TestRunner或者在setUp()方法中显示调用MockitoAnnotations.initMocks(this);生成Mock类即可。
二.自动注入Mock类到被测试类
只要在被测试类上标记@InjectMocks，Mockito就会自动将标记@Mock、@Spy等注解的属性值注入到被测试类中。
import static org.mockito.Mockito.when;import java.util.Collections;import java.util.List; import javax.annotation.Resource; import org.junit.Assert;import org.junit.Before;import org.junit.Test;import org.junit.runner.RunWith;import org.mockito.InjectMocks;import org.mockito.Mock;import org.mockito.MockitoAnnotations;importorg.springframework.test.context.ContextConfiguration;import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; import com.cdai.ssi.user.dao.UserDao;import com.cdai.ssi.user.domain.UserDomain;import com.cdai.ssi.user.dto.UserDto;importcom.cdai.ssi.user.service.UserService; @RunWith(SpringJUnit4ClassRunner.class)@ContextConfiguration({"classpath:spring/spring-config.xml"})public class UserServiceTest { @InjectMocks@Resource(name= "userService")privateUserService userService;@MockprivateUserDao userDao;@Beforepublicvoid setUp() { MockitoAnnotations.initMocks(this);}@Testpublicvoid testQueryAll() { when(userDao.selectAll()).thenReturn(Collections.<UserDomain>emptyList()); List<UserDto>dtoList = userService.queryAll(); Assert.assertTrue(dtoList.isEmpty());} } @InjectMocks的问题是：如果被测试类是代理类，那么注入会失效。比如上面的UserService如果是事务或者其他AOP代理类，那么进入@Test方法时UserService中的DAO属性不会被Mock类替换。
三.有些方法想Mock定制，有些想调用真实方法
因为@Mock针对接口生成Mock类，所以我们是没法调用到真实的实现类的方法。可以使用@Spy注解标注属性，并且标注@Resource注解让Spring注入真实实现类，那么Mockito就会自动生成Spy类。
例如：
@Resource(name = "userService")privateUserService userService;@Spy@ResourceprivateUserDao userDao;
Spy类就可以满足我们的要求。如果一个方法定制了返回值或者异常，那么就会按照定制的方式被调用执行；如果一个方法没被定制，那么调用的就是真实类的方法。
如果我们定制了一个方法A后，再下一个测试方法中又想调用真实方法，那么只需在方法A被调用前，调用Mockito.reset(spyObject);就行了。
import static org.mockito.Mockito.when; import org.mockito.Mockito; public class TestMockObject implementsITestMock { publicstatic void main(String[] args) { ITestMockmock = Mockito.mock(TestMockObject.class); System.out.println(mock.test1()); System.out.println(mock.test2()); ITestMockspy = Mockito.spy(new TestMockObject()); System.out.println(spy.test1()); System.out.println(spy.test2()); when(spy.test1()).thenReturn(100); System.out.println(spy.test1()); Mockito.reset(spy); System.out.println(spy.test1()); System.out.println(spy.test2()); when(spy.test1()).thenReturn(104); System.out.println(spy.test1()); } @Override publicint test1() { System.out.print("RealTest1()!!! - "); return1; } @Override publicint test2() { System.out.print("RealTest2()!!! - "); return2; }} 输出为：
00Real Test1()!!! - 1Real Test2()!!! - 2Real Test1()!!! - 100Real Test1()!!! - 1Real Test2()!!! - 2Real Test1()!!! - 104
请勿发布不友善或者负能量的内容。与人为善，比聪明更重要！
技术博客集 - 网站简介：
前后端技术：
后端基于Hyperf2.1框架开发,前端使用Bootstrap可视化布局系统生成
网站主要作用：
1.编程技术分享及讨论交流，内置聊天系统;
2.测试交流框架问题，比如：Hyperf、Laravel、TP、beego;
3.本站数据是基于大数据采集等爬虫技术为基础助力分享知识，如有侵权请发邮件到站长邮箱，站长会尽快处理;
4.站长邮箱：[email protected];
文章归档
文章标签
友情链接
