今天，在為一種Java方法編寫單元測試用例時，如下所示：

public ApplicationUser getApplicationUser() {ApplicationUser applicationUser = (ApplicationUser) SecurityContextHolder.getContext().getAuthentication().getPrincipal();return applicationUser; }

我想模擬Spring Security Context以獲取Principal，以實現與我模擬方法的每個級別相同的目標，如下所示：

import org.junit.Before; import org.junit.Rule; import org.junit.Test; import static org.mockito.Mockito.when; import static org.mockito.Mockito.mock; import org.mockito.MockitoAnnotations; import org.springframework.security.core.Authentication; import org.springframework.security.core.context.SecurityContext; import org.springframework.security.core.context.SecurityContextHolder; import com.arpit.security.user.ApplicationUser;public class BaseTest {@Beforepublic void setupMock() {MockitoAnnotations.initMocks(this);}@Testpublic void mockApplicationUser() {ApplicationUser applicationUser = mock(ApplicationUser.class);Authentication authentication = mock(Authentication.class);SecurityContext securityContext = mock(SecurityContext.class);when(securityContext.getAuthentication()).thenReturn(authentication);SecurityContextHolder.setContext(securityContext);when(SecurityContextHolder.getContext().getAuthentication().getPrincipal()).thenReturn(applicationUser);}}

翻譯自: https://www.javacodegeeks.com/2017/05/mocking-spring-security-context-unit-testing.html

總結

以上是生活随笔為你收集整理的模拟Spring Security上下文进行单元测试的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。