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上下文中创建一个模拟对象,并将其注入到被测试对象中,以便进行单元测试。

相关推荐
仰望星空@脚踏实地1 小时前
Spring Boot Web 服务单元测试设计指南
spring boot·后端·单元测试
一勺菠萝丶3 小时前
Spring Boot + MyBatis/MyBatis Plus:XML中循环处理List参数的终极指南
xml·spring boot·mybatis
RainbowSea4 小时前
问题:后端由于字符内容过长,前端展示精度丢失修复
java·spring boot·后端
风象南4 小时前
SpringBoot 控制器的动态注册与卸载
java·spring boot·后端
我是一只代码狗5 小时前
springboot中使用线程池
java·spring boot·后端
hello早上好5 小时前
JDK 代理原理
java·spring boot·spring
PanZonghui5 小时前
Centos项目部署之运行SpringBoot打包后的jar文件
linux·spring boot
沉着的码农6 小时前
【设计模式】基于责任链模式的参数校验
java·spring boot·分布式
zyxzyx6666 小时前
Flyway 介绍以及与 Spring Boot 集成指南
spring boot·笔记
一头生产的驴7 小时前
java整合itext pdf实现自定义PDF文件格式导出
java·spring boot·pdf·itextpdf