|
24 | 24 | import org.easymock.EasyMock; |
25 | 25 | import org.junit.Assert; |
26 | 26 | import org.junit.Test; |
| 27 | +import org.pac4j.core.context.Cookie; |
27 | 28 | import org.pac4j.core.profile.CommonProfile; |
28 | 29 | import org.pac4j.core.util.Pac4jConstants; |
29 | 30 | import org.pac4j.jee.context.JEEContext; |
30 | 31 | import org.pac4j.saml.profile.SAML2Profile; |
31 | 32 |
|
32 | 33 | import javax.servlet.http.HttpServletResponse; |
33 | 34 | import java.util.Arrays; |
| 35 | +import java.util.Collections; |
34 | 36 | import java.util.HashMap; |
35 | 37 | import java.util.HashSet; |
36 | 38 | import java.util.Map; |
|
44 | 46 | import static org.apache.knox.gateway.pac4j.filter.Pac4jDispatcherFilter.PAC4J_SESSION_STORE_EXCLUDE_ROLES; |
45 | 47 | import static org.apache.knox.gateway.pac4j.filter.Pac4jDispatcherFilter.PAC4J_SESSION_STORE_EXCLUDE_ROLES_DEFAULT; |
46 | 48 | import static org.apache.knox.gateway.pac4j.session.KnoxSessionStore.PAC4J_PASSWORD; |
| 49 | +import static org.apache.knox.gateway.pac4j.session.KnoxSessionStore.PAC4J_SESSION_PREFIX; |
47 | 50 |
|
48 | 51 | public class KnoxSessionStoreTest { |
49 | 52 | private static final String CLUSTER_NAME = "knox"; |
@@ -158,4 +161,29 @@ public void filterConfigParamsTest() |
158 | 161 | Assert.assertNotNull(samlProfile.getAttribute("https://knox.apache.org/SAML/Attributes/groups")); |
159 | 162 | Assert.assertNotNull(samlProfile.getAttribute("https://knox.apache.org/SAML/Attributes/groups2")); |
160 | 163 | } |
| 164 | + |
| 165 | + @Test |
| 166 | + public void testNullCookieValue() throws AliasServiceException { |
| 167 | + final AliasService aliasService = EasyMock.createNiceMock(AliasService.class); |
| 168 | + EasyMock.expect(aliasService.getPasswordFromAliasForCluster(CLUSTER_NAME, PAC4J_PASSWORD, true)) |
| 169 | + .andReturn(PAC4J_PASSWORD.toCharArray()).anyTimes(); |
| 170 | + EasyMock.expect(aliasService.getPasswordFromAliasForCluster(CLUSTER_NAME, PAC4J_PASSWORD)) |
| 171 | + .andReturn(PAC4J_PASSWORD.toCharArray()).anyTimes(); |
| 172 | + EasyMock.replay(aliasService); |
| 173 | + |
| 174 | + final DefaultCryptoService cryptoService = new DefaultCryptoService(); |
| 175 | + cryptoService.setAliasService(aliasService); |
| 176 | + |
| 177 | + final Map<String, String> sessionStoreConfigs = new HashMap<>(); |
| 178 | + |
| 179 | + final JEEContext mockContext = EasyMock.createNiceMock(JEEContext.class); |
| 180 | + String keyWithNullValue = "CasClient$attemptedAuthentication"; |
| 181 | + Cookie cookie = new Cookie(PAC4J_SESSION_PREFIX + keyWithNullValue, "null"); |
| 182 | + EasyMock.expect(mockContext.getRequestCookies()).andReturn(Collections.singletonList(cookie)); |
| 183 | + EasyMock.replay(mockContext); |
| 184 | + |
| 185 | + final KnoxSessionStore sessionStore = new KnoxSessionStore(cryptoService, CLUSTER_NAME, null, sessionStoreConfigs); |
| 186 | + Assert.assertTrue(sessionStore.get(mockContext, keyWithNullValue).isEmpty()); |
| 187 | + } |
| 188 | + |
161 | 189 | } |
0 commit comments