Add database initialization to XACML PDP test
[policy/engine.git] / ONAP-PAP-REST / src / test / java / org / onap / policy / pap / test / XACMLPAPTest.java
index ba5539f..5208ad7 100644 (file)
@@ -2,7 +2,7 @@
  * ============LICENSE_START=======================================================
  * ONAP-PAP-REST
  * ================================================================================
- * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * Copyright (C) 2017-2018 AT&T Intellectual Property. All rights reserved.
  * ================================================================================
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -22,6 +22,7 @@ package org.onap.policy.pap.test;
 
 import static org.junit.Assert.assertTrue;
 import static org.junit.Assert.fail;
+import static org.mockito.Mockito.when;
 
 import java.io.BufferedReader;
 import java.io.IOException;
@@ -34,25 +35,41 @@ import java.util.List;
 import java.util.Map;
 import java.util.Properties;
 
+import javax.persistence.EntityManager;
+import javax.persistence.EntityManagerFactory;
+import javax.persistence.EntityTransaction;
+import javax.persistence.Persistence;
 import javax.servlet.ServletConfig;
 import javax.servlet.ServletException;
 import javax.servlet.ServletOutputStream;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
 import org.apache.tomcat.dbcp.dbcp2.BasicDataSource;
 import org.hibernate.SessionFactory;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
 import org.mockito.Mockito;
+import org.onap.policy.common.ia.IntegrityAuditProperties;
 import org.onap.policy.pap.xacml.rest.XACMLPapServlet;
+import org.onap.policy.pap.xacml.rest.controller.ActionPolicyDictionaryController;
 import org.onap.policy.pap.xacml.rest.controller.ClosedLoopDictionaryController;
+import org.onap.policy.pap.xacml.rest.controller.DecisionPolicyDictionaryController;
+import org.onap.policy.pap.xacml.rest.controller.DescriptiveDictionaryController;
+import org.onap.policy.pap.xacml.rest.controller.DictionaryController;
 import org.onap.policy.pap.xacml.rest.controller.FirewallDictionaryController;
+import org.onap.policy.pap.xacml.rest.controller.MicroServiceDictionaryController;
+import org.onap.policy.pap.xacml.rest.controller.PolicyScopeDictionaryController;
+import org.onap.policy.pap.xacml.rest.controller.SafePolicyController;
 import org.onap.policy.pap.xacml.rest.daoimpl.CommonClassDaoImpl;
 import org.onap.policy.pap.xacml.rest.policycontroller.PolicyCreation;
+import org.onap.policy.pap.xacml.rest.util.DictionaryUtils;
 import org.onap.policy.rest.dao.CommonClassDao;
 import org.onap.policy.rest.jpa.BRMSParamTemplate;
+import org.onap.policy.rest.jpa.Category;
 import org.onap.policy.rest.jpa.PolicyEditorScopes;
 import org.onap.policy.rest.jpa.UserInfo;
 import org.onap.policy.utils.PolicyUtils;
@@ -63,8 +80,8 @@ import org.springframework.orm.hibernate4.LocalSessionFactoryBuilder;
 
 import com.mockrunner.mock.web.MockServletInputStream;
 
-
 public class XACMLPAPTest {
+       private static final Log logger = LogFactory.getLog(XACMLPAPTest.class);
 
     private static final String ENVIRONMENT_HEADER = "Environment";
     private List<String> headers = new ArrayList<>();
@@ -74,6 +91,49 @@ public class XACMLPAPTest {
     private ServletConfig servletConfig;
     private XACMLPapServlet pap;
     private SessionFactory sessionFactory;
+    private CommonClassDao commonClassDao;
+
+       private static final String DEFAULT_DB_DRIVER = "org.h2.Driver";
+       private static final String DEFAULT_DB_USER = "sa";
+       private static final String DEFAULT_DB_PWD = "";
+
+       @Before
+       public void setUpDB() throws Exception {
+               logger.info("setUpDB: Entering");
+
+               Properties properties = new Properties();
+               properties.put(IntegrityAuditProperties.DB_DRIVER, XACMLPAPTest.DEFAULT_DB_DRIVER);
+               properties.put(IntegrityAuditProperties.DB_URL, "jdbc:h2:file:./sql/xacmlTest");
+               properties.put(IntegrityAuditProperties.DB_USER, XACMLPAPTest.DEFAULT_DB_USER);
+               properties.put(IntegrityAuditProperties.DB_PWD, XACMLPAPTest.DEFAULT_DB_PWD);
+               properties.put(IntegrityAuditProperties.SITE_NAME, "SiteA");
+               properties.put(IntegrityAuditProperties.NODE_TYPE, "pap");
+
+               //Clean the iaTest DB table for IntegrityAuditEntity entries
+               cleanDb("testPapPU", properties);
+
+               logger.info("setUpDB: Exiting");
+       }
+
+       public void cleanDb(String persistenceUnit, Properties properties){
+               logger.debug("cleanDb: enter");
+
+               EntityManagerFactory emf = Persistence.createEntityManagerFactory(persistenceUnit, properties);
+
+               EntityManager em = emf.createEntityManager();
+               // Start a transaction
+               EntityTransaction et = em.getTransaction();
+
+               et.begin();
+
+               // Clean up the DB
+               em.createQuery("Delete from IntegrityAuditEntity").executeUpdate();
+
+               // commit transaction
+               et.commit();
+               em.close();
+               logger.debug("cleanDb: exit");
+       }
 
     @Before
     public void setUp() throws ServletException {
@@ -89,6 +149,10 @@ public class XACMLPAPTest {
                 .thenReturn("src/test/resources/xacml.pap.properties");
         pap = new XACMLPapServlet();
         pap.init(servletConfig);
+        commonClassDao = Mockito.mock(CommonClassDao.class);
+        new DictionaryUtils(commonClassDao);
+        DictionaryUtils.setDictionaryUtils(new DictionaryUtils());
+        Mockito.mock(DictionaryUtils.class);  
     }
     
     @Test
@@ -518,6 +582,170 @@ public class XACMLPAPTest {
         Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
     }
     
+    @Test
+    public void testCommonCreateDictionary() throws IOException, SQLException, ServletException {
+       new DictionaryController(commonClassDao);
+       new ActionPolicyDictionaryController(commonClassDao);
+       new SafePolicyController(commonClassDao);
+       new DescriptiveDictionaryController(commonClassDao);
+       List<Object> object = new ArrayList<>();
+        object.add(new Category());
+        when(commonClassDao.getDataById(Category.class, "shortName", "resource")).thenReturn(object);
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       String json = "{\"dictionaryFields\": {\"onapName\": \"testMMRestAPI1\",\"description\": \"testing update response message\"}}";
+       dictionaryTestSetup(false, "OnapName", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+       
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       json = "{\"dictionaryFields\": {\"xacmlId\": \"testMMRestAPI1\",\"datatypeBean\": {\"shortName\": \"string\"}, \"description\": \"testing update\",\"priority\": \"High\",\"userDataTypeValues\": [{\"attributeValues\": \"testAttr\"}, {\"attributeValues\": \"testAttr2\"}, {\"attributeValues\": \"testAttr3\"}]}}";
+       dictionaryTestSetup(false, "Attribute", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+       
+       
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       json = "{\"dictionaryFields\":{\"attributeName\":\"TestMMrestAPI1\",\"type\":\"REST\",\"url\":\"testsomeurl.com\",\"method\":\"GET\",\"description\":\"test create\",\"body\":\"Testing Create\",\"headers\":[{\"option\":\"test1\",\"number\":\"test\"},{\"option\":\"test2\",\"number\":\"test\"}]}}";
+       dictionaryTestSetup(false, "Action", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+       
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       json = "{\"dictionaryFields\":{\"scopeName\":\"testMMRestAPI1\",\"description\":\"test\",\"attributes\":[{\"option\":\"test1\",\"number\":\"test\"},{\"option\":\"test2\",\"number\":\"test\"}]}}";
+       dictionaryTestSetup(false, "DescriptiveScope", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+       
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       json = "{\"dictionaryFields\":{\"riskName\":\"testMMrestAPI1\",\"description\":\"test\"}}";
+       dictionaryTestSetup(false, "RiskType", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+       
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       json = "{\"dictionaryFields\":{\"name\":\"testMMrestAPI1\",\"message\":\"test\",\"riskType\":\"testMMrestAPI1\"}}";
+       dictionaryTestSetup(false, "SafePolicyWarning", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+    }
+    
+    @Test
+    public void testDecisionCreateDictionary() throws IOException, SQLException, ServletException {
+       new DecisionPolicyDictionaryController(commonClassDao);
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       String json = "{\"dictionaryFields\":{\"xacmlId\":\"testMMRestAPI1\",\"datatypeBean\":{\"shortName\":\"string\"},\"description\":\"test\",\"priority\":\"High\"}}";
+       dictionaryTestSetup(false, "Settings", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+       
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       json = "{\"dictionaryFields\":{\"bbid\":\"BB1\",\"workstep\":\"1\",\"treatments\":\"Manual Handling,Abort,Retry\"}}";
+       dictionaryTestSetup(false, "RainyDayTreatments", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+    }
+    
+    @Test
+    public void testMSCreateDictionary() throws IOException, SQLException, ServletException {
+       new MicroServiceDictionaryController(commonClassDao);
+       new PolicyScopeDictionaryController(commonClassDao);
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       String json = "{\"dictionaryFields\":{\"name\":\"testMMrestAPI1\",\"descriptionValue\":\"test\"}}";
+       dictionaryTestSetup(false, "MicroServiceLocation", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+       
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       json = "{\"dictionaryFields\":{\"name\":\"testMMrestAPI1\",\"descriptionValue\":\"test\"}}";
+       dictionaryTestSetup(false, "MicroServiceConfigName", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+       
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       json = "{\"dictionaryFields\":{\"name\":\"testMMrestAPI1\",\"description\":\"test\"}}";
+       dictionaryTestSetup(false, "DCAEUUID", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+       
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       json = "{\"dictionaryFields\":{\"name\":\"testApiANY\",\"descriptionValue\":\"default test\"}}";
+       dictionaryTestSetup(false, "PolicyScopeService", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+       
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       json = "{\"dictionaryFields\":{\"name\":\"testApiANY\",\"descriptionValue\":\"default test\"}}";
+       dictionaryTestSetup(false, "PolicyScopeResource", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+       
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       json = "{\"dictionaryFields\":{\"name\":\"testApiANY\",\"descriptionValue\":\"default test\"}}";
+       dictionaryTestSetup(false, "PolicyScopeType", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+       
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       json = "{\"dictionaryFields\":{\"name\":\"testApiANY\",\"descriptionValue\":\"default test\"}}";
+       dictionaryTestSetup(false, "PolicyScopeClosedLoop", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+       
+       httpServletRequest = Mockito.mock(HttpServletRequest.class);
+       httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
+       json = "{\"dictionaryFields\":{\"groupName\":\"testMMrestAPI1\",\"description\":\"testing\"},\"groupPolicyScopeListData1\":{\"resource\":\"ANY\",\"type\":\"ANY\",\"service\":\"ANY\",\"closedloop\":\"ANY\"}}";
+       dictionaryTestSetup(false, "GroupPolicyScopeList", json);
+       // send Request to PAP
+       pap.service(httpServletRequest, httpServletResponse);
+       // Verify 
+       Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
+    }
+    
     private void dictionaryTestSetup(Boolean updateFlag, String dictionaryType, String json) throws IOException, SQLException {
         Mockito.when(httpServletRequest.getHeader(ENVIRONMENT_HEADER)).thenReturn("DEVL");
         Mockito.when(httpServletRequest.getHeader("ClientScope")).thenReturn("dictionaryItem");
@@ -536,7 +764,7 @@ public class XACMLPAPTest {
         setDBDao();
     }
 
-    private void setDBDao() throws SQLException {
+    public void setDBDao() throws SQLException {
         BasicDataSource dataSource = new BasicDataSource();
         dataSource.setDriverClassName("org.h2.Driver");
         // In-memory DB for testing
@@ -560,6 +788,27 @@ public class XACMLPAPTest {
         PolicyCreation.setCommonClassDao(new CommonClassDaoImpl());
     }
 
+    @Test
+    public void getDictionary() throws ServletException, IOException{
+       String[] dictionarys = new String[]{"Attribute", "OnapName", "Action", "BRMSParamTemplate","VSCLAction"
+                       ,"VNFType","PEPOptions","Varbind","Service","Site", "Settings", "RainyDayTreatments",
+                       "DescriptiveScope", "ActionList", "ProtocolList", "Zone", "SecurityZone",
+                       "PrefixList", "AddressGroup", "ServiceGroup", "ServiceList", "TermList",
+                       "MicroServiceLocation", "MicroServiceConfigName", "DCAEUUID", "MicroServiceModels",
+                       "PolicyScopeService", "PolicyScopeResource", "PolicyScopeType", "PolicyScopeClosedLoop",
+                       "GroupPolicyScopeList", "RiskType", "SafePolicyWarning", "MicroServiceDictionary"};
+       for(String dictionary : dictionarys){
+               httpServletRequest = Mockito.mock(HttpServletRequest.class);
+               httpServletResponse = new MockHttpServletResponse();
+               Mockito.when(httpServletRequest.getHeader(ENVIRONMENT_HEADER)).thenReturn("DEVL");
+               Mockito.when(httpServletRequest.getMethod()).thenReturn("GET");
+               Mockito.when(httpServletRequest.getParameter("apiflag")).thenReturn("api");
+               Mockito.when(httpServletRequest.getParameter("dictionaryType")).thenReturn(dictionary);
+               pap.service(httpServletRequest, httpServletResponse);
+               assertTrue(HttpServletResponse.SC_OK == httpServletResponse.getStatus());
+       }
+    }
+    
     @Test
     public void testDummy() throws ServletException, IOException {