ecomp-portal-BE-common/src/test/java/org/openecomp/portalapp/portal/controller/UserNotificationControllerTest.java

   1 /*-
   2  * ============LICENSE_START==========================================
   3  * ONAP Portal
   4  * ===================================================================
   5  * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
   6  * ===================================================================
   7  *
   8  * Unless otherwise specified, all software contained herein is licensed
   9  * under the Apache License, Version 2.0 (the “License”);
  10  * you may not use this software except in compliance with the License.
  11  * You may obtain a copy of the License at
  12  *
  13  *             http://www.apache.org/licenses/LICENSE-2.0
  14  *
  15  * Unless required by applicable law or agreed to in writing, software
  16  * distributed under the License is distributed on an "AS IS" BASIS,
  17  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  18  * See the License for the specific language governing permissions and
  19  * limitations under the License.
  20  *
  21  * Unless otherwise specified, all documentation contained herein is licensed
  22  * under the Creative Commons License, Attribution 4.0 Intl. (the “License”);
  23  * you may not use this documentation except in compliance with the License.
  24  * You may obtain a copy of the License at
  25  *
  26  *             https://creativecommons.org/licenses/by/4.0/
  27  *
  28  * Unless required by applicable law or agreed to in writing, documentation
  29  * distributed under the License is distributed on an "AS IS" BASIS,
  30  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  31  * See the License for the specific language governing permissions and
  32  * limitations under the License.
  33  *
  34  * ============LICENSE_END============================================
  35  *
  36  * ECOMP is a trademark and service mark of AT&T Intellectual Property.
  37  */
  38 package org.openecomp.portalapp.portal.controller;
  39
  40 import static org.junit.Assert.assertTrue;
  41
  42 import java.util.ArrayList;
  43 import java.util.Calendar;
  44 import java.util.Date;
  45 import java.util.List;
  46
  47 import javax.servlet.http.HttpServletRequest;
  48 import javax.servlet.http.HttpServletResponse;
  49 import javax.servlet.http.HttpSession;
  50
  51 import org.junit.Before;
  52 import org.junit.Test;
  53 import org.junit.runner.RunWith;
  54 import org.mockito.InjectMocks;
  55 import org.mockito.Matchers;
  56 import org.mockito.Mock;
  57 import org.mockito.Mockito;
  58 import org.mockito.MockitoAnnotations;
  59 import org.openecomp.portalapp.portal.framework.MockitoTestSuite;
  60 import org.openecomp.portalapp.portal.controller.UserNotificationController;
  61 import org.openecomp.portalapp.portal.core.MockEPUser;
  62 import org.openecomp.portalapp.portal.domain.EPUser;
  63 import org.openecomp.portalapp.portal.ecomp.model.PortalRestResponse;
  64 import org.openecomp.portalapp.portal.ecomp.model.PortalRestStatusEnum;
  65 import org.openecomp.portalapp.portal.service.FunctionalMenuService;
  66 import org.openecomp.portalapp.portal.service.FunctionalMenuServiceImpl;
  67 import org.openecomp.portalapp.portal.service.UserNotificationService;
  68 import org.openecomp.portalapp.portal.service.UserNotificationServiceImpl;
  69 import org.openecomp.portalapp.portal.transport.EpNotificationItem;
  70 import org.openecomp.portalapp.portal.transport.EpNotificationItemVO;
  71 import org.openecomp.portalapp.portal.transport.FunctionalMenuRole;
  72 import org.openecomp.portalapp.util.EPUserUtils;
  73 import org.openecomp.portalsdk.core.util.SystemProperties;
  74 import org.openecomp.portalsdk.core.web.support.UserUtils;
  75 import org.powermock.api.mockito.PowerMockito;
  76 import org.powermock.core.classloader.annotations.PrepareForTest;
  77 import org.powermock.modules.junit4.PowerMockRunner;
  78
  79 @RunWith(PowerMockRunner.class)
  80 @PrepareForTest(UserUtils.class)
  81 public class UserNotificationControllerTest {
  82
  83         @Mock
  84         FunctionalMenuService functionalMenuService = new FunctionalMenuServiceImpl();
  85
  86         @Mock
  87         UserNotificationService userNotificationService = new UserNotificationServiceImpl();
  88
  89         @InjectMocks
  90         UserNotificationController userNotificationController = new UserNotificationController();
  91
  92         @Before
  93         public void setup() {
  94                 MockitoAnnotations.initMocks(this);
  95         }
  96
  97         MockEPUser mockUser = new MockEPUser();
  98         MockitoTestSuite mockitoTestSuite = new MockitoTestSuite();
  99
 100         HttpServletRequest mockedRequest = mockitoTestSuite.getMockedRequest();
 101         HttpServletResponse mockedResponse = mockitoTestSuite.getMockedResponse();
 102         NullPointerException nullPointerException = new NullPointerException();
 103
 104         @Test
 105         public void getMenuIdRoleIdTest() {
 106                 List<FunctionalMenuRole> expectedMenuRoleList = new ArrayList<FunctionalMenuRole>();
 107                 FunctionalMenuRole functionalMenuRole = new FunctionalMenuRole();
 108                 functionalMenuRole.setId(new Integer(99999999));
 109                 functionalMenuRole.setMenuId((long) 137);
 110                 functionalMenuRole.setAppId(new Integer(456));
 111                 functionalMenuRole.setRoleId(new Integer(6214));
 112                 expectedMenuRoleList.add(functionalMenuRole);
 113                 List<FunctionalMenuRole> actualFunctionalMenuRoleList = null;
 114                 Mockito.when(functionalMenuService.getFunctionalMenuRole()).thenReturn(expectedMenuRoleList);
 115                 actualFunctionalMenuRoleList = userNotificationController.getMenuIdRoleId(mockedRequest, mockedResponse);
 116                 assertTrue(actualFunctionalMenuRoleList.equals(expectedMenuRoleList));
 117
 118         }
 119
 120         @Test
 121         public void getNotificationsTest() {
 122                 EPUser user = mockUser.mockEPUser();
 123                 HttpSession session = mockedRequest.getSession();
 124                 session.setAttribute(SystemProperties.getProperty(SystemProperties.USER_ATTRIBUTE_NAME), user);
 125                 Mockito.when(EPUserUtils.getUserSession(mockedRequest)).thenReturn(user);
 126                 List<EpNotificationItem> expectedEpNotificationList = new ArrayList<EpNotificationItem>();
 127                 EpNotificationItem epNotificationItem = new EpNotificationItem();
 128                 epNotificationItem.setNotificationId((long) 200);
 129                 expectedEpNotificationList.add(epNotificationItem);
 130                 PortalRestResponse<List<EpNotificationItem>> expectedportalRestResponse = new PortalRestResponse<List<EpNotificationItem>>();
 131                 expectedportalRestResponse.setMessage("success");
 132                 expectedportalRestResponse.setResponse(expectedEpNotificationList);
 133                 expectedportalRestResponse.setStatus(PortalRestStatusEnum.OK);
 134                 PortalRestResponse<List<EpNotificationItem>> actualPortalRestResponse = null;
 135                 Mockito.when(userNotificationService.getNotifications(user.getId())).thenReturn(expectedEpNotificationList);
 136                 actualPortalRestResponse = userNotificationController.getNotifications(mockedRequest, mockedResponse);
 137                 assertTrue(expectedportalRestResponse.equals(actualPortalRestResponse));
 138
 139         }
 140
 141         @Test
 142         public void getNotificationsCatchesExceptionTest() {
 143                 EPUser user = mockUser.mockEPUser();
 144                 HttpSession session = mockedRequest.getSession();
 145                 session.setAttribute(SystemProperties.getProperty(SystemProperties.USER_ATTRIBUTE_NAME), user);
 146                 Mockito.when(EPUserUtils.getUserSession(mockedRequest)).thenReturn(user);
 147                 List<EpNotificationItem> expectedEpNotificationList = null;
 148                 PortalRestResponse<List<EpNotificationItem>> expectedportalRestResponse = new PortalRestResponse<List<EpNotificationItem>>();
 149                 expectedportalRestResponse.setMessage(null);
 150                 expectedportalRestResponse.setResponse(expectedEpNotificationList);
 151                 expectedportalRestResponse.setStatus(PortalRestStatusEnum.ERROR);
 152                 PortalRestResponse<List<EpNotificationItem>> actualPortalRestResponse = null;
 153                 Mockito.when(userNotificationService.getNotifications(user.getId())).thenThrow(new NullPointerException());
 154                 actualPortalRestResponse = userNotificationController.getNotifications(mockedRequest, mockedResponse);
 155                 assertTrue(expectedportalRestResponse.equals(actualPortalRestResponse));
 156         }
 157
 158         @Test
 159         public void getAdminNotificationsTest() {
 160                 EPUser user = mockUser.mockEPUser();
 161                 HttpSession session = mockedRequest.getSession();
 162                 session.setAttribute(SystemProperties.getProperty(SystemProperties.USER_ATTRIBUTE_NAME), user);
 163                 Mockito.when(EPUserUtils.getUserSession(mockedRequest)).thenReturn(user);
 164                 List<EpNotificationItemVO> actualEpNotificationsList = new ArrayList<EpNotificationItemVO>();
 165                 List<EpNotificationItemVO> expectedEpNotificationsList = new ArrayList<EpNotificationItemVO>();
 166                 EpNotificationItemVO epNotificationItemVO = new EpNotificationItemVO();
 167                 epNotificationItemVO.setId((long) 1);
 168                 expectedEpNotificationsList.add(epNotificationItemVO);
 169                 Mockito.when(userNotificationService.getAdminNotificationVOS(Matchers.anyLong())).thenReturn(expectedEpNotificationsList);
 170                 actualEpNotificationsList = userNotificationController.getAdminNotifications(mockedRequest, mockedResponse);
 171                 assertTrue(actualEpNotificationsList.equals(expectedEpNotificationsList));
 172         }
 173
 174         @Test
 175         public void saveTestWhenNotificationIsNull() throws Exception {
 176                 EPUser user = mockUser.mockEPUser();
 177                 HttpSession session = mockedRequest.getSession();
 178                 session.setAttribute(SystemProperties.getProperty(SystemProperties.USER_ATTRIBUTE_NAME), user);
 179                 Mockito.when(EPUserUtils.getUserSession(mockedRequest)).thenReturn(user);
 180
 181                 PortalRestResponse<String> actualPortalRestResponse = new PortalRestResponse<String>();
 182                 PortalRestResponse<String> expectedPortalRestResponse = new PortalRestResponse<String>();
 183                 expectedPortalRestResponse.setMessage("FAILURE");
 184                 expectedPortalRestResponse.setResponse("Notification Header cannot be null or empty");
 185                 expectedPortalRestResponse.setStatus(PortalRestStatusEnum.ERROR);
 186
 187                 EpNotificationItem notificationItem = null;
 188                 actualPortalRestResponse = userNotificationController.save(mockedRequest, mockedResponse, notificationItem);
 189                 assertTrue(actualPortalRestResponse.equals(expectedPortalRestResponse));
 190         }
 191
 192         @Test
 193         public void saveTestWhenEndTimeIsGreater() throws Exception {
 194                 EPUser user = mockUser.mockEPUser();
 195                 HttpSession session = mockedRequest.getSession();
 196                 session.setAttribute(SystemProperties.getProperty(SystemProperties.USER_ATTRIBUTE_NAME), user);
 197                 Mockito.when(EPUserUtils.getUserSession(mockedRequest)).thenReturn(user);
 198                 PortalRestResponse<String> actualPortalRestResponse = new PortalRestResponse<String>();
 199                 PortalRestResponse<String> expectedPortalRestResponse = new PortalRestResponse<String>();
 200                 expectedPortalRestResponse.setMessage("FAILURE");
 201                 expectedPortalRestResponse.setResponse("End Time should be greater than  start time");
 202                 expectedPortalRestResponse.setStatus(PortalRestStatusEnum.ERROR);
 203                 EpNotificationItem notificationItem = new EpNotificationItem();
 204                 notificationItem.setNotificationId((long) 1);
 205                 notificationItem.setMsgHeader("Test");
 206                 Date currentDate = new Date();
 207                 Calendar c = Calendar.getInstance();
 208                 c.setTime(currentDate);
 209                 c.add(Calendar.DATE, 1);
 210                 Date currentDatePlusOne = c.getTime();
 211                 notificationItem.setStartTime(currentDatePlusOne);
 212                 notificationItem.setEndTime(currentDate);
 213
 214                 actualPortalRestResponse = userNotificationController.save(mockedRequest, mockedResponse, notificationItem);
 215                 assertTrue(actualPortalRestResponse.equals(expectedPortalRestResponse));
 216
 217         }
 218
 219         @Test
 220         public void saveTestWhenNoRoleIDExists() throws Exception {
 221                 EPUser user = mockUser.mockEPUser();
 222                 HttpSession session = mockedRequest.getSession();
 223                 session.setAttribute(SystemProperties.getProperty(SystemProperties.USER_ATTRIBUTE_NAME), user);
 224                 Mockito.when(EPUserUtils.getUserSession(mockedRequest)).thenReturn(user);
 225                 PortalRestResponse<String> actualPortalRestResponse = new PortalRestResponse<String>();
 226                 PortalRestResponse<String> expectedPortalRestResponse = new PortalRestResponse<String>();
 227                 expectedPortalRestResponse.setMessage("FAILURE");
 228                 expectedPortalRestResponse.setResponse("No Roles Ids Exist for the selected Roles");
 229                 expectedPortalRestResponse.setStatus(PortalRestStatusEnum.ERROR);
 230                 EpNotificationItem notificationItem = new EpNotificationItem();
 231                 notificationItem.setNotificationId((long) 1);
 232                 notificationItem.setMsgHeader("Test");
 233                 notificationItem.setIsForAllRoles("N");
 234                 Date currentDate = new Date();
 235                 Calendar c = Calendar.getInstance();
 236                 c.setTime(currentDate);
 237                 c.add(Calendar.DATE, 1);
 238                 Date currentDatePlusOne = c.getTime();
 239                 notificationItem.setStartTime(currentDate);
 240                 notificationItem.setEndTime(currentDatePlusOne);
 241                 List<Long> roleList = new ArrayList<Long>();
 242                 notificationItem.setRoleIds(roleList);
 243                 actualPortalRestResponse = userNotificationController.save(mockedRequest, mockedResponse, notificationItem);
 244                 assertTrue(actualPortalRestResponse.equals(expectedPortalRestResponse));
 245         }
 246
 247         @Test
 248         public void saveTest() throws Exception {
 249                 EPUser user = mockUser.mockEPUser();
 250                 HttpSession session = mockedRequest.getSession();
 251                 session.setAttribute("user", user);
 252                 Mockito.when(EPUserUtils.getUserSession(mockedRequest)).thenReturn(user);
 253                 PortalRestResponse<String> actualPortalRestResponse = new PortalRestResponse<String>();
 254                 PortalRestResponse<String> expectedPortalRestResponse = new PortalRestResponse<String>();
 255                 expectedPortalRestResponse.setMessage("SUCCESS");
 256                 expectedPortalRestResponse.setResponse("");
 257                 expectedPortalRestResponse.setStatus(PortalRestStatusEnum.OK);
 258                 EpNotificationItem notificationItem = new EpNotificationItem();
 259                 notificationItem.setNotificationId((long) 1);
 260                 notificationItem.setMsgHeader("Test");
 261                 notificationItem.setIsForAllRoles("Y");
 262                 Date currentDate = new Date();
 263                 Calendar c = Calendar.getInstance();
 264                 c.setTime(currentDate);
 265                 c.add(Calendar.DATE, 1);
 266                 Date currentDatePlusOne = c.getTime();
 267                 notificationItem.setStartTime(currentDate);
 268                 notificationItem.setEndTime(currentDatePlusOne);
 269                 List<Long> roleList = new ArrayList<Long>();
 270                 Long role1 = (long) 1;
 271                 roleList.add(role1);
 272                 notificationItem.setRoleIds(roleList);
 273                 HttpServletRequest request = mockitoTestSuite.getMockedRequest();
 274                 PowerMockito.mockStatic(UserUtils.class);
 275                 Mockito.when(UserUtils.getUserIdAsLong(request)).thenReturn((long) 1);
 276                 Mockito.when(userNotificationService.saveNotification(notificationItem)).thenReturn("Test");
 277                 actualPortalRestResponse = userNotificationController.save(mockedRequest, mockedResponse, notificationItem);
 278                 assertTrue(actualPortalRestResponse.equals(expectedPortalRestResponse));
 279         }
 280
 281 }