policy/engine jdk11 upgrades
[policy/engine.git] / POLICY-SDK-APP / src / test / java / org / onap / policy / controller / AdminTabControllerTest.java
index 17ad564..eccab04 100644 (file)
@@ -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.
  * 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));
+    }
 }