SecurityMockMvcResultHandlers
Spring Security 提供一些 ResultHandler
的实现。为使用 Spring Security 的 ResultHandler
的实现确保已使用了以下静态导入:
import static org.springframework.security.test.web.servlet.response.SecurityMockMvcResultHandlers.*;
Exporting the SecurityContext
通常我们希望查询一个存储库来查看某些 MockMvc
请求是否实际保存在数据库中。在某些情况下,我们的存储库查询使用 Spring Data Integration 根据当前用户的用户名或任何其他属性过滤结果。让我们看一个示例:
一个数据仓库接口:
private interface MessageRepository extends JpaRepository<Message, Long> {
@Query("SELECT m.content FROM Message m WHERE m.sentBy = ?#{ principal?.name }")
List<String> findAllUserMessages();
}
我们的测试场景:
mvc
.perform(post("/message")
.content("New Message")
.contentType(MediaType.TEXT_PLAIN)
)
.andExpect(status().isOk());
List<String> userMessages = messageRepository.findAllUserMessages();
assertThat(userMessages).hasSize(1);
此测试无法通过,因为在我们的请求完成后,SecurityContextHolder
将被过滤器链清除。然后,我们可以将 TestSecurityContextHolder
导出到 SecurityContextHolder
中,并按需要使用它:
mvc
.perform(post("/message")
.content("New Message")
.contentType(MediaType.TEXT_PLAIN)
)
.andDo(exportTestSecurityContext())
.andExpect(status().isOk());
List<String> userMessages = messageRepository.findAllUserMessages();
assertThat(userMessages).hasSize(1);
记住在测试之间清除 |