/**
 * Licensed to Apereo under one or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information regarding copyright ownership. Apereo
 * licenses this file to you under the Apache License, Version 2.0 (the "License"); you may not use
 * this file except in compliance with the License. You may obtain a copy of the License at the
 * following location:
 *
 * <p>http://www.apache.org/licenses/LICENSE-2.0
 *
 * <p>Unless required by applicable law or agreed to in writing, software distributed under the
 * License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied. See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.apereo.portal.portlets.login;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
import static org.mockito.Mockito.when;

import java.util.HashMap;
import java.util.Map;
import javax.portlet.PortletRequest;
import javax.portlet.PortletSession;
import javax.servlet.http.HttpServletRequest;
import org.apereo.portal.IUserPreferencesManager;
import org.apereo.portal.IUserProfile;
import org.apereo.portal.layout.profile.SessionAttributeProfileMapperImpl;
import org.apereo.portal.url.IPortalRequestUtils;
import org.apereo.portal.user.IUserInstance;
import org.apereo.portal.user.IUserInstanceManager;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

public class LoginPortletHelperTest {

    LoginPortletHelper helper = new LoginPortletHelper();
    Map<String, String> mappings = new HashMap<String, String>();
    @Mock IPortalRequestUtils portalRequestUtils;
    @Mock PortletRequest portletRequest;
    @Mock PortletSession session;
    @Mock HttpServletRequest servletRequest;
    @Mock IUserInstanceManager userInstanceManager;
    @Mock IUserInstance userInstance;
    @Mock IUserPreferencesManager userPreferencesManager;
    @Mock IUserProfile userProfile;

    @Before
    public void setUp() {
        MockitoAnnotations.initMocks(this);
        helper.setPortalRequestUtils(portalRequestUtils);
        helper.setUserInstanceManager(userInstanceManager);

        when(portletRequest.getPortletSession()).thenReturn(session);
        when(portalRequestUtils.getPortletHttpRequest(portletRequest)).thenReturn(servletRequest);
        when(userInstanceManager.getUserInstance(servletRequest)).thenReturn(userInstance);
        when(userInstance.getPreferencesManager()).thenReturn(userPreferencesManager);
        when(userPreferencesManager.getUserProfile()).thenReturn(userProfile);
        when(userProfile.getProfileFname()).thenReturn("default");

        mappings.put("mobile", "mobileDefault");
        mappings.put("desktop", "default");
        helper.setProfileMappings(mappings);
    }

    @Test
    public void testInSessionProfile() {
        when(session.getAttribute(
                        SessionAttributeProfileMapperImpl.DEFAULT_SESSION_ATTRIBUTE_NAME,
                        PortletSession.APPLICATION_SCOPE))
                .thenReturn("mobile");
        String profile = helper.getSelectedProfile(portletRequest);
        assertEquals("mobile", profile);
    }

    @Test
    public void testCurrentProfile() {
        String profile = helper.getSelectedProfile(portletRequest);
        assertEquals("desktop", profile);
    }

    @Test
    public void testUnmatchedProfile() {
        when(userProfile.getProfileFname()).thenReturn("somethingelse");
        String profile = helper.getSelectedProfile(portletRequest);
        assertNull(profile);
    }
}