初识 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));
}
}
