Spring Boot单元测试中如何注入Mock KubernetesClient？

针对你遇到的真实KubernetesClient在测试时自动加载、setter注入mock不够优雅的问题，这里提供几个简洁可行的方案：

方案一：基于Profile条件控制真实Bean的创建 #

修改你的K8sClientConfig，给真实客户端Bean加上@Profile("!test")注解，让它只在非测试环境生效：

@Configuration
public class K8sClientConfig {
    @Bean
    @Profile("!test") // 仅当激活的profile不是test时，才创建真实KubernetesClient
    public KubernetesClient kubernetesClient() {
        return new KubernetesClientBuilder().build();
    }
}

然后在测试类中激活test profile，同时保留@EnableKubernetesMockClient——此时Spring会自动使用mock的KubernetesClient注入ProductManager，不需要再用setter替换：

@SpringBootTest(profiles = "test")
@EnableKubernetesMockClient(crud = true)
public class ProductManagerTest {

   @Autowired
   private KubernetesClient kubernetesClient;
   @Autowired
   private KubernetesMockServer server;
   @Autowired
   private ProductManager productManager;

   @Test
   public void testCreateConfigMap() {
       // 先配置mock server的预期响应
       server.expect().post().withPath("/api/v1/namespaces/default/configmaps")
             .andReturn(HttpStatus.CREATED, new ConfigMapBuilder()
                     .withNewMetadata().withName("test-cm").endMetadata()
                     .build())
             .once();

       productManager.createConfigMap();
       // 验证请求是否符合预期
       server.verify();
   }
}

方案二：用@ConditionalOnMissingBean避免Bean冲突 #

如果不想用Profile，也可以给真实的KubernetesClient Bean加上@ConditionalOnMissingBean注解，这样当测试环境中已经存在mock的KubernetesClient时，真实客户端不会被创建：

@Configuration
public class K8sClientConfig {
    @Bean
    @ConditionalOnMissingBean // 仅当没有其他KubernetesClient Bean存在时才创建
    public KubernetesClient kubernetesClient() {
        return new KubernetesClientBuilder().build();
    }
}

测试类无需额外修改Profile，只保留@EnableKubernetesMockClient即可——mock客户端会优先被注入：

@SpringBootTest
@EnableKubernetesMockClient(crud = true)
public class ProductManagerTest {

   @Autowired
   private KubernetesClient kubernetesClient;
   @Autowired
   private KubernetesMockServer server;
   @Autowired
   private ProductManager productManager;

   @Test
   public void testCreateConfigMap() {
       // 测试逻辑
   }
}

方案三：精简测试上下文（快速单元测试） #

如果不需要加载整个Spring Boot应用上下文，可以改用轻量级的测试方式，只加载ProductManager和mock配置，避免不必要的Bean初始化：

@ExtendWith(SpringExtension.class)
@EnableKubernetesMockClient(crud = true)
@ContextConfiguration(classes = {ProductManager.class}) // 仅加载必要组件
public class ProductManagerTest {

   @Autowired
   private KubernetesClient kubernetesClient;
   @Autowired
   private KubernetesMockServer server;
   @Autowired
   private ProductManager productManager;

   @Test
   public void testCreateConfigMap() {
       // 测试逻辑
   }
}

这种方式测试启动更快，也完全不会触发真实集群的连接。

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