SecurityContextHolder Deferred SecurityContext
Closes gh-10913
This commit is contained in:
+3
-2
@@ -17,6 +17,7 @@
|
||||
package org.springframework.security.web.context;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.function.Supplier;
|
||||
|
||||
import javax.servlet.FilterChain;
|
||||
import javax.servlet.ServletException;
|
||||
@@ -62,9 +63,9 @@ public class SecurityContextHolderFilter extends OncePerRequestFilter {
|
||||
@Override
|
||||
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain)
|
||||
throws ServletException, IOException {
|
||||
SecurityContext securityContext = this.securityContextRepository.loadContext(request).get();
|
||||
Supplier<SecurityContext> deferredContext = this.securityContextRepository.loadContext(request);
|
||||
try {
|
||||
this.securityContextHolderStrategy.setContext(securityContext);
|
||||
this.securityContextHolderStrategy.setDeferredContext(deferredContext);
|
||||
filterChain.doFilter(request, response);
|
||||
}
|
||||
finally {
|
||||
|
||||
+5
-1
@@ -16,6 +16,8 @@
|
||||
|
||||
package org.springframework.security.web.context;
|
||||
|
||||
import java.util.function.Supplier;
|
||||
|
||||
import javax.servlet.FilterChain;
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
@@ -94,7 +96,9 @@ class SecurityContextHolderFilterTests {
|
||||
this.filter.setSecurityContextHolderStrategy(this.strategy);
|
||||
this.filter.doFilter(this.request, this.response, filterChain);
|
||||
|
||||
verify(this.strategy).setContext(expectedContext);
|
||||
ArgumentCaptor<Supplier<SecurityContext>> deferredContextArg = ArgumentCaptor.forClass(Supplier.class);
|
||||
verify(this.strategy).setDeferredContext(deferredContextArg.capture());
|
||||
assertThat(deferredContextArg.getValue().get()).isEqualTo(expectedContext);
|
||||
verify(this.strategy).clearContext();
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user