如何在Java中使用Spring切换用户进行测试？

Question

我使用Spring Security SwitchUserFilter编写了功能。在应用程序中，我可以使用/j_spring_security_switch_user?j_username=xxx URL切换用户，并使用/j_spring_security_exit_user返回上一页。我还实现了几种取决于切换用户的事实的方法，所以我想为它们编写单元测试。因此我的问题是如何在jUnit测试环境中切换用户？

Answer 1

我写的方法正在准备用户使用SwitchUserGrantedAuthority并登录他。它似乎正常我的测试目的，但任何提示和评论将非常感激。

@SuppressWarnings({ "rawtypes", "unchecked" })
private User logAdminAsUser(User admin, String roleName) {
    SecurityContextHolder.getContext().setAuthentication(
        new TestingAuthenticationToken(admin, null, "ROLE_ADMIN"));
    Authentication adminAuth = SecurityContextHolder.getContext().getAuthentication();
    SwitchUserGrantedAuthority switchUserGrantedAuthority =
        new SwitchUserGrantedAuthority("ROLE_ADMIN", adminAuth);
    List authorities = new LinkedList();
    authorities.add(switchUserGrantedAuthority);
    User user = populator.storeUser("ROLE_USER");
    SecurityContextHolder.getContext().setAuthentication(
        new TestingAuthenticationToken(user, null, authorities));
    return user;
}

Answer 2

如果您想进行集成测试，您应该考虑使用自定义http客户端，或者如果您的测试逻辑依赖于它，甚至是像Selenium这样的GUI驱动程序。

如果我们讨论的是单元测试，请参考Springs http://spring.io/blog/2014/05/07/preview-spring-security-test-method-security文档，它们支持大量测试，@ WithMockUser注释似乎是您正在寻找的，它允许您指定应该运行此测试的角色或用户。

Answer 3

我用过这个：

private void switchUser(User user, String roleName) 
{
    Authentication authentication = SecurityContextHolder.getContext().getAuthentication();

    Collection<GrantedAuthority> authorities = 
            new ArrayList<>();
    GrantedAuthority ga = new SimpleGrantedAuthority(roleName);
    authorities.add(ga);

    Authentication result = new UsernamePasswordAuthenticationTokenExt(
            user,
            authentication.getCredentials(),
            null,
            System.currentTimeMillis()
    );

    SecurityContextHolder.getContext().setAuthentication( result );
}

其中User是新用户，而roleName是新设置的权限（当然这种方法可以修改得更多的参数等）