From 0f3f8485db14015be5905e18302bcb6eca124590 Mon Sep 17 00:00:00 2001 From: Joe Grandja Date: Mon, 1 Aug 2022 15:15:10 -0400 Subject: [PATCH] Fix OAuth2AuthorizationCodeGrantTests Issue gh-482 --- .../web/configurers/OAuth2AuthorizationCodeGrantTests.java | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/oauth2-authorization-server/src/test/java/org/springframework/security/oauth2/server/authorization/config/annotation/web/configurers/OAuth2AuthorizationCodeGrantTests.java b/oauth2-authorization-server/src/test/java/org/springframework/security/oauth2/server/authorization/config/annotation/web/configurers/OAuth2AuthorizationCodeGrantTests.java index 5d6d65d6..1f31b281 100644 --- a/oauth2-authorization-server/src/test/java/org/springframework/security/oauth2/server/authorization/config/annotation/web/configurers/OAuth2AuthorizationCodeGrantTests.java +++ b/oauth2-authorization-server/src/test/java/org/springframework/security/oauth2/server/authorization/config/annotation/web/configurers/OAuth2AuthorizationCodeGrantTests.java @@ -660,9 +660,8 @@ public class OAuth2AuthorizationCodeGrantTests { ArgumentCaptor securityContextCaptor = ArgumentCaptor.forClass(org.springframework.security.core.context.SecurityContext.class); - verify(securityContextRepository, times(2)).saveContext(securityContextCaptor.capture(), any(), any()); - securityContextCaptor.getAllValues().forEach(securityContext -> - assertThat(securityContext.getAuthentication()).isInstanceOf(UsernamePasswordAuthenticationToken.class)); + verify(securityContextRepository, times(1)).saveContext(securityContextCaptor.capture(), any(), any()); + assertThat(securityContextCaptor.getValue().getAuthentication()).isInstanceOf(UsernamePasswordAuthenticationToken.class); reset(securityContextRepository); String authorizationCode = extractParameterFromRedirectUri(mvcResult.getResponse().getRedirectedUrl(), "code");