Question

我正在为我的Spring MVC控制器设置单元测试，我正在尝试利用Spring MVC测试框架。对于我的控制器中的每个端点，我想确保只有具有指定权限的用户才能访问。我的问题是我在使用mockMvc工具时使用自定义User实现并获得类强制转换异常。

对于每个请求，我希望它看起来像这样：

 mockMvc.perform(MockMvcRequestBuilders.get(path).with(user("user").authorities(authorities)))
                .andExpect(status().isOk())
                .andExpect(authenticated().withUsername("user"));

我想以某种方式调整上面的语句来指定我的自定义用户主体。请参阅下面的Spring的用户方法。我的第一个想法是我会覆盖类UserRequestPostProcessor并调整它以便它使用我的自定义用户类型而不是标准的Spring Security用户，但是这个类被声明为final并且不能被子类化。是否支持覆盖默认行为并使用自定义用户类型??

public static UserRequestPostProcessor user(String username) {
        return new UserRequestPostProcessor(username);
}

从我所看到的，我是用@WithSecurityContext注释我的测试的候选人，以便我可以在身份验证中设置我的自定义用户主体。我在这里担心的是，我将仅限于为每种方法测试一个用户，但这并不适合我想要做的事情。

如何测试多个自定义用户发出的请求？

Answer 1

我了解了两种方法：

我可以创建一个实现UserDetails的类并扩展我的自定义用户主体类。然后我可以将其作为参数传递用户方法。
我可以完全抓住用户方法，并传入我的已使用我的自定义用户主体设置身份验证。

我选择了后者。

protected void performTest(HttpMethod method, String path, Object[] pathVariables, UserRoleEnum role,
        boolean expectAllowed) throws Exception {
    mockMvc.perform(buildRequest(method, path, pathVariables).with(authentication(createAuthentication(role))))
            .andExpect(expectAllowed ? status().isNotFound() : status().isForbidden())
            .andExpect(authenticated().withUsername("user"));
}

注意 - buildRequest和createAuthentication方法是我创建的辅助方法，而所有其他方法都是由Spring提供的。辅助方法分别返回MockHttpServletRequestBuilder和Authentication。

Spring Security Unit Test - MockMvc使用自定义用户执行测试

1 个答案: