单元测试之@Mock与@InjectMocks(一文搞懂)-Toy模板网

这篇具有很好参考价值的文章主要介绍了单元测试之@Mock与@InjectMocks(一文搞懂)。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

@Mock与@InjectMocks一般搭配组合使用，是单元测试必不可少的注解

@Mock：需要模拟的类，我们需要模拟哪些类，就用它修饰哪些类的变量，常用于第三方服务service
@InjectMocks：要测试的类，使用@Mock修饰的对象，就是我们测试哪个类，就用它修饰对应的变量，会整合使用@Mock修饰的对象

直接上代码文章来源地址https://www.toymoban.com/news/detail-740934.html

@Service
public class ThirdService {

    public Object getThirdUser(String userId) {
        return new Object();
    }
}

@Service
public class UserService {

    @Autowired
    private ThirdService thirdService;

    public Object getUser(String userId) {
        return thirdService.getThirdUser(userId);
    }
}

//测试UserService
@SpringBootTest
class UserServiceTest {

    //需要模拟的类(因为UserService中使用了这个类)
    @Mock
    private ThirdService thirdService;

    //要测试的类，使用@Mock修饰的对象
    //这时候userService对象中持有的thirdService变量就是模拟的对象了
    @InjectMocks
    private UserService userService;

    @Test
    public void testGetUser() {
        //设定行为返回数据，我们可以设定模拟对象的行为，当然也可以不设定
        //any代表任意参数，返回a对象
        Object a = new Object();
        when(thirdService.getThirdUser(any())).thenReturn(a);

        //执行被测试的方法，这时候内部调用的ThirdService就是模拟对象了，行为结果就是上面设置的
        Object abc = userService.getUser("abc");

        //验证结果
        assertEquals(a,abc);
    }

}