Spring单元测试中，如何Mock被@Primary注解标记的类？

你遇到的核心问题很明确：测试时Spring容器里同时存在两个标记了@Primary的EntityService Bean——原项目里的EntityServiceImpl，还有你在测试配置中新定义的Bean，导致容器无法确定注入哪一个，进而报错。下面给你几个实用的解决方案，按推荐程度排序：

1. 用@MockBean直接替换（最推荐） #

Spring Boot Test提供的@MockBean注解可以自动帮你替换容器中对应的Bean，完全不需要额外写配置类，还能彻底避免冲突。

直接在测试类里这么写：

// 如果是测试Web控制器，用@WebMvcTest只加载必要的Bean，测试更轻量
@WebMvcTest(EntityController.class)
public class EntityControllerTest {

    @Autowired
    private MockMvc mockMvc;

    // 用@MockBean标记后，Spring会自动把容器里所有的EntityService实现（包括@Primary的那个）替换成这个Mock对象
    @MockBean
    private EntityService entityService;

    @Test
    void testControllerMethod() {
        // 先给Mock对象设置预期行为
        when(entityService.someMethod(any())).thenReturn(expectedResult);

        // 执行接口测试
        mockMvc.perform(get("/your-endpoint"))
               .andExpect(status().isOk())
               .andExpect(content().json(expectedJson));
    }
}

这个方法不用修改任何生产代码，也不用维护测试配置类，Spring会帮你处理Bean的替换逻辑，是最省心的方案。

2. 在测试配置中排除原有的@Primary Bean #

如果你必须用自定义测试配置类，可以通过排除原有的EntityServiceImpl来解决冲突。比如结合@SpringBootTest和@EnableAutoConfiguration：

@SpringBootTest
// 指定测试配置类
@ContextConfiguration(classes = EntityControllerTestConfig.class)
// 排除原有的@Primary的EntityServiceImpl，避免冲突
@EnableAutoConfiguration(exclude = EntityServiceImpl.class)
public class EntityControllerTest {

    @Autowired
    private EntityController controller;

    // 这里注入的就是测试配置里的Mock对象
    @Autowired
    private EntityService entityService;

    // 测试方法...
}

@Configuration
class EntityControllerTestConfig {
    // 这里不用加@Primary，因为原有的实现类已经被排除了
    @Bean
    public EntityService entityService() {
        return Mockito.mock(EntityService.class);
    }
}

也可以在测试配置类的@ComponentScan里排除原实现类：

@Configuration
// 扫描项目包，但排除掉EntityServiceImpl
@ComponentScan(basePackages = "com.your.project.package", 
               excludeFilters = @ComponentScan.Filter(type = FilterType.ASSIGNABLE_TYPE, classes = EntityServiceImpl.class))
class EntityControllerTestConfig {
    @Bean
    @Primary
    public EntityService entityService() {
        return Mockito.mock(EntityService.class);
    }
}

3. 用@Qualifier指定Bean（不推荐） #

这个方法需要修改生产代码的控制器，在@Autowired上加上@Qualifier指定测试时要注入的Bean，除非万不得已不建议这么做。示例：

// 生产代码的控制器
@RestController
public class EntityController {

    // 加上@Qualifier指定Bean名称
    @Autowired
    @Qualifier("testEntityService")
    private EntityService entityService;

    // 控制器方法...
}

// 测试配置类
@Configuration
class EntityControllerTestConfig {
    @Bean("testEntityService")
    public EntityService entityService() {
        return Mockito.mock(EntityService.class);
    }
}

这种方式破坏了生产代码的简洁性，所以优先推荐前两种方案。

内容的提问来源于stack exchange，提问作者DamDam