4

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

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

 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注释我的测试的候选人，以便我可以在身份验证中设置我的自定义用户主体。我在这里担心的是，我将仅限于为每种方法测试一个用户，而这不适用于我正在尝试做的事情。

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

4

1 回答 1

2

我学到了两种方法来做到这一点：

  1. 我可以创建一个实现 UserDetails 并扩展我的自定义用户主体类的类。然后我可以将它作为参数传递给用户方法。

  2. 我可以完全从头开始使用用户方法，并传入我已经使用我的自定义用户主体设置的身份验证。

我选择了后者。

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。

于 2016-04-12T02:52:32.030 回答