如何在SpringBoot测试中添加仅特定测试可见的REST控制器路由？

当然可以！你完全可以在Spring Boot的测试代码里定义只用于测试的API端点，还能精准控制这些端点只在特定测试类中生效。下面一步步给你拆解实现方式：

第一步：创建测试专用的端点控制器 #

首先把你的HelloAPI控制器从src/main/java目录移到src/test/java目录下——这样它就只会存在于测试代码中，绝对不会被打包到生产环境里：

@RestController
public class HelloAPI {
    @GetMapping("/hello")
    public String ok() {
        return "world";
    }
}

不过这时候有个小问题：这个Bean会对所有标注了@SpringBootTest的测试类可见，如果我们只想让它在特定测试中生效，就需要做进一步的限制。

第二步：限制端点到特定测试类 #

这里给你推荐两种实用的方法，按需选择即可：

方法1：用@TestConfiguration + @Import #

先创建一个专门的测试配置类，用来注册这个测试控制器：

@TestConfiguration
public class TestApiConfig {
    @Bean
    public HelloAPI helloAPI() {
        return new HelloAPI();
    }
}

然后只在需要用到这个端点的测试类上，通过@Import引入这个配置类：

@SpringBootTest
@Import(TestApiConfig.class)
public class HelloApiTest {
    @Autowired
    private TestRestTemplate restTemplate;

    @Test
    void testHelloEndpoint() {
        String response = restTemplate.getForObject("/hello", String.class);
        assertEquals("world", response);
    }
}

这样一来，只有添加了@Import(TestApiConfig.class)的测试类，它的ApplicationContext才会加载HelloAPI Bean；其他@SpringBootTest测试类完全不会感知到这个端点的存在。

方法2：直接用@ContextConfiguration指定类 #

如果不想额外写配置类，也可以直接在目标测试类上用@ContextConfiguration指定要加载的HelloAPI类：

@SpringBootTest
@ContextConfiguration(classes = {HelloAPI.class})
public class HelloApiTest {
    @Autowired
    private TestRestTemplate restTemplate;

    @Test
    void testHelloEndpoint() {
        String response = restTemplate.getForObject("/hello", String.class);
        assertEquals("world", response);
    }
}

这种方式更简洁，直接把测试控制器作为配置类传入，同样能达到“仅该测试类加载此端点”的效果。

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