Spring/spring-security
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add SecurityContextHolderStrategy for Jaas

Browse Source 
Issue gh-11060
Issue gh-11061
This commit is contained in:
Josh Cummings
2022-06-27 16:58:52 -06:00
parent ec1bfa12f0
commit b316a3217b
5 changed files with 65 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
core/src/test/java/org/springframework/security/authentication/jaas/SecurityContextLoginModuleTests.java
View File

@@ -29,9 +29,13 @@ import org.junit.jupiter.api.Test;
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.core.context.SecurityContextHolderStrategy;
import org.springframework.security.core.context.SecurityContextImpl;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
import static org.mockito.BDDMockito.given;
import static org.mockito.Mockito.mock;
/**
* Tests SecurityContextLoginModule
@@ -84,6 +88,18 @@ public class SecurityContextLoginModuleTests {
.withFailMessage("Principals should contain the authentication").isTrue();
}
@Test
public void loginWhenCustomSecurityContextHolderStrategyThenUses() throws Exception {
SecurityContextHolderStrategy securityContextHolderStrategy = mock(SecurityContextHolderStrategy.class);
given(securityContextHolderStrategy.getContext()).willReturn(new SecurityContextImpl(this.auth));
this.module.setSecurityContextHolderStrategy(securityContextHolderStrategy);
assertThat(this.module.login()).as("Login should succeed, there is an authentication set").isTrue();
assertThat(this.module.commit()).withFailMessage("The authentication is not null, this should return true")
.isTrue();
assertThat(this.subject.getPrincipals().contains(this.auth))
.withFailMessage("Principals should contain the authentication").isTrue();
}
@Test
public void testLogout() throws Exception {
SecurityContextHolder.getContext().setAuthentication(this.auth);