强大的Mockito测试框架

最新推荐文章于 2024-04-25 23:47:22 发布

ccecwg

最新推荐文章于 2024-04-25 23:47:22 发布

阅读量586

点赞数

分类专栏： Mockito

Mockito 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

本文详细介绍了如何使用Mockito框架进行Java单元测试，包括生成Mock类、自动注入Mock类到被测试类、Mock方法定制以及如何在不同场景下灵活运用Mockito进行测试。同时，还讨论了如何在测试类中使用@Mock、@InjectMocks、@Spy等注解，并通过实例展示了如何在测试中模拟和验证业务逻辑。

摘要由CSDN通过智能技术生成

1自动生成Mock类

在需要Mock的属性上标记@Mock注解，然后@RunWith(MockitoJUnitRunner.class)或者在setUp()方法中显示调用MockitoAnnotations.initMocks(this);生成Mock类即可。

2自动注入Mock类到被测试类

只要在被测试类上标记@InjectMocks，Mockito就会自动将标记@Mock、@Spy等注解的属性值注入到被测试类中。

[java]view plaincopy 
    
 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;  
           
          @Mock  
          privateUserDao userDao;  
    
          @Before  
          publicvoid setUp() {  
                    MockitoAnnotations.initMocks(this);  
          }  
           
          @Test  
          publicvoid testQueryAll() {  
                    when(userDao.selectAll()).  
                             thenReturn(Collections.<UserDomain>emptyList());  
                     
                    List<UserDto>dtoList = userService.queryAll();  
                     
                    Assert.assertTrue(dtoList.isEmpty());  
          }  
           
 }  

@InjectMocks的问题是：如果被测试类是代理类，那么注入会失效。比如上面的UserService如果是事务或者其他AOP代理类，那么进入@Test方法时UserService中的DAO属性不会被Mock类替换。

3 Mock方法定制再也不用录制、播放了

Mockito的Mock方法定制可读性很强，而且也不需要像EasyMock那样录制播放，定制后就可以使用。

例如：

when(userDao.selectAll()).

thenReturn(Collections.<UserDomain>emptyList());

4有些方法想Mock定制，有些想调用真实方法

因为@Mock针对接口生成Mock类，所以我们是没法调用到真实的实现类的方法。可以使用@Spy注解标注属性，并且标注@Resource注解让Spring注入真实实现类，那么Mockito就会自动生成Spy类。

例如：

@InjectMocks

@Resource(name ="userService")

privateUserService userService;

@Spy

@Resource

privateUserDao userDao;

Spy类就可以满足我们的要求。如果一个方法定制了返回值或者异常，那么就会按照定制的方式被调用执行；如果一个方法没被定制，那么调用的就是真实类的方法。

如果我们定制了一个方法A后，再下一个测试方法中又想调用真实方法，那么只需在方法A被调用前，调用Mockito.reset(spyObject);就行了。

[java]view plaincopy 
    
 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;  
          }  
    
 }  