我花了一天的时间想办法解决这个问题。我最终得到了一个我认为还不错的解决方案，并且可以帮助很多人。
根据您在测试中尝试执行的操作，您可以发送mockMvc 来测试您的控制器。请注意，未调用 AuthorizationServer。您只需要在资源服务器中进行测试。
- 创建将在
OAuth2AuthenticationProcessingFilter 中使用的bean InMemoryTokenStore 来验证您的用户。很棒的是，您可以在执行测试之前将令牌添加到您的InMemoryTokenStore。 OAuth2AuthenticationProcessingFilter 将根据用户使用的令牌对用户进行身份验证，并且不会调用任何远程服务器。
@Configuration
public class AuthenticationManagerProvider {
@Bean
public TokenStore tokenStore() {
return new InMemoryTokenStore();
}
}
- 注解
@WithMockUser 不适用于OAuth2。事实上，OAuth2AuthenticationProcessingFilter 总是检查你的令牌，而不考虑SecurityContext。我建议使用与@WithMockUser 相同的方法，但使用您在代码库中创建的注释。 （进行一些易于维护和清洁的测试）：
@WithMockOAuth2Scope 几乎包含自定义身份验证所需的所有参数。你可以删除那些你永远不会使用的，但我做了很多以确保你看到可能性。 （将这 2 个类放在您的测试文件夹中）
@Retention(RetentionPolicy.RUNTIME)
@WithSecurityContext(factory = WithMockOAuth2ScopeSecurityContextFactory.class)
public @interface WithMockOAuth2Scope {
String token() default "";
String clientId() default "client-id";
boolean approved() default true;
String redirectUrl() default "";
String[] responseTypes() default {};
String[] scopes() default {};
String[] resourceIds() default {};
String[] authorities() default {};
String username() default "username";
String password() default "";
String email() default "";
}
然后，我们需要一个类来解释这个注解，并用测试所需的数据填充我们的 `InMemoryTokenStore。
@Component
public class WithMockOAuth2ScopeSecurityContextFactory implements WithSecurityContextFactory<WithMockOAuth2Scope> {
@Autowired
private TokenStore tokenStore;
@Override
public SecurityContext createSecurityContext(WithMockOAuth2Scope mockOAuth2Scope) {
OAuth2AccessToken oAuth2AccessToken = createAccessToken(mockOAuth2Scope.token());
OAuth2Authentication oAuth2Authentication = createAuthentication(mockOAuth2Scope);
tokenStore.storeAccessToken(oAuth2AccessToken, oAuth2Authentication);
return SecurityContextHolder.createEmptyContext();
}
private OAuth2AccessToken createAccessToken(String token) {
return new DefaultOAuth2AccessToken(token);
}
private OAuth2Authentication createAuthentication(WithMockOAuth2Scope mockOAuth2Scope) {
OAuth2Request oauth2Request = getOauth2Request(mockOAuth2Scope);
return new OAuth2Authentication(oauth2Request,
getAuthentication(mockOAuth2Scope));
}
private OAuth2Request getOauth2Request(WithMockOAuth2Scope mockOAuth2Scope) {
String clientId = mockOAuth2Scope.clientId();
boolean approved = mockOAuth2Scope.approved();
String redirectUrl = mockOAuth2Scope.redirectUrl();
Set<String> responseTypes = new HashSet<>(asList(mockOAuth2Scope.responseTypes()));
Set<String> scopes = new HashSet<>(asList(mockOAuth2Scope.scopes()));
Set<String> resourceIds = new HashSet<>(asList(mockOAuth2Scope.resourceIds()));
Map<String, String> requestParameters = Collections.emptyMap();
Map<String, Serializable> extensionProperties = Collections.emptyMap();
List<GrantedAuthority> authorities = AuthorityUtils.createAuthorityList(mockOAuth2Scope.authorities());
return new OAuth2Request(requestParameters, clientId, authorities,
approved, scopes, resourceIds, redirectUrl, responseTypes, extensionProperties);
}
private Authentication getAuthentication(WithMockOAuth2Scope mockOAuth2Scope) {
List<GrantedAuthority> grantedAuthorities = AuthorityUtils.createAuthorityList(mockOAuth2Scope.authorities());
String username = mockOAuth2Scope.username();
User userPrincipal = new User(username,
mockOAuth2Scope.password(),
true, true, true, true, grantedAuthorities);
HashMap<String, String> details = new HashMap<>();
details.put("user_name", username);
details.put("email", mockOAuth2Scope.email());
TestingAuthenticationToken token = new TestingAuthenticationToken(userPrincipal, null, grantedAuthorities);
token.setAuthenticated(true);
token.setDetails(details);
return token;
}
}
- 一切都设置好后，在
src/test/java/your/package/ 下创建一个简单的测试类。此类将执行mockMvc 操作，并使用@ WithMockOAuth2Scope 创建测试所需的令牌。
@WebMvcTest(SimpleController.class)
@Import(AuthenticationManagerProvider.class)
class SimpleControllerTest {
@Autowired
private MockMvc mockMvc;
@Test
@WithMockOAuth2Scope(token = "123456789",
authorities = "CAN_READ")
public void test() throws Exception {
mockMvc.perform(get("/whoami")
.header("Authorization", "Bearer 123456789"))
.andExpect(status().isOk())
.andExpect(content().string("username"));
}
}
我想出了这个解决方案，感谢：
对于好奇的人：
测试时，Spring 会加载一个InMemoryTokenStore，并让您有可能采用您提供的一个（作为@Bean）。在生产中运行时，就我而言，Spring 使用RemoteTokenStore，它调用远程授权服务器来检查令牌（http://authorization_server/oauth/check_token）。
当您决定使用 OAuth2 时，Spring 会触发 OAuth2AuthenticationProcessingFilter。这是我在所有调试会话期间的切入点。
我学到了很多，谢谢你。
您可以找到source code here。
希望对您有所帮助！