單元測試是軟件開發過程中不可或缺的一部分，它能有效地提高代碼質量，確保代碼功能的正確性。在 Spring 應用中，JUnit 和 Mockito 是常用的單元測試工具，而 Spring Test 提供了豐富的測試支持。本文將介紹如何在 Spring 中使用 JUnit、Mockito 和 Spring Test 進行單元測試。

1. 引入必要的依賴

首先，在 pom.xml 文件中添加 JUnit、Mockito 和 Spring Test 的依賴：

<dependencies><!-- JUnit 5 依賴 --><dependency><groupId>org.junit.jupiter</groupId><artifactId>junit-jupiter-engine</artifactId><version>5.8.1</version><scope>test</scope></dependency><!-- Mockito 依賴 --><dependency><groupId>org.mockito</groupId><artifactId>mockito-core</artifactId><version>4.0.0</version><scope>test</scope></dependency><!-- Spring Test 依賴 --><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-test</artifactId><version>2.5.6</version><scope>test</scope><exclusions><exclusion><groupId>org.junit.vintage</groupId><artifactId>junit-vintage-engine</artifactId></exclusion></exclusions></dependency>
</dependencies>

2. 使用 JUnit 編寫單元測試

JUnit 是最流行的 Java 單元測試框架。我們將使用 JUnit 5 編寫一個簡單的單元測試。

假設我們有一個 CalculatorService 類：

public class CalculatorService {public int add(int a, int b) {return a + b;}public int subtract(int a, int b) {return a - b;}
}

我們可以為其編寫如下測試類：

import org.junit.jupiter.api.Test;
import static org.junit.jupiter.api.Assertions.assertEquals;public class CalculatorServiceTest {private final CalculatorService calculatorService = new CalculatorService();@Testpublic void testAdd() {assertEquals(5, calculatorService.add(2, 3));}@Testpublic void testSubtract() {assertEquals(1, calculatorService.subtract(3, 2));}
}

3. 使用 Mockito 進行單元測試

Mockito 是一個用于創建模擬對象的流行框架，特別適用于測試依賴注入的類。假設我們有以下 UserService 類和 UserRepository 接口：

public interface UserRepository {User findByUsername(String username);
}public class UserService {private UserRepository userRepository;public UserService(UserRepository userRepository) {this.userRepository = userRepository;}public User getUserByUsername(String username) {return userRepository.findByUsername(username);}
}

我們可以使用 Mockito 為 UserService 編寫單元測試：

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.Mockito.when;public class UserServiceTest {private UserRepository userRepository;private UserService userService;@BeforeEachpublic void setUp() {userRepository = Mockito.mock(UserRepository.class);userService = new UserService(userRepository);}@Testpublic void testGetUserByUsername() {User user = new User("john_doe");when(userRepository.findByUsername("john_doe")).thenReturn(user);User result = userService.getUserByUsername("john_doe");assertEquals("john_doe", result.getUsername());}
}

4. 使用 Spring Test 進行單元測試

Spring Test 提供了對 Spring 應用的全面測試支持。通過 @SpringBootTest 注解，可以加載整個 Spring 應用的上下文進行集成測試。

假設我們有一個簡單的 Spring Boot 應用，包括 UserController 和 UserService：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;@RestController
public class UserController {@Autowiredprivate UserService userService;@GetMapping("/user")public User getUser(@RequestParam String username) {return userService.getUserByUsername(username);}
}

我們可以使用 Spring Test 編寫如下測試類：

import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
import org.springframework.test.web.servlet.result.MockMvcResultMatchers;@SpringBootTest
@AutoConfigureMockMvc
public class UserControllerTest {@Autowiredprivate MockMvc mockMvc;@Testpublic void testGetUser() throws Exception {mockMvc.perform(MockMvcRequestBuilders.get("/user").param("username", "john_doe")).andExpect(MockMvcResultMatchers.status().isOk()).andExpect(MockMvcResultMatchers.jsonPath("$.username").value("john_doe"));}
}

結論

通過本文的介紹，我們了解了如何在 Spring 應用中使用 JUnit、Mockito 和 Spring Test 進行單元測試。JUnit 提供了基礎的測試框架，Mockito 允許我們方便地創建模擬對象，而 Spring Test 則提供了對 Spring 應用的全面測試支持。