* ============LICENSE_START=======================================================
* ONAP Policy Engine
* ================================================================================
- * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * Copyright (C) 2017, 2019-2020 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Modifications Copyright (C) 2019 Samsung
* ================================================================================
* Licensed 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
- *
+ *
* 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.
* limitations under the License.
* ============LICENSE_END=========================================================
*/
+
package org.onap.policy.controller;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import java.io.BufferedReader;
+import java.io.IOException;
import java.io.StringReader;
import java.io.UnsupportedEncodingException;
import java.util.ArrayList;
import java.util.List;
-
import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
-
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mockito;
public class AdminTabControllerTest {
- private static Logger logger = FlexLogger.getLogger(AdminTabControllerTest.class);
- private static CommonClassDao commonClassDao;
- private HttpServletRequest request;
- private MockHttpServletResponse response;
-
- @Before
- public void setUp() throws Exception {
+ private static Logger logger = FlexLogger.getLogger(AdminTabControllerTest.class);
+ private static CommonClassDao commonClassDao;
+ private HttpServletRequest request;
+ private MockHttpServletResponse response;
+ private AdminTabController admin;
- logger.info("setUp: Entering");
+ /**
+ * Before.
+ *
+ * @throws Exception Exception
+ */
+ @Before
+ public void setUp() throws Exception {
+ admin = new AdminTabController();
+ logger.info("setUp: Entering");
commonClassDao = mock(CommonClassDao.class);
-
- request = mock(HttpServletRequest.class);
- response = new MockHttpServletResponse();
-
- HttpSession mockSession = mock(HttpSession.class);
- User user = new User();
- user.setOrgUserId("Test");
- Mockito.when(mockSession.getAttribute(SystemProperties.getProperty("user_attribute_name"))).thenReturn(user);
- Mockito.when(request.getSession(false)).thenReturn(mockSession);
-
- AdminTabController.setCommonClassDao(commonClassDao);
-
+ request = mock(HttpServletRequest.class);
+ response = new MockHttpServletResponse();
+ HttpSession mockSession = mock(HttpSession.class);
+ User user = new User();
+ user.setOrgUserId("Test");
+ Mockito.when(mockSession.getAttribute(SystemProperties.getProperty("user_attribute_name"))).thenReturn(user);
+ Mockito.when(request.getSession(false)).thenReturn(mockSession);
+ AdminTabController.setCommonClassDao(commonClassDao);
GlobalRoleSettings globalRole = new GlobalRoleSettings();
globalRole.setLockdown(true);
globalRole.setRole("super-admin");
List<Object> globalRoles = new ArrayList<>();
globalRoles.add(globalRole);
when(commonClassDao.getData(GlobalRoleSettings.class)).thenReturn(globalRoles);
- }
-
- @Test
- public void testGetAdminRole(){
- AdminTabController admin = new AdminTabController();
- try {
- admin.getAdminTabEntityData(request, response);
- assertTrue(response.getContentAsString() != null && response.getContentAsString().contains("lockdowndata"));
- } catch (UnsupportedEncodingException e) {
- logger.error("Exception Occured"+e);
- fail();
- }
- }
-
- @Test
- public void testSaveAdminRole() throws Exception{
- AdminTabController admin = new AdminTabController();
- String data = "{\"lockdowndata\":{\"lockdown\":true}}";
- BufferedReader reader = new BufferedReader(new StringReader(data));
- try {
- when(request.getReader()).thenReturn(reader);
- admin.saveAdminTabLockdownValue(request, response);
- assertTrue(response.getContentAsString() != null && response.getContentAsString().contains("descriptiveScopeDictionaryDatas"));
- } catch (UnsupportedEncodingException e) {
- logger.error("Exception Occured"+e);
- fail();
- }
- }
-
+ }
+
+ @Test
+ public void testGetAdminRole() {
+ assertNotNull(AdminTabController.getCommonClassDao());
+ try {
+ admin.getAdminTabEntityData(request, response);
+ assertTrue(response.getContentAsString() != null && response.getContentAsString().contains("lockdowndata"));
+ } catch (UnsupportedEncodingException e) {
+ logger.error("Exception Occured" + e);
+ fail();
+ }
+ }
+
+ @Test
+ public void testSaveAdminRole() throws Exception {
+ String data = "{\"lockdowndata\":{\"lockdown\":true}}";
+ BufferedReader reader = new BufferedReader(new StringReader(data));
+ try {
+ when(request.getReader()).thenReturn(reader);
+ admin.saveAdminTabLockdownValue(request, response);
+ assertTrue(response.getContentAsString() != null
+ && response.getContentAsString().contains("descriptiveScopeDictionaryDatas"));
+ } catch (UnsupportedEncodingException e) {
+ logger.error("Exception Occured" + e);
+ fail();
+ }
+ }
+
+ @SuppressWarnings("unchecked")
+ @Test
+ public void testGetAdminTabEntityDataException() throws IOException {
+ HttpServletResponse mockResponse = Mockito.mock(HttpServletResponse.class);
+ when(mockResponse.getWriter()).thenThrow(IOException.class);
+ admin.getAdminTabEntityData(request, mockResponse);
+ verify(mockResponse).getWriter();
+ }
+
+ @Test
+ public void testSaveAdminTabLockdownValueException() throws IOException {
+ assertNull(admin.saveAdminTabLockdownValue(request, response));
+ }
}