如何在spring boot应用程序中模拟安全上下文持有者

Question

我们在Spring boot rest api中有下面的代码片段来获取用户角色。

Authentication auth = SecurityContextHolder.getContext().getAuthentication();
if(auth.getAuthorities().contains(new SimpleGrantedAuthority("ROLE_roleName1"))){
// some conditions
} else if (auth.getAuthorities().contains(new SimpleGrantedAuthority("ROLE_roleName2"))){
// some conditions
}

我们在一个测试课上有这个。

@Before
public void setUp(){
Authentication auth = Mockito.mock(Authentication.class);
SecurityContext securityContext = Mockito.mock(SecurityContext.class);
Mockito.when(securityContext.getAuthentication()).thenReturn(authentication);
SecurityContextHolder.setContext(securityContext);
}

但junit测试用例失败了。请建议如何为安全上下文持有者编写junit测试用例（针对上述代码）。

Answer 1

您可以使用@WitMockUser（自Spring4.0起可用）注释您的测试方法，例如@WitMockUser（角色="MANAGER"）

@Test
@WithMockUser(username = "myuser", password = "pass", roles = "USER")
public void test() throws Exception {
    mockMvc.perform(get("/foo"))
        .andExpect(status().isOk());
}

请记住添加以下依赖项来测试项目的依赖项：

<代码>组织。springframework。安全：Spring安全测试