Question

我正在使用Spring OAuth2和JWT令牌来保护应用程序。我正在扩展 org.springframework.security.core.userdetails ，以便向令牌添加一些其他属性，然后可以将这些属性用于执行称为端点的授权。

public class CustomUser extends User {
     private Set<String> bookIds;

     public CustomUser(String username, String password, Collection<? extends GrantedAuthority> authorities) {
         super(username, password, authorities);
     }

     public CustomUser(String username, String password, boolean enabled, boolean accountNonExpired, boolean credentialsNonExpired, boolean accountNonLocked, Collection<? extends GrantedAuthority> authorities) {
         super(username, password, enabled, accountNonExpired, credentialsNonExpired, accountNonLocked, authorities);
     }
}

我也有一个org.springframework.security.access.PermissionEvaluator的自定义实现，它能够反序列化令牌并检查自定义属性是否是其中的一部分，可以将其本身添加到控制器端点。

@PreAuthorize("hasPermission(authentication, #bookId,'read')")

现在一切正常，我可以通过邮递员测试我的应用程序，只有拥有有效JWT令牌且其bookIds设置的URL部分中具有bookID的用户才能访问资源

@PreAuthorize("hasPermission(authentication, #bookId,'read')")
@GetMapping(value = "api/books/{bookId}")
public Book getBook(@PathVariable String bookId) {}

但是，我正在努力对此进行测试，因为这是微服务应用程序的一部分，在该应用程序中，身份验证和服务不是同一项目的一部分，而是可以在单独的VM上运行。理想情况下，我希望能够在每个服务中模拟令牌并在bookId集中添加所需的任何值。我知道在春季4之后，我们可以使用@WithMockUser，但是据我所知，这仅限于用户名/密码/角色/权限，例如：

@WithMockUser(username = "ram", roles={"ADMIN"})

我真正想做的是扩展此注释以支持我的自定义属性'bookId'或在其中注入模拟集的方法。这有可能吗？如果没有，那我有什么选择，因为我无法在单元测试中调用身份验证提供程序，因为该实现将存在于单独的应用程序中托管的另一个spring上下文中。

非常感谢！

Answer 1

您可以使用@WithUserDetails注释。
如果您有一个用户名“ user1”的用户，则可以用@WithUserDetails("user1")注释测试，它将以CustomUser主体执行，包括与“ user1”关联的自定义属性“ bookId”。
您可以在spring-security docs中找到有关注释的更多信息。

如果需要更大的灵活性，还可以使用@WithSecurityContext注释。
这使您可以创建自己的注释，例如@WithMockCustomUser，您可以在其中自行设置安全上下文。
您可以在spring-security docs中找到更多详细信息。

还有一个选项，如果您使用MockMvc来运行测试，则可以使用request post processor。
您的测试看起来像这样

mvc
    .perform(get("/api/books/1")
    .with(user(customUser)));

其中customUser是您在测试中创建的CustomUser的实例。

@WithMockUser和自定义用户实现

1 个答案: