Mockito 框架的 @Mock 和 @InjectMocks 注解使用指南

Mockito 与 @InjectMocks 简介

Mockito 是一个流行的 Java 测试框架，用于创建和验证单元测试中的模拟对象。@InjectMocks 注解是 Mockito 的一个特性，它能够自动注入使用 @Mock 注解创建的模拟对象。

手动创建测试实例的传统方法

在没有使用 @InjectMocks 之前，开发者可能需要手动创建被测试类的实例，并为其注入模拟的依赖对象。这通常涉及到直接使用 new 关键字创建对象，并将模拟对象传递给构造函数或 setter 方法。

@Mock 和 @InjectMocks 的自动化测试实例创建

Mockito 提供了 @Mock 注解来创建模拟对象，并通过 @InjectMocks 注解自动将这些模拟对象注入到被测试类的实例中。这种方式简化了测试代码的编写，提高了测试的效率。 UserService 类和测试类的示例
 

public class UserService {
    private UserDao userDao;

    public UserService(UserDao userDao) {
        this.userDao = userDao;
    }

    public UserDao getUserDao() {
        return userDao;
    }
}

@RunWith(MockitoJUnitRunner.class)
public class UserServiceTest {
    @Mock
    private UserDao userDao;

    @InjectMocks
    private UserService testMe;

    @Test
    public void testInjectMocks() {
        System.out.println(testMe.getUserDao().getClass());
    }
}

 
上面的测试用例演示了如何使用 @Mock 和 @InjectMocks 注解来创建 UserService 的测试实例。

@InjectMocks 的工作原理

@InjectMocks 注解会根据测试类中的 @Mock 字段，自动创建被测试类的实例，并将模拟对象注入到相应的字段中。它首先尝试使用非默认构造函数，如果没有合适的构造函数，则会使用 setter 方法或直接赋值。

@InjectMocks 注解的高级特性

@InjectMocks 注解在选择注入方式时有一定的顺序：首先是构造函数注入，然后是 setter 方法注入，最后是属性注入。如果存在多个构造函数，它会根据参数的数量选择最合适的构造函数。

 

@InjectMocks 与私有构造函数和 setter 方法

即使被测试类的构造函数或 setter 方法是私有的，@InjectMocks 也能够进行注入，因为 Mockito 在注入时会忽略 Java 的访问控制。

@InjectMocks 忽略含有基本类型参数的构造函数

如果构造函数包含基本类型参数，@InjectMocks 将不会使用该构造函数进行注入，因为基本类型无法模拟。

多个相同类型的 Mock 对象的名称匹配

当存在多个相同类型的 @Mock 对象时，@InjectMocks 可以通过 @Mock 注解的 name 属性进行匹配，确保正确的模拟对象被注入到正确的字段中。

@InjectMocks 与普通对象的区别

@InjectMocks 只能注入使用 @Mock 创建的模拟对象，如果字段是一个普通对象，@InjectMocks 将无法进行注入。

@Mock 和 @InjectMocks 的赋值行为

@Mock 和 @InjectMocks 会覆盖字段的初始值，即使字段已经被赋值，最终的值也是由 Mockito 模拟的对象。

使用 @Mock 对象创建实例的错误示例

错误的使用方式是在测试类中直接使用 new 关键字创建被测试类的实例，然后再尝试使用 @Mock 对象进行注入，这会导致模拟对象没有被正确注入。

静态测试类的使用示例

使用 PowerMock 可以对静态方法进行模拟，结合 @InjectMocks 使用，可以对涉及静态方法的测试类进行测试。

@RunWith(PowerMockRunner.class)
@PrepareForTest({
    SpringContext.class, KeywordRuleCacheData.class
})
public class KeywordRuleCacheDataRefreshDealTest {
    @InjectMocks
    KeywordRuleCacheDataRefreshDeal keywordRuleCacheDataRefreshDeal;

    @Test
    public void run() throws BaseAppException {
        PowerMockito.mockStatic(SpringContext.class);
        PowerMockito.mockStatic(KeywordRuleCacheData.class);
        PowerMockito.when(KeywordRuleCacheData.refushKeywordRuleCacheData()).thenReturn(true);
        keywordRuleCacheDataRefreshDeal.run();
    }
}

上面的示例展示了如何使用 PowerMock 和 Mockito 进行静态方法的模拟测试。