@InjectMocks

初识 Mockito 这个测试框架后,我们要使用 Mock 的属性创建一个被测试类实例时,大概会下面这么纯手工来打造。

假定类 MyService 有一个属性 MyRepository myRepository:

@Repository
public class MyRepository {
 
    public void doSomething() {
        System.out.println("here's dosomething");
    }
 
    public Model findById(Long id) {
        return new Model(id, "Real Repository");
    }
}

@Service
public class MyService {
 
    @Autowired
    private MyRepository myRepository;
 
    public void doSomething() {
        this.myRepository.doSomething();
    }
 
    public Model findById(Long id) {
        return this.myRepository.findById(id);
    }
}

需要构造 MyService 实例时 Mock 内部状态:

MyRepository myRepository = Mockito.mock(MyRepository.class); 
MyService myService = new MyService(myRepository);

如果所有的 Mock 对象全部通过手工来创建,那就不容易体现出 Mockito 的优越性出来。因此对于被测试对象的创建,Mock 属性的注入应该让 @Mock 和 @InjectMocks这两个注解大显身手了。

  • @Mock:创建一个Mock。
  • @InjectMocks:创建一个实例,其余用@Mock(或@Spy)注解创建的mock将被注入到用该实例中。

@Autowird 等方式完成自动注入。在单元测试中,没有启动 spring 框架,此时就需要通过 @ InjectMocks完成依赖注入。@InjectMocks会将带有@Spy 和@Mock 注解的对象尝试注入到被 测试的目标类中。记住下面这两句话即可:

  • Usually when you are unit testing, you shouldn't initialize Spring context. So remove Autowiring.
  • Usually when you do integration testing, you should use real dependencies. So remove mocking.

所以我们可以得出如下代码:

@RunWith(MockitoJUnitRunner.class)
public class UserServiceTest {
 
    @Mock
    private MyRepository myRepository;
 
    @InjectMocks
    private MyService myService;
 
    @Test
    public void testInjectMocks() {
        System.out.println(myService.getMyRepository().getClass());
    }
}

MyService 被标记了 @InjectMocks,在 setUp方法中 执行 MockitoAnnotations.initMocks(this); 的时候,会将标记了 @Mock 或 @Spy 的属性注入到 service 中。MyService 里面的 MyRepository 完全被Mock实例替换,所有的调用都是针对Mock生成类的。

如果我们还有一个MyController如下,需要注入MyService应该怎么解决呢?

@Controller
public class MyController {
 
    @Autowired
    private MyService myService;
 
    public void doSomething() {
        this.myService.doSomething();
    }
 
    public Model findById(Long id) {
        return this.myService.findById(id);
    }
}

如果我用如下的写法:

@RunWith(MockitoJUnitRunner.class)
public class MyControllerTest {
 
    @Mock
    private MyRepository myRepository;
 
    @InjectMocks
    private MyService myService;
 
    @InjectMocks
    private MyController myController;
 
    @Before
    public void setUp() throws Exception {
        Model model = new Model(11L, "AAA");
        doNothing().when(myRepository).doSomething();
        when(myRepository.findById(11L)).thenReturn(model);
    }
 
    @Test
    public void doSomething() throws Exception {
        this.myController.doSomething();
    }
 
    @Test
    public void findById() throws Exception {
        System.out.println(this.myController.findById(11L));
    }
}

使用Mock打桩的为MyRepository,原本以为使用InjectMocks后,MyService会自动注入MyRepository,MyController会自动注入前的MyService,但是结果并不是这样的。MyController无法识别MyService。MyController实例后,没有给myService属性赋值。于是想在MyService上加个@Mock,虽然编译没问题,但是运行起来异常了:

org.mockito.exceptions.base.MockitoException: This combination of annotations is not permitted on a single field:
@Mock and @InjectMocks

所以InjectMocks字段是无法注入其他InjectMocks字段的。所以我们可以考虑使用Spring来做容器管理,修改Test类:

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {"classpath:beans.xml"})
public class MyControllerTest {
 
    @Mock
    private MyRepository myRepository;
 
    @InjectMocks
    @Autowired
    private MyService myService;
 
    @Autowired
    private MyController myController;
 
    @Before
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);
        Model model = new Model(11L, "AAA");
        doNothing().when(myRepository).doSomething();
        when(myRepository.findById(11L)).thenReturn(model);
    }
 
    @Test
    public void doSomething() throws Exception {
        this.myController.doSomething();
    }
 
    @Test
    public void findById() throws Exception {
        System.out.println(this.myController.findById(11L));
    }
}

其实不借助容器,也可以手动来赋值。在setup方法中做下修改:

@RunWith(MockitoJUnitRunner.class)
public class MyControllerTest {
 
    @Mock
    private MyRepository myRepository;
 
    @InjectMocks
    private MyService myService;
 
    @InjectMocks
    private MyController myController;
 
    @Before
    public void setUp() throws Exception {
        //通过ReflectionTestUtils注入需要的非public字段数据
        ReflectionTestUtils.setField(myController, "myService", myService);
        Model model = new Model(11L, "AAA");
        doNothing().when(myRepository).doSomething();
        when(myRepository.findById(11L)).thenReturn(model);
    }
 
    @Test
    public void doSomething() throws Exception {
        this.myController.doSomething();
    }
 
    @Test
    public void findById() throws Exception {
        System.out.println(this.myController.findById(11L));
    }
}

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 70,584评论 12 116
  • 1.1 spring IoC容器和beans的简介 Spring 框架的最核心基础的功能是IoC(控制反转)容器,...
    simoscode阅读 2,944评论 2 20
  • pdf下载地址:Java面试宝典 第一章内容介绍 20 第二章JavaSE基础 21 一、Java面向对象 21 ...
    王震阳阅读 73,101评论 25 504
  • Spring Boot 参考指南 介绍 转载自:https://www.gitbook.com/book/qbgb...
    毛宇鹏阅读 34,851评论 5 337
  • 单元测试的目标和挑战 单元测试的思路是在不涉及依赖关系的情况下测试代码(隔离性),所以测试代码与其他类或者系统的关...
    jiangmo阅读 394评论 0 2