為什么需要單元測試

單元測試擁有保證代碼質(zhì)量、盡早發(fā)現(xiàn)軟件 Bug、簡化調(diào)試過程、促進變化并簡化集成、使流程更靈活等優(yōu)勢。單元測試是針對代碼單元的獨立測試，核心是“獨立”，優(yōu)勢來源也是這種獨立性，而所面臨的不足也正是因為其獨立性：既然是“獨立”，就難以測試與其他代碼和依賴環(huán)境的相互關系。單元測試與系統(tǒng)測試是互補而非代替關系。單元測試的優(yōu)勢，正是系統(tǒng)測試的不足，單元測試的不足，又恰是系統(tǒng)測試的優(yōu)勢。不能將單元測試當做解決所有問題的萬金油，而需理解其優(yōu)勢與不足，揚長避短，與系統(tǒng)測試相輔相成，實現(xiàn)測試的最大效益。

OAuth2 系統(tǒng)單元測試困難

接口測試依賴于 UPMS (用戶權限管理)，無法做到解耦獨立 spring-security-test 模塊未提供相關標準實現(xiàn) 場景復雜既要包含無狀態(tài) token 調(diào)用，又要保證上線文傳遞業(yè)務

解決方案

參考 @WithMockUser ，在 Mock 攔截器中自動執(zhí)行相關的增強（token 獲取），并通過擴展 WithSecurityContextFactory 實現(xiàn)上下文 token 的傳遞。具體可以參考源碼 pig-common-test[1]

引入依賴

<dependency> <groupId>com.pig4cloud</groupId> <artifactId>pig-common-test</artifactId> <version>${last.version}</version> <scope>test</scope></dependency>

單元測試 Controller 接口

指定認證中心接口

配置在 test/resources/application.yml

security: oauth2: client: access-token-uri: http://pig-gateway:3000/oauth/token

模擬測試 controller 接口

@RunWith(SpringRunner.class)@SpringBootTestpublic class SysLogControllerTest { private MockMvc mvc; @Autowired private WebApplicationContext applicationContext; // 注入WebApplicationContext @Before public void setUp() { this.mvc = MockMvcBuilders.webAppContextSetup(applicationContext).build(); } @Test @SneakyThrows @WithMockOAuth2User public void testMvcToken() { mvc.perform(delete('/log/1').with(token())).andExpect(status().isOk()); }}

模擬測試 FeignClient 傳遞 token

直接注入 FeignClient 實現(xiàn)即可 使用 @WithMockOAuth2User 注解測試類即可

WithMockOAuth2User 屬性說明

當前用例獲取 token 使用的用戶名

String username() default 'admin';

當前用例獲取 token 使用的密碼

String password() default '123456';

寫在最后

源碼參考 pig-common-test[1] 模塊

目前僅在 pig 2.10 做了實現(xiàn)，理論支持低版本，直接 install 此模塊即可

參考資料

[1] pig-common-test: https://gitee.com/log4j/pig/tree/master/pig-common/pig-common-test

到此這篇關于Spring OAuth2.0 單元測試解決方案的文章就介紹到這了,更多相關Spring OAuth2.0 單元測試內(nèi)容請搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持好吧啦網(wǎng)！