使用依赖于 Spring Security 的 JUnit 测试 Spring Controller

Question

我有一个 Spring 应用程序，我正在构建 JUnit 测试来测试某个 Controller。

问题是在Controller里面我调用了这个代码：

final Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
final String userName = authentication.getName();

换句话说，我需要在调用此Controller 之前进行身份验证。我用这段代码写了一个JUnit 测试：

private MockMvc mockMvc;

    @Test
    public void getPageTest() throws Exception{
        final ProcessFileController controller = new ProcessFileController();
        mockMvc = standaloneSetup(controller).build();

    mockMvc.perform(get(URI.create("/processFile.html")).sessionAttr("freeTrialEmailAddress", "")).andExpect(view().name("processFile"));
        }

当我运行它时，它会在final String userName = authentication.getName(); 上给我一个NullPointerException，因为我的authentication 是null，因为我没有登录。

问题是：有没有办法模拟身份验证？欢迎所有想法。

谢谢。

Answer 1

Spring Security 版本 4 对此进行了一些巧妙的改进。

首先确保您在类路径中有用于测试的测试框架，使用 Maven，它看起来像：

<dependency>
  <groupId>org.springframework.security</groupId>
  <artifactId>spring-security-test</artifactId>
  <version>4.0.4.RELEASE</version>
  <scope>test</scope>
</dependency>

有用的导入：

import static org.springframework.security.test.web.servlet.setup.SecurityMockMvcConfigurers.*;
import static org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors.*;

测试设置：

mockMvc = webAppContextSetup(applicationContext).apply(springSecurity()).build();

（我认为您需要 WebApplicationContext 而不是单个控制器。）

然后测试类似：

mockMvc.perform(get(...).with(user("username").roles("USER"))).andExpect(...);

Answer 2

理想情况下，您应该使用@AuthenticationPrincipal，但如果这不是一个选项，您需要使用Authentication 实例设置SecurityContext，然后该实例将在测试中可用。

您可以在帮助类中使用静态方法来执行此操作。

public static void setupSecurityContext(String username, String password, String... groups)
{
  List<GrantedAuthority> authorities = new ArrayList<>();
  for (String group : groups)
  {
    authorities.add(new SimpleGrantedAuthority(group));
  }

  UserDetails user = new UserDetails(username, password, authorities);
  UsernamePasswordAuthenticationToken token = new UsernamePasswordAuthenticationToken(user, password);
  SecurityContextHolder.getContext().setAuthentication(token);
}

然后在测试中你可以简单地调用

SecurityHelper.setupSecurityContext("user", "password", "g1", "g2");