Spring Boot单元测试使用MockBean注解向Service注入Mock对象

1. 背景介绍

我们在测试时有一个Service,我们需要测试Service,但Service内部依赖ServiceA、ServiceB,此时我们希望Mock ServiceA,ServiceB 注入真实对象。

java 复制代码
class Service {
    private ServiceA A;
    private ServiceB B;
     
    public int methodA() { return A.a();}
    public int methodB() { return B.b();}
}

2.使用@MockBean注解来模拟依赖对象

在Spring Boot Test中使用@Spy注解标记Service对象,但是其余真实对象无法注入,因为@Spy注解只能部分模拟对象,而无法注入真实对象。在这种情况下,可以使用@Autowired注解来自动注入Service对象,并使用@MockBean注解来模拟依赖对象。

java 复制代码
@RunWith(SpringRunner.class)
@SpringBootTest
public class ServiceTest {

    @Autowired
    private Service service;

    @MockBean
    private ServiceA serviceA;

    @Test
    public void testMethodA() {
        // 模拟UserDao的部分方法
        Mockito.doReturn(1).when(serviceA).a(Mockito.any());

        // 调用UserService的方法
        int s = service.methodA();

        // 验证结果
        assertEquals(1, s);
    }

    @Test
    public void testOtherMethod() {
        // 使用真实的方法
        service.methodB();
        // 验证结果
        // ...
    }
}

我们使用了@Autowired注解来自动注入Service对象,并使用@MockBean注解来模拟serviceA对象。在testMethodA方法中,我们使用Mockito.doReturn方法来模拟ServiceA对象的部分方法,并调用userService.methodA方法进行测试。在testOtherMethod方法中,我们直接调用userService.otherMethod方法进行测试。

在使用Mockito进行单元测试时,可以使用@MockBean注解来模拟依赖对象,并使用@Autowired注解来自动注入需要测试的对象。这样,就可以在Spring上下文中同时使用模拟对象和真实对象,并进行更加全面和准确的测试

3. 总结

@MockBean注解是Spring Boot提供的一个注解,用于模拟依赖对象。它的作用是在Spring上下文中创建一个模拟对象,并将其注入到被测试对象中,以便进行单元测试。

相关推荐
程序员二黑9 分钟前
状态迁移与场景法:搞定复杂业务流测试的利器
面试·单元测试·测试
问道飞鱼1 小时前
【微服务组件】Springboot结合Dubbo实现RPC调用
spring boot·微服务·rpc·dubbo
I'm Jie1 小时前
(二)Gradle 依赖仓库及安全凭证配置
java·spring boot·spring·gradle·maven
李少兄2 小时前
记一次 Spring Boot 项目中 Redis 工具类的重构实践
spring boot·redis·重构
摇滚侠4 小时前
Spring Boot3零基础教程,生命周期启动加载机制,笔记64
spring boot·笔记
摇滚侠4 小时前
Spring Boot3零基础教程,整合 Redis,笔记69
spring boot·redis·笔记
bug攻城狮5 小时前
Spring Boot 2.6+ 整合 PageHelper 启动报错:循环依赖解决方案全解析
java·spring boot·后端
摇滚侠6 小时前
Spring Boot3零基础教程,生命周期监听,自定义监听器,笔记59
java·开发语言·spring boot·笔记
摇滚侠6 小时前
Spring Boot3零基础教程,事件驱动开发,设计登录成功后增加积分记录信息功能,笔记61
java·spring boot·笔记·后端
这周也會开心7 小时前
SpringBoot的搭建方式
java·spring boot·后端