[Policy-52, Policy-92, Policy-93] Policy Enhancements and bugfixes

[policy/engine.git] / POLICY-SDK-APP / src / test / java / org / openecomp / policy / controller / AdminTabControllerTest.java

diff --git a/POLICY-SDK-APP/src/test/java/org/openecomp/policy/controller/AdminTabControllerTest.java b/POLICY-SDK-APP/src/test/java/org/openecomp/policy/controller/AdminTabControllerTest.java
index 43b8a6f..07bd549 100644 (file)
--- a/POLICY-SDK-APP/src/test/java/org/openecomp/policy/controller/AdminTabControllerTest.java
+++ b/POLICY-SDK-APP/src/test/java/org/openecomp/policy/controller/AdminTabControllerTest.java
@@ -19,10 +19,12 @@
  */
 package org.openecomp.policy.controller;
 
-import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.*;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
+import java.io.BufferedReader;
+import java.io.StringReader;
 import java.io.UnsupportedEncodingException;
 import java.util.ArrayList;
 import java.util.List;
@@ -41,12 +43,20 @@ 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 {
 
                logger.info("setUp: Entering");
         commonClassDao = mock(CommonClassDao.class);
+        
+        request = mock(HttpServletRequest.class);       
+               response =  new MockHttpServletResponse();
+               
+               AdminTabController.setCommonClassDao(commonClassDao);
+               
         GlobalRoleSettings globalRole = new GlobalRoleSettings();
         globalRole.setLockdown(true);
         globalRole.setRole("super-admin");
@@ -57,17 +67,28 @@ public class AdminTabControllerTest {
        
        @Test
        public void testGetAdminRole(){
-               HttpServletRequest request = mock(HttpServletRequest.class);       
-               MockHttpServletResponse response =  new MockHttpServletResponse();
-               
                AdminTabController admin = new AdminTabController();
-               AdminTabController.setCommonClassDao(commonClassDao);
-               admin.getAdminTabEntityData(request, response);
-               
                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();
                }
        }