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

最新推荐文章于 2025-07-25 17:19:50 发布

FserSuN

最新推荐文章于 2025-07-25 17:19:50 发布

阅读量7.5k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：工程质量文章标签： spring boot 单元测试 log4j

本文链接：https://blog.youkuaiyun.com/Revivedsun/article/details/132259624

工程质量专栏收录该内容

9 篇文章

订阅专栏

本文介绍了如何在SpringBoot测试中使用@MockBean注解模拟ServiceA和服务B的依赖，结合@Autowired自动注入Service，以及如何利用Mockito进行方法模拟，实现全面的单元测试。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 背景介绍

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

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注解来模拟依赖对象。

@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上下文中同时使用模拟对象和真实对象，并进行更加全面和准确的测试