Spring安全测试和MockMvc为REST控制器提供空的自定义UserDetails参数

Question

我正在尝试编写一个集成测试，该测试可以访问REST端点并获取特定身份验证用户(我在测试中设置的用户)的数据。我最初在mockMvc = webAppContextSetup(wac).apply(springSecurity()).build()上尝试了我的设置，但在BeanInstantiationException上一直失败。在使用自定义UserDetails实现测试Spring安全性和MvcMock的帮助下，我能够通过切换我的设置来克服这个问题。现在，我已经切换到我的设置使用standaloneSetup，我可以调用到我的控制器。但是，无论我做什么，我都无法将我在测试中创建的自定义UserDetails对象获取到对MockMvc调用结束的控制器中的方法中。

我使用Spring 4.2.2和Spring Security 4.0.1.

我的测试代码看起来像这样:

import static org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors.user;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
import static org.springframework.test.web.servlet.setup.MockMvcBuilders.standaloneSetup;
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(classes = { RestControllerITConfig.class })
@WebAppConfiguration
public class ControllerIT {
  private MockMvc mockMvc;
  @Before
  public void setUp() {
    mockMvc = standaloneSetup(new Controller())
                .setCustomArgumentResolvers(new AuthenticationPrincipalArgumentResolver())
                .build();
  }
  @Test
  public void testGetEndpoint() throws Exception {
    CustomUserDetails userDetails = new CustomUserDetails("John","Smith", "abc123");
    assertNotNull(userDetails);
    MvcResult result = mockMvc.perform(
      get("/somepath").with(user(userDetails)))    
      .andExpect(status().isOk())
      .andExpect(content().contentType("text/json")));
  }
}
@Configuration
@ComponentScan(basePackageClasses = { AuthenticationConfig.class })
public class RestControllerITConfig {
}
@Configuration
@EnableWebSecurity
@ComponentScan("com.my.authentication.package")
public class AuthenticationConfig extends WebSecurityConfigurerAdapter {
  // snip
}
@Target({ ElementType.PARAMETER, ElementType.TYPE })
@Retention(RetentionPolicy.RUNTIME)
@Documented
@AuthenticationPrincipal
public @interface MyUser {
}
@RestController
@RequestMapping("/somepath")
public class Controller {
  @RequestMapping(value = "", method = RequestMethod.GET)
  public ResponseEntity<Object> getSomePath(@MyUser CustomUserDetails customUser) {
    customUser.getName(); // <== Causes NPE
  }
}

为简洁起见，省略了其他一些不相关的配置。我真的不明白为什么我在测试中显式创建的自定义UserDetails对象没有传递到我的REST控制器中，而是传递了一个空对象。我哪里做错了?有人能举个类似的例子吗?

Answer 1

我终于能够通过显式地将我的AuthenticationConfig的过滤器添加到StandaloneMockMvcBuilder来实现这一点。所以我的设置现在看起来像:

private MockMvc mockMvc;
@Autowired
private MyController controller;
@Autowired
private AuthenticationConfig authConfig;
@Before
public void setUp() {
  mockMvc = standaloneSetup(controller)
              .setCustomArgumentResolvers(new AuthenticationPrincipalArgumentResolver())
              .addFilters(authConfig.getCustomFilter())
              .build();
}

Answer 2

如果您正在使用带有自定义用户详细信息实现的Spring 5，则可以轻松使用@WithUserDetails注释

例如

@Test
@WithUserDetails
void createInstitution

您可以在spring安全文档