/* * ============LICENSE_START========================================== * ONAP Portal SDK * =================================================================== * Copyright © 2018 AT&T Intellectual Property. All rights reserved. * =================================================================== * * Unless otherwise specified, all software contained herein is licensed * under the Apache License, Version 2.0 (the "License"); * you may not use this software except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * 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. * * Unless otherwise specified, all documentation contained herein is licensed * under the Creative Commons License, Attribution 4.0 Intl. (the "License"); * you may not use this documentation except in compliance with the License. * You may obtain a copy of the License at * * https://creativecommons.org/licenses/by/4.0/ * * Unless required by applicable law or agreed to in writing, documentation * 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. * * ============LICENSE_END============================================ * * */ package org.onap.portalsdk.core.web.support; import java.util.ArrayList; import java.util.Enumeration; import java.util.HashSet; import java.util.List; import java.util.Set; import java.util.SortedSet; import java.util.TreeSet; import javax.servlet.ServletContext; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpSession; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.mockito.Mock; import org.mockito.Mockito; import org.onap.portalsdk.core.domain.MenuData; import org.onap.portalsdk.core.domain.Role; import org.onap.portalsdk.core.domain.RoleFunction; import org.onap.portalsdk.core.domain.User; import org.onap.portalsdk.core.util.SystemProperties; import org.powermock.api.mockito.PowerMockito; import org.powermock.core.classloader.annotations.PrepareForTest; import org.powermock.modules.junit4.PowerMockRunner; @RunWith(PowerMockRunner.class) @PrepareForTest({AppUtils.class, SystemProperties.class}) public class UserUtilsTest { @Mock HttpServletRequest request; @Mock HttpSession session; @Mock User user; @Mock ServletContext servletContext; SortedSet roles; @Before public void setUp() { PowerMockito.mockStatic(AppUtils.class); PowerMockito.mockStatic(SystemProperties.class); Mockito.when(AppUtils.getSession(request)).thenReturn(session); Mockito.when(request.getSession(true)).thenReturn(session); Mockito.when(request.getSession()).thenReturn(session); Mockito.when(session.getServletContext()).thenReturn(servletContext); Mockito.when(SystemProperties.getProperty(SystemProperties.USER_ATTRIBUTE_NAME)).thenReturn("user"); Mockito.when(SystemProperties.getProperty(SystemProperties.APPLICATION_USER_ID)).thenReturn("1"); Mockito.when(session.getAttribute("user")).thenReturn(user); SortedSet roles = new TreeSet<>(); Role role = new Role(); role.setActive(true); role.setId(1L); Set roleFunctionSet = new HashSet<>(); RoleFunction rf = new RoleFunction(); rf.setName("test"); rf.setCode("test"); rf.setType("test"); rf.setAction("test"); roleFunctionSet.add(rf); role.setRoleFunctions(roleFunctionSet); Role childRole = new Role(); childRole.setActive(true); childRole.setId(1L); HashSet childRoles = new HashSet<>(); childRoles.add(childRole); role.setChildRoles(childRoles); roles.add(role); Mockito.when(user.getRoles()).thenReturn(roles); } @Test public void testSetUserSession() { Set set = new HashSet<>(); MenuData menuData = new MenuData(); menuData.setFunctionCd("test"); set.add(menuData); List roleFunctionList = new ArrayList<>(); roleFunctionList.add(new RoleFunction()); UserUtils.setUserSession(request, user, set, set, "test", roleFunctionList); } @Test public void testHasRole() { UserUtils.hasRole(request, "1"); } @Test public void testHasRoleByUser() { UserUtils.hasRole(user, "1"); } @Test public void testGetLoginMethod() { UserUtils.getLoginMethod(request); } @Test public void testGetUserId() { UserUtils.getUserId(request); } @Test public void testGetStackTrace() { UserUtils.getStackTrace(new Exception("test exception")); } @Test public void testConvertToEcompUser() { UserUtils.convertToEcompUser(user); } @Test public void testGetRequestId() { Enumeration headerNames = Mockito.mock(Enumeration.class); Mockito.when(headerNames.hasMoreElements()).thenReturn(true); Mockito.when(headerNames.nextElement()).thenReturn("X-ECOMP-RequestID"); Mockito.when(request.getHeader("X-ECOMP-RequestID")).thenReturn("test"); Mockito.when(request.getHeaderNames()).thenReturn(headerNames); UserUtils.getRequestId(request); } }