单元测试是指对软件中的最小可测试单元进行检查和验证。在Java中，单元测试的最小单元是类。通过编写针对类或方法的小段代码，来检验被测代码是否符合预期结果或行为。执行单元测试可以帮助开发者验证代码是否正确实现了功能需求，以及是否能够适应应用环境或需求变化。
本文将介绍如何在Spring Boot中编写优雅的单元测试，包括如何添加单元测试依赖，如何对不同层次的组件进行单元测试，以及如何使用Mock对象来模拟真实对象行为。本文假设读者已经对Spring Boot和单元测试有一定的了解和基础。
目录
二、Spring Boot中不同层次的单元测试
Service层
在Spring Boot项目中，要进行单元测试，首先需要添加相应的依赖。如果使用Maven作为构建工具，可以在pom.xml文件中添加以下依赖：
|
1 2 3 4 5 |
<dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-test</artifactId> <scope>test</scope> </dependency> |
这个依赖包含了多个库和功能，主要有以下几个：
除了以上这些库外，spring-boot-starter-test还包含了其他一些库和功能，如JsonPath、JsonAssert、XmlUnit等。这些库和功能可以根据不同的测试场景进行选择和使用。
Service层是指封装业务逻辑和处理数据的层，它通常使用@Service或@Component注解来标识。在Spring Boot中，对Service层进行单元测试，可以使用@SpringBootTest注解来加载完整的Spring上下文，从而可以自动注入Service层的Bean。同时，可以使用@MockBean注解来创建和注入其他层次的Mock对象，从而避免真实地调用其他层次的方法，而是模拟其行为。
例如，假设有一个UserService类，它提供了一个根据用户ID查询用户信息的方法：
|
1 2 3 4 5 6 7 8 9 10 |
@Service public class UserService { @Autowired private UserRepository userRepository; public User getUserById(Long id) { return userRepository.findById(id).orElse(null); } } |
要对这个类进行单元测试，可以编写以下测试类：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
@SpringBootTest public class UserServiceTest { @Autowired private UserService userService; @MockBean private UserRepository userRepository; @Test public void testGetUserById() { // 创建一个User对象 User user = new User(); user.setId(1L); user.setName("Alice"); user.setEmail("alice@example.com"); // 当调用userRepository.findById(1L)时，返回一个包含user对象的Optional对象 when(userRepository.findById(1L)).thenReturn(Optional.of(user)); // 调用userService.getUserById方法，传入1L作为参数，得到一个User对象 User result = userService.getUserById(1L); // 验证结果对象与user对象相等 assertThat(result).isEqualTo(user); // 验证userRepository.findById(1L)方法被调用了一次 verify(userRepository, times(1)).findById(1L); } } |
在这个测试类中，使用了以下几个关键点和技巧：
Controller层是指处理用户请求和响应的层，它通常使用@RestController或@Controller注解来标识。在Spring Boot中，对Controller层进行单元测试，可以使用@WebMvcTest注解来启动一个轻量级的Spring MVC上下文，只加载Controller层的组件。同时，可以使用@AutoConfigureMockMvc注解来自动配置一个MockMvc对象，用来模拟Http请求和验证Http响应。
例如，假设有一个UserController类，它提供了一个根据用户ID查询用户信息的接口：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
@RestController @RequestMapping("/users") public class UserController { @Autowired private UserService userService; @GetMapping("/{id}") public ResponseEntity<User> getUserById(@PathVariable Long id) { User user = userService.getUserById(id); if (user == null) { return ResponseEntity.notFound().build(); } else { return ResponseEntity.ok(user); } } } |
要对这个类进行单元测试，可以编写以下测试类：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
@WebMvcTest(UserController.class) @AutoConfigureMockMvc public class UserControllerTest { @Autowired private MockMvc mockMvc; @MockBean private UserService userService; @Test public void testGetUserById() throws Exception { // 创建一个User对象 User user = new User(); user.setId(1L); user.setName("Alice"); user.setEmail("alice@example.com"); // 当调用userService.getUserById(1L)时，返回user对象 when(userService.getUserById(1L)).thenReturn(user); // 模拟发送GET请求到/users/1，并验证响应状态码为200，响应内容为JSON格式的user对象 mockMvc.perform(get("/users/1")) .andExpect(status().isOk()) .andExpect(content().contentType(MediaType.APPLICATION_JSON)) .andExpect(jsonPath("$.id").value(1L)) .andExpect(jsonPath("$.name").value("Alice")) .andExpect(jsonPath("$.email").value("alice@example.com")); // 验证userService.getUserById(1L)方法被调用了一次 verify(userService, times(1)).getUserById(1L); } } |
在这个测试类中，使用了以下几个关键点和技巧：
Repository层是指封装数据访问和持久化的层，它通常使用@Repository或@JpaRepository注解来标识。在Spring Boot中，对Repository层进行单元测试，可以使用@DataJpaTest注解来启动一个嵌入式数据库，并自动配置JPA相关的组件。同时，可以使用@TestEntityManager注解来获取一个TestEntityManager对象，用来操作和验证数据库数据。
例如，假设有一个UserRepository接口，它继承了JpaRepository接口，并提供了一个根据用户姓名查询用户列表的方法：
|
1 2 3 4 5 |
@Repository public interface UserRepository extends JpaRepository<User, Long> { List<User> findByName(String name); } |
要对这个接口进行单元测试，可以编写以下测试类：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
@DataJpaTest public class UserRepositoryTest { @Autowired private UserRepository userRepository; @Autowired private TestEntityManager testEntityManager; @Test public void testFindByName() { // 创建两个User对象，并使用testEntityManager.persist方法将其保存到数据库中 User user1 = new User(); user1.setName("Bob"); user1.setEmail("bob@example.com"); testEntityManager.persist(user1); User user2 = new User(); user2.setName("Bob"); user2.setEmail("bob2@example.com"); testEntityManager.persist(user2); // 调用userRepository.findByName方法，传入"Bob"作为参数，得到一个用户列表 List<User> users = userRepository.findByName("Bob"); // 验证用户列表的大小为2，且包含了user1和user2 assertThat(users).hasSize(2); assertThat(users).contains(user1, user2); } } |
在这个测试类中，使用了以下几个关键点和技巧：
在Spring Boot中，除了使用@WebMvcTest和@DataJpaTest等注解来加载特定层次的组件外，还可以使用@SpringBootTest注解来加载完整的Spring上下文，从而进行更加集成的测试。但是，在这种情况下，可能会遇到一些问题，例如：
– 测试过程中需要依赖外部资源，如数据库、消息队列、Web服务等。这些资源可能不稳定或不可用，导致测试失败或超时。
– 测试过程中需要调用其他组件或服务的方法，但是这些方法的实现或行为不确定或不可控，导致测试结果不可预测或不准确。
– 测试过程中需要验证一些难以观察或测量的结果，如日志输出、异常抛出、私有变量值等。这些结果可能需要使用复杂或侵入式的方式来获取或验证。
为了解决这些问题，可以使用Mock对象来模拟真实对象行为。Mock对象是指在测试过程中替代真实对象的虚拟对象，它可以根据预设的规则来返回特定的值或执行特定的操作。使用Mock对象有以下好处：
– 降低测试依赖：通过使用Mock对象来替代外部资源或其他组件，可以减少测试过程中对真实环境的依赖，使得测试更加稳定和可靠。
– 提高测试控制：通过使用Mock对象来模拟特定的行为或场景，可以提高测试过程中对真实对象行为的控制，使得测试更加灵活和精确。
– 简化测试验证：通过使用Mock对象来返回特定的结果或触发特定的事件，可以简化测试过程中对真实对象结果或事件的验证，使得测试更加简单和直观。
在Spring Boot中，要使用Mock对象，可以使用@MockBean注解来创建和注入一个Mock对象。这个注解会自动使用Mockito库来创建一个Mock对象，并将其添加到Spring上下文中。同时，可以使用when方法来定义Mock对象的行为，以及verify方法来验证Mock对象的方法调用。
例如，假设有一个EmailService接口，它提供了一个发送邮件的方法：
|
1 2 3 4 |
public interface EmailService { void sendEmail(String to, String subject, String content); } |
要对这个接口进行单元测试，可以编写以下测试类：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
@SpringBootTest public class EmailServiceTest { @Autowired private UserService userService; @MockBean private EmailService emailService; @Test public void testSendEmail() { // 创建一个User对象 User user = new User(); user.setId(1L); user.setName("Alice"); user.setEmail("alice@example.com"); // 当调用emailService.sendEmail方法时，什么也不做 doNothing().when(emailService).sendEmail(anyString(), anyString(), anyString()); // 调用userService.sendWelcomeEmail方法，传入user对象作为参数 userService.sendWelcomeEmail(user); // 验证emailService.sendEmail方法被调用了一次，并且参数分别为user.getEmail()、"Welcome"、"Hello, Alice" verify(emailService, times(1)).sendEmail(user.getEmail(), "Welcome", "Hello, Alice"); } } |
在这个测试类中，使用了以下几个关键点和技巧：
除了使用@MockBean注解来创建和注入Mock对象外，还可以使用@SpyBean注解来创建和注入Spy对象。Spy对象是指在测试过程中部分替代真实对象的虚拟对象，它可以根据预设的规则来返回特定的值或执行特定的操作，同时保留真实对象的其他行为。使用Spy对象有以下好处：
在Spring Boot中，要使用Spy对象，可以使用@SpyBean注解来创建和注入一个Spy对象。这个注解会自动使用Mockito库来创建一个Spy对象，并将其添加到Spring上下文中。同时，可以使用when方法来定义Spy对象的行为，以及verify方法来验证Spy对象的方法调用。
例如，假设有一个LogService接口，它提供了一个记录日志的方法：
|
1 2 3 4 |
public interface LogService { void log(String message); } |
要对这个接口进行单元测试，可以编写以下测试类：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
@SpringBootTest public class LogServiceTest { @Autowired private UserService userService; @SpyBean private LogService logService; @Test public void testLog() { // 创建一个User对象 User user = new User(); user.setId(1L); user.setName("Alice"); user.setEmail("alice@example.com"); // 当调用logService.log方法时，调用真实的方法，并打印参数到控制台 doAnswer(invocation -> { String message = invocation.getArgument(0); System.out.println(message); invocation.callRealMethod(); return null; }).when(logService).log(anyString()); // 调用userService.createUser方法，传入user对象作为参数 userService.createUser(user); // 验证logService.log方法被调用了两次，并且参数分别为"Creating user: Alice"、"User created: Alice" verify(logService, times(2)).log(anyString()); verify(logService, times(1)).log("Creating user: Alice"); verify(logService, times(1)).log("User created: Alice"); } } |
在这个测试类中，使用了以下几个关键点和技巧：
本文介绍了如何在Spring Boot中编写优雅的单元测试，包括如何添加单元测试依赖，如何对不同层次的组件进行单元测试，以及如何使用Mock对象和Spy对象来模拟真实对象行为。本文还给出了每种类型的单元测试的示例代码，并解释了其中的关键点和技巧。
通过编写单元测试，可以提高Spring Boot应用的质量和稳定性，同时也可以提高开发者的编程水平和信心。希望本文能够对你有所帮助和启发，让你能够在Spring Boot中编写优雅的单元测试。
from:https://blog.csdn.net/TaloyerG/article/details/132487310