Spring Boot 测试策略：构建高质量的测试体系

Spring Boot 测试策略：构建高质量的测试体系

引言

在现代软件开发中，测试是确保应用质量的关键环节。Spring Boot 提供了丰富的测试支持，包括单元测试、集成测试、端到端测试等。本文将深入探讨 Spring Boot 的测试策略，介绍各种测试类型的最佳实践和实现方法。

一、测试类型概述

1.1 测试金字塔

┌─────────────────────────────────────────────────────────┐ │ 端到端测试 (E2E) │ │ 少量但覆盖核心业务流程 │ ├─────────────────────────────────────────────────────────┤ │ 集成测试 │ │ 中等数量，验证组件协作 │ ├─────────────────────────────────────────────────────────┤ │ 单元测试 │ │ 大量，测试单个组件 │ └─────────────────────────────────────────────────────────┘

1.2 测试类型对比

二、单元测试

2.1 基础单元测试

@SpringBootTest class UserServiceTest { @MockBean private UserRepository userRepository; @Autowired private UserService userService; @Test void findById_shouldReturnUser() { // 准备 String userId = "1"; User expectedUser = new User(userId, "John", "john@example.com"); Mockito.when(userRepository.findById(userId)) .thenReturn(Optional.of(expectedUser)); // 执行 User actualUser = userService.findById(userId); // 验证 Assertions.assertEquals(expectedUser.getId(), actualUser.getId()); Assertions.assertEquals(expectedUser.getName(), actualUser.getName()); Mockito.verify(userRepository).findById(userId); } @Test void findById_shouldThrowException_whenUserNotFound() { // 准备 String userId = "999"; Mockito.when(userRepository.findById(userId)) .thenReturn(Optional.empty()); // 执行 & 验证 Assertions.assertThrows(UserNotFoundException.class, () -> { userService.findById(userId); }); Mockito.verify(userRepository).findById(userId); } }

2.2 参数化测试

@SpringBootTest class CalculatorServiceTest { @Autowired private CalculatorService calculatorService; @ParameterizedTest @CsvSource({ "10, 5, 15", "20, 10, 30", "-5, 5, 0", "0, 0, 0" }) void add_shouldReturnCorrectSum(int a, int b, int expected) { int result = calculatorService.add(a, b); Assertions.assertEquals(expected, result); } @ParameterizedTest @ValueSource(strings = {"valid@example.com", "test@domain.org", "user123@mail.com"}) void isValidEmail_shouldReturnTrue_forValidEmails(String email) { boolean result = calculatorService.isValidEmail(email); Assertions.assertTrue(result); } }

2.3 使用 MockMvc 测试控制器

@WebMvcTest(UserController.class) class UserControllerTest { @Autowired private MockMvc mockMvc; @MockBean private UserService userService; @Test void getUser_shouldReturnUser() throws Exception { // 准备 User user = new User("1", "John", "john@example.com"); Mockito.when(userService.findById("1")) .thenReturn(user); // 执行 & 验证 mockMvc.perform(get("/api/users/1")) .andExpect(status().isOk()) .andExpect(jsonPath("$.id").value("1")) .andExpect(jsonPath("$.name").value("John")) .andExpect(jsonPath("$.email").value("john@example.com")); } @Test void getUser_shouldReturnNotFound_whenUserNotExists() throws Exception { Mockito.when(userService.findById("999")) .thenThrow(new UserNotFoundException("User not found")); mockMvc.perform(get("/api/users/999")) .andExpect(status().isNotFound()); } }

三、集成测试

3.1 数据库集成测试

@SpringBootTest @TestMethodOrder(MethodOrderer.OrderAnnotation.class) class UserRepositoryIntegrationTest { @Autowired private UserRepository userRepository; @Test @Order(1) void save_shouldPersistUser() { // 准备 User user = new User(); user.setName("Test User"); user.setEmail("test@example.com"); // 执行 User savedUser = userRepository.save(user); // 验证 Assertions.assertNotNull(savedUser.getId()); Assertions.assertEquals("Test User", savedUser.getName()); Assertions.assertEquals("test@example.com", savedUser.getEmail()); } @Test @Order(2) void findByEmail_shouldReturnUser() { // 执行 Optional<User> found = userRepository.findByEmail("test@example.com"); // 验证 Assertions.assertTrue(found.isPresent()); Assertions.assertEquals("Test User", found.get().getName()); } }

3.2 使用 Testcontainers

@SpringBootTest @Testcontainers class DatabaseIntegrationTest { @Container static PostgreSQLContainer<?> postgres = new PostgreSQLContainer<>( "postgres:15-alpine" ) .withDatabaseName("testdb") .withUsername("testuser") .withPassword("testpass"); @Autowired private UserRepository userRepository; @DynamicPropertySource static void registerPgProperties(DynamicPropertyRegistry registry) { registry.add("spring.datasource.url", postgres::getJdbcUrl); registry.add("spring.datasource.username", postgres::getUsername); registry.add("spring.datasource.password", postgres::getPassword); } @Test void testUserRepository() { User user = new User(); user.setName("Container Test"); user.setEmail("container@test.com"); User saved = userRepository.save(user); Optional<User> found = userRepository.findById(saved.getId()); Assertions.assertTrue(found.isPresent()); Assertions.assertEquals("Container Test", found.get().getName()); } }

四、端到端测试

4.1 使用 Selenium

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT) class E2ETest { @Autowired private TestRestTemplate restTemplate; @LocalServerPort private int port; @Test void createAndGetUser() { // 创建用户 UserCreateDTO createDTO = new UserCreateDTO("E2E User", "e2e@test.com"); ResponseEntity<User> created = restTemplate.postForEntity( "http://localhost:" + port + "/api/users", createDTO, User.class ); Assertions.assertEquals(HttpStatus.CREATED, created.getStatusCode()); Assertions.assertNotNull(created.getBody().getId()); // 获取用户 ResponseEntity<User> retrieved = restTemplate.getForEntity( "http://localhost:" + port + "/api/users/" + created.getBody().getId(), User.class ); Assertions.assertEquals(HttpStatus.OK, retrieved.getStatusCode()); Assertions.assertEquals("E2E User", retrieved.getBody().getName()); Assertions.assertEquals("e2e@test.com", retrieved.getBody().getEmail()); } }

五、性能测试

5.1 使用 JMeter

@SpringBootTest class PerformanceTest { @Autowired private UserService userService; @Test void testUserServicePerformance() { // 预热 for (int i = 0; i < 100; i++) { userService.findById("1"); } // 测试 int iterations = 1000; long startTime = System.currentTimeMillis(); for (int i = 0; i < iterations; i++) { userService.findById("1"); } long endTime = System.currentTimeMillis(); long duration = endTime - startTime; System.out.println("执行 " + iterations + " 次查询耗时: " + duration + "ms"); System.out.println("平均每次查询耗时: " + (duration / (double) iterations) + "ms"); // 性能断言 Assertions.assertTrue(duration < 500, "性能不达标"); } }

5.2 使用 JMH 进行基准测试

@BenchmarkMode(Mode.AverageTime) @OutputTimeUnit(TimeUnit.MICROSECONDS) @Fork(1) @Warmup(iterations = 5) @Measurement(iterations = 10) public class UserServiceBenchmark { private UserService userService; private UserRepository userRepository; @Setup public void setup() { userRepository = Mockito.mock(UserRepository.class); User user = new User("1", "Test", "test@example.com"); Mockito.when(userRepository.findById("1")).thenReturn(Optional.of(user)); userService = new UserService(userRepository); } @Benchmark public User findById() { return userService.findById("1"); } }

六、测试最佳实践

6.1 测试命名规范

// 方法名格式: [被测试方法]_[场景]_[预期结果] @Test void createUser_withValidData_shouldReturnUser() {} @Test void createUser_withNullName_shouldThrowException() {} @Test void getUser_whenUserExists_shouldReturnUser() {} @Test void getUser_whenUserNotExists_shouldReturnNull() {}

6.2 使用 AssertJ

@Test void testUserAssertions() { User user = userService.findById("1"); // 使用 AssertJ 进行流式断言 Assertions.assertThat(user) .isNotNull() .extracting(User::getId, User::getName, User::getEmail) .containsExactly("1", "John", "john@example.com"); }

6.3 测试数据生成

@Component public class TestDataGenerator { public User createTestUser() { User user = new User(); user.setName("Test User"); user.setEmail("test@example.com"); user.setAge(25); return user; } public User createTestUser(String name, String email) { User user = new User(); user.setName(name); user.setEmail(email); user.setAge(25); return user; } public List<User> createTestUsers(int count) { List<User> users = new ArrayList<>(); for (int i = 0; i < count; i++) { User user = new User(); user.setName("User " + i); user.setEmail("user" + i + "@example.com"); user.setAge(20 + i); users.add(user); } return users; } }

七、测试配置

7.1 application-test.yml

spring: datasource: url: jdbc:h2:mem:testdb;DB_CLOSE_DELAY=-1;DB_CLOSE_ON_EXIT=FALSE username: sa password: driver-class-name: org.h2.Driver h2: console: enabled: true jpa: hibernate: ddl-auto: create-drop show-sql: true properties: hibernate: format_sql: true logging: level: org.hibernate.SQL: DEBUG org.hibernate.type.descriptor.sql.BasicBinder: TRACE

7.2 MockBean 与 SpyBean

@SpringBootTest class ServiceTest { // 使用 MockBean 替换整个 bean @MockBean private UserRepository userRepository; // 使用 SpyBean 包装真实 bean，可以部分 mock @SpyBean private EmailService emailService; @Autowired private UserService userService; @Test void testWithSpyBean() { // 调用真实方法，但可以验证调用 userService.registerUser(createTestUser()); Mockito.verify(emailService).sendWelcomeEmail(Mockito.anyString()); } }

八、总结

Spring Boot 测试策略应包含以下层次：

1. 单元测试：覆盖单个组件，使用 Mockito 隔离依赖
2. 集成测试：验证多个组件协作，使用 Testcontainers 模拟真实环境
3. 端到端测试：测试完整业务流程，使用 RestTemplate 或 Selenium
4. 性能测试：评估系统性能，使用 JMeter 或 JMH

通过建立完善的测试体系，可以提高代码质量，减少回归缺陷，加速开发迭代。

参考资料

1. Spring Boot 测试文档：https://docs.spring.io/spring-boot/docs/current/reference/html/features.html#features.testing
2. Mockito 官方文档：https://site.mockito.org/
3. AssertJ 文档：https://assertj.github.io/doc/
4. Testcontainers：https://www.testcontainers.org/