449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings/*
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings * The contents of this file are subject to the terms of the Common Development and
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings * Distribution License (the License). You may not use this file except in compliance with the
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings * License.
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings *
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings * You can obtain a copy of the License at legal/CDDLv1.0.txt. See the License for the
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings * specific language governing permission and limitations under the License.
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings *
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings * When distributing Covered Software, include this CDDL Header Notice in each file and include
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings * the License file at legal/CDDLv1.0.txt. If applicable, add the following below the CDDL
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings * Header, with the fields enclosed by brackets [] replaced by your own identifying
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings * information: "Portions copyright [year] [name of copyright owner]".
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings *
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings * Copyright 2015 ForgeRock AS.
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings */
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingspackage com.sun.identity.saml2.profile;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport static org.mockito.Matchers.any;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport static org.mockito.Matchers.anyBoolean;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport static org.mockito.Mockito.when;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport org.forgerock.openam.saml2.IDPRequestValidator;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport org.forgerock.openam.saml2.IDPSSOFederateRequest;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport org.forgerock.openam.saml2.SAML2ActorFactory;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport org.forgerock.openam.saml2.SAMLAuthenticator;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport org.forgerock.openam.saml2.SAMLAuthenticatorLookup;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport org.mockito.Mock;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport org.mockito.Mockito;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport org.mockito.MockitoAnnotations;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport org.testng.annotations.BeforeMethod;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport org.testng.annotations.Test;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport javax.servlet.http.HttpServletRequest;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport javax.servlet.http.HttpServletResponse;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingsimport java.io.PrintWriter;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbingspublic class IDPSSOFederateTest {
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings @Mock
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings private HttpServletRequest mockRequest;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings @Mock
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings private HttpServletResponse mockResponse;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings @Mock
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings private PrintWriter mockPrintWriter;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings @Mock
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings private FederateCookieRedirector cookieRedirector;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings @Mock
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings private SAML2ActorFactory actorFactory;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings @Mock
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings private IDPRequestValidator validator;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings @Mock
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings private SAMLAuthenticator authenticator;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings @Mock
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings private SAMLAuthenticatorLookup authenticationLookup;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings private IDPSSOFederate idpSsoFederateRequest;
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings @BeforeMethod
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings public void initMocks() throws ServerFaultException, ClientFaultException {
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings MockitoAnnotations.initMocks(this);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings when(actorFactory.getIDPRequestValidator(Mockito.anyString(), anyBoolean())).thenReturn(validator);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings when(actorFactory.getSAMLAuthenticator(
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings any(IDPSSOFederateRequest.class),
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings any(HttpServletRequest.class),
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings any(HttpServletResponse.class),
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings any(PrintWriter.class),
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings anyBoolean())).thenReturn(authenticator);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings when(actorFactory.getSAMLAuthenticatorLookup(
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings any(IDPSSOFederateRequest.class),
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings any(HttpServletRequest.class),
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings any(HttpServletResponse.class),
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings any(PrintWriter.class))).thenReturn(authenticationLookup);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings idpSsoFederateRequest = new IDPSSOFederate(false, cookieRedirector, actorFactory);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings }
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings @Test
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings public void shouldBeTestable() throws Exception {
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings idpSsoFederateRequest.process(mockRequest, mockResponse, mockPrintWriter, null);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings }
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings @Test
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings public void shouldNotCallAnyFurtherFunctionsAfterNeedSetLBCookieAndRedirectReturnsTrue() throws Exception {
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings // Arrange
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings when(cookieRedirector.needSetLBCookieAndRedirect(
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings any(HttpServletRequest.class),
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings any(HttpServletResponse.class),
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings anyBoolean())).thenReturn(true);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings // Act
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings idpSsoFederateRequest.process(mockRequest, mockResponse, mockPrintWriter, null);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings // Assert
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings Mockito.verifyZeroInteractions(authenticator, authenticationLookup);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings }
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings @Test
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings public void shouldCallAuthenticateIfThereIsNoRequestId() throws Exception {
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings // Arrange
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings when(mockRequest.getParameter("ReqID")).thenReturn("");
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings when(cookieRedirector.needSetLBCookieAndRedirect(
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings any(HttpServletRequest.class),
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings any(HttpServletResponse.class),
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings anyBoolean())).thenReturn(false);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings // Act
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings idpSsoFederateRequest.process(mockRequest, mockResponse, mockPrintWriter, null);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings // Assert
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings Mockito.verify(authenticator).authenticate();
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings Mockito.verifyZeroInteractions(authenticationLookup);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings }
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings @Test
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings public void shouldCallAuthenticateLookupIfThereIsARequestId() throws Exception {
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings // Arrange
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings when(mockRequest.getParameter("ReqID")).thenReturn("12345");
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings when(cookieRedirector.needSetLBCookieAndRedirect(
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings any(HttpServletRequest.class),
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings any(HttpServletResponse.class),
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings anyBoolean())).thenReturn(false);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings // Act
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings idpSsoFederateRequest.process(mockRequest, mockResponse, mockPrintWriter, null);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings // Assert
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings Mockito.verify(authenticationLookup).retrieveAuthenticationFromCache();
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings Mockito.verifyZeroInteractions(authenticator);
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings }
449854c2a07b50ea64d9d6a8b03d18d4afeeee43Ken Stubbings}