其他分享
首页 > 其他分享> > mockito的常规使用

mockito的常规使用

作者:互联网

简介

  在实际的开发中,单元测试是不可避免的一环。很多时候针对复杂的类依赖和调用关系的场景(某个方法可能会层层调用其他类的方法,且逻辑复杂),造数据去测这种场景的代码,是非常难受且难以全覆盖的。针对这样的问题,mockito的工具可以很好的解决。mockito工具可以只针对测试你当前测试的方法的主要平层逻辑,而不用去关注当前方法的内部调用的其他方法(当前私有方法还不能实现)。因为通过mockito可以给内部调用的方法创建一个虚拟对象,根据测试的需要返回需要的数据,不用真正执行调用其他方法的真正逻辑。

常用的注解

  @injectMock 

  主要是用于来创建你需要测试的对象,

  @Mock 

  主要是用来创建在测试对象中使用的对象,也就是虚拟的对象。

  @Spy

  主要是用来创建在测试对象中使用的对象,它是真实的对象。

       不管是@Mock还是,@Spy是不受spring托管的。

  

demo 1 在测试的方法内调用其他类的方法

  先建两个service层的类 CompanyServiceImpl和PersonServiceImpl,PersonServiceImpl.getUserAllInfo()方法会调用CompanyServiceImpl.getCompanyInfo();

  

package bear.tw.service;

public class PersonServiceImpl {

    private CompanyServiceImpl companyService = new CompanyServiceImpl();

    public void getUserAllInfo(Long userId){
        String companyNameByUserId = companyService.getCompanyNameByUserId(userId);
        System.out.println(companyNameByUserId);
        System.out.println("current userId is:" + userId);
    }
}

 

  

 

package bear.tw.service;

public class CompanyServiceImpl {

    public String getCompanyNameByUserId(Long userId){
        String name = "real company ";
        return name;
    }


}

 

  单元测试类

package bear.tw;

import bear.tw.service.CompanyServiceImpl;
import bear.tw.service.PersonServiceImpl;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;
import static org.mockito.Mockito.*;
;

@RunWith(MockitoJUnitRunner.class)
public class PersonServiceTest {

    @InjectMocks
    private PersonServiceImpl personService;

    @Mock
    private CompanyServiceImpl companyService;


    @Test
    public void getUserAllInfo(){
        when(companyService.getCompanyNameByUserId(anyLong())).thenReturn("virtual company 1");
        personService.getUserAllInfo(1l);
    }


}

 



  测试方法getUserAllInfo的结果是:

  根据截图的结果可知 companyService.getCompanytInfo的执行结果是测试方法的里设置的 虚拟值,而不是真实的执行结果。表明调用的是虚拟的对象方法

demo2 当前测试方法调用同类的其他方法(将被调用的方法虚拟为一个方法)

在PersonServiceImpl中添加另外一个方法 countUser ,然后在getUserAllInfo()中调用

package bear.tw.service;

public class PersonServiceImpl {

    private CompanyServiceImpl companyService = new CompanyServiceImpl();

    public void getUserAllInfo(Long userId){
        String companyNameByUserId = companyService.getCompanyNameByUserId(userId);
        System.out.println(companyNameByUserId);
        System.out.println("current userId is:" + userId);

        System.out.println(countUser());

    }

    public int countUser(){
        return  1;
    }
}

  然后修改测试类

  

@Test
    public void getUserAllInfo2(){
        when(companyService.getCompanyNameByUserId(anyLong())).thenReturn("virtual company 2");
        PersonServiceImpl spy = spy(personService);
        doReturn(2).when(spy).countUser();
        spy.getUserAllInfo(1l);
    }

  测试结果

 

   从测试结果看,companyService对象的虚拟对象成功了, PersonServiceImpl的countUser方法的虚拟对象也执行成功了。

 

标签:companyService,PersonServiceImpl,CompanyServiceImpl,mockito,userId,常规,使用,public
来源: https://www.cnblogs.com/whltw/p/15412161.html