X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=POLICY-SDK-APP%2Fsrc%2Ftest%2Fjava%2Forg%2Fonap%2Fpolicy%2Fcontroller%2FAdminTabControllerTest.java;h=eccab040c2d68b80348a1a10ef0b585d791cea6f;hb=e37f5a70bcf3fd8073418c13416cc79c0ef47d59;hp=17ad56457c189e6a18b57d3d498204af1d1ef2c6;hpb=428150834ee60899b9a8da019bae3c8bf009adf1;p=policy%2Fengine.git

diff --git a/POLICY-SDK-APP/src/test/java/org/onap/policy/controller/AdminTabControllerTest.java b/POLICY-SDK-APP/src/test/java/org/onap/policy/controller/AdminTabControllerTest.java
index 17ad56457..eccab040c 100644
--- a/POLICY-SDK-APP/src/test/java/org/onap/policy/controller/AdminTabControllerTest.java
+++ b/POLICY-SDK-APP/src/test/java/org/onap/policy/controller/AdminTabControllerTest.java
@@ -2,14 +2,16 @@
  * ============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.
@@ -17,22 +19,26 @@
  * 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;
@@ -46,61 +52,76 @@ import org.springframework.mock.web.MockHttpServletResponse;
 
 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));
+    }
 }