使用JUnit測試Spring Security身份驗證-java教程-PHP中文網

使用JUnit測試Spring Security身份驗證

王林

發布： 2023-09-10 10:01:05

轉載

663 人瀏覽過

使用JUnit测试Spring Security身份验证

簡介

Spring Security 是一個高度可自訂的身份驗證和存取控制框架，適用於 Java 應用程序，特別是基於 Spring 的應用程式。測試這些安全措施對於確保應用程式的安全至關重要。在本文中，我們將探討如何使用 JUnit（Java 中領先的單元測試框架）有效測試 Spring Security。

理解Spring Security和JUnit

#Spring Security是一個強大的框架，為企業級應用程式提供身份驗證、授權和其他安全功能。它既全面又靈活，適用於各種安全需求。

JUnit 是一個簡單的開源框架，用於用 Java 編寫可重複的測試。它提供註釋來標識測試方法和斷言來檢查這些測試的結果。

使用JUnit測試Spring Security

##設定測試環境

要使用 JUnit 測試 Spring Security，我們首先需要將必要的依賴項新增到 Maven 或 Gradle 建置檔中。對於 Maven，我們將包括 -

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-security</artifactId>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-test</artifactId>
    <scope>test</scope>
</dependency>

登入後複製

編寫Spring Security的測試案例

現在，我們將開始編寫我們的測試案例。假設我們有一個REST API端點（"/api/data"），只有經過驗證的使用者才能存取。我們可以寫一個JUnit測試來驗證這一點−

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 static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;

@SpringBootTest
@AutoConfigureMockMvc
public class WebSecurityTest {

   @Autowired
   private MockMvc mockMvc;

   @Test
   public void shouldReturnUnauthorizedForUnauthenticatedUsers() throws Exception {
      mockMvc.perform(get("/api/data"))
         .andExpect(status().isUnauthorized());
   }
}

登入後複製

在這個測試中，我們使用MockMvc執行一個GET請求到"/api/data"。由於使用者未經身份驗證，我們期望HTTP狀態碼為401（未授權）。

測試身份驗證存取

如果我們想要為已經認證的使用者測試端點，那麼Spring Security Test提供了@WithMockUser註解來實現這個目的 -

import org.springframework.security.test.context.support.WithMockUser;

@Test
@WithMockUser
public void shouldReturnOkForAuthenticatedUsers() throws Exception {
   mockMvc.perform(get("/api/data")).andExpect(status().isOk());
}

登入後複製

在此測試中，@WithMockUser 設定一個模擬用戶，使請求「經過身份驗證」。然後，我們預計 HTTP 狀態為 200（正常）。