ONAP-PAP-REST/src/test/java/org/onap/policy/pap/test/XACMLPAPTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP-PAP-REST
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.policy.pap.test;
  22
  23 import static org.junit.Assert.assertTrue;
  24 import static org.junit.Assert.fail;
  25
  26 import java.io.IOException;
  27 import java.sql.SQLException;
  28 import java.util.ArrayList;
  29 import java.util.Collections;
  30 import java.util.List;
  31 import java.util.Properties;
  32
  33 import javax.servlet.ServletConfig;
  34 import javax.servlet.ServletException;
  35 import javax.servlet.ServletOutputStream;
  36 import javax.servlet.http.HttpServletRequest;
  37 import javax.servlet.http.HttpServletResponse;
  38
  39 import org.apache.tomcat.dbcp.dbcp2.BasicDataSource;
  40 import org.hibernate.SessionFactory;
  41 import org.junit.After;
  42 import org.junit.Before;
  43 import org.junit.Test;
  44 import org.mockito.Mockito;
  45 import org.onap.policy.pap.xacml.rest.XACMLPapServlet;
  46 import org.onap.policy.pap.xacml.rest.daoimpl.CommonClassDaoImpl;
  47 import org.onap.policy.pap.xacml.rest.policycontroller.PolicyCreation;
  48 import org.onap.policy.utils.PolicyUtils;
  49 import org.onap.policy.xacml.std.pap.StdPAPPolicy;
  50 import org.springframework.mock.web.MockHttpServletResponse;
  51 import org.springframework.mock.web.MockServletConfig;
  52 import org.springframework.orm.hibernate4.LocalSessionFactoryBuilder;
  53
  54 import com.mockrunner.mock.web.MockServletInputStream;
  55
  56
  57 public class XACMLPAPTest {
  58
  59     private static final String ENVIRONMENT_HEADER = "Environment";
  60     private List<String> headers = new ArrayList<>();
  61     private HttpServletRequest httpServletRequest;
  62     private HttpServletResponse httpServletResponse;
  63     private ServletOutputStream mockOutput;
  64     private ServletConfig servletConfig;
  65     private XACMLPapServlet pap;
  66     private SessionFactory sessionFactory;
  67
  68     @Before
  69     public void setUp() throws ServletException {
  70         httpServletRequest = Mockito.mock(HttpServletRequest.class);
  71         httpServletResponse = Mockito.mock(MockHttpServletResponse.class);
  72         Mockito.when(httpServletRequest.getHeaderNames()).thenReturn(Collections.enumeration(headers));
  73         Mockito.when(httpServletRequest.getAttributeNames()).thenReturn(Collections.enumeration(headers));
  74
  75         servletConfig = Mockito.mock(MockServletConfig.class);
  76         System.setProperty("com.sun.management.jmxremote.port", "9993");
  77         Mockito.when(servletConfig.getInitParameterNames()).thenReturn(Collections.enumeration(headers));
  78         Mockito.when(servletConfig.getInitParameter("XACML_PROPERTIES_NAME"))
  79                 .thenReturn("src/test/resources/xacml.pap.properties");
  80         pap = new XACMLPapServlet();
  81         pap.init(servletConfig);
  82     }
  83
  84     @Test
  85     public void testFirwallCreatePolicy() throws IOException, ServletException, SQLException {
  86         httpServletRequest = Mockito.mock(HttpServletRequest.class);
  87         String json = "";
  88         Mockito.when(httpServletRequest.getHeader(ENVIRONMENT_HEADER)).thenReturn("DEVL");
  89         Mockito.when(httpServletRequest.getMethod()).thenReturn("PUT");
  90         Mockito.when(httpServletRequest.getParameter("apiflag")).thenReturn("api");
  91         Mockito.when(httpServletRequest.getParameter("operation")).thenReturn("create");
  92         Mockito.when(httpServletRequest.getParameter("policyType")).thenReturn("Config");
  93         StdPAPPolicy newPAPPolicy = new StdPAPPolicy("Firewall Config", "test", "testDescription", "Test", false, "test", json, 0,
  94                 "5","default", "false", "");
  95         MockServletInputStream mockInput = new MockServletInputStream(PolicyUtils.objectToJsonString(newPAPPolicy).getBytes());
  96         Mockito.when(httpServletRequest.getInputStream()).thenReturn(mockInput);
  97
  98         // set DBDao
  99         setDBDao();
 100         pap.service(httpServletRequest, httpServletResponse);
 101
 102         Mockito.verify(httpServletResponse).setStatus(HttpServletResponse.SC_OK);
 103         Mockito.verify(httpServletResponse).addHeader("successMapKey", "success");
 104         Mockito.verify(httpServletResponse).addHeader("policyName", "test.Config_FW_test.1.xml");
 105
 106     }
 107
 108
 109
 110     private void setDBDao() throws SQLException {
 111         BasicDataSource dataSource = new BasicDataSource();
 112         dataSource.setDriverClassName("org.h2.Driver");
 113         // In-memory DB for testing
 114         dataSource.setUrl("jdbc:h2:mem:test");
 115         dataSource.setUsername("sa");
 116         dataSource.setPassword("");
 117         LocalSessionFactoryBuilder sessionBuilder = new LocalSessionFactoryBuilder(dataSource);
 118         sessionBuilder.scanPackages("org.onap.*", "com.*");
 119
 120         Properties properties = new Properties();
 121         properties.put("hibernate.show_sql", "false");
 122         properties.put("hibernate.dialect", "org.hibernate.dialect.H2Dialect");
 123         properties.put("hibernate.hbm2ddl.auto", "drop");
 124         properties.put("hibernate.hbm2ddl.auto", "create");
 125
 126         sessionBuilder.addProperties(properties);
 127         sessionFactory = sessionBuilder.buildSessionFactory();
 128
 129         // Set up dao with SessionFactory
 130         CommonClassDaoImpl.setSessionfactory(sessionFactory);
 131         PolicyCreation.setCommonClassDao(new CommonClassDaoImpl());
 132     }
 133
 134     @Test
 135     public void testDummy() throws ServletException, IOException {
 136
 137         Mockito.when(httpServletRequest.getMethod()).thenReturn("POST");
 138         mockOutput = Mockito.mock(ServletOutputStream.class);
 139
 140         try {
 141             Mockito.when(httpServletResponse.getOutputStream()).thenReturn(mockOutput);
 142         } catch (IOException e) {
 143             fail();
 144         }
 145
 146         try {
 147             pap.service(httpServletRequest, httpServletResponse);
 148             assertTrue(true);
 149         } catch (Exception e) {
 150             fail();
 151         }
 152     }
 153
 154     @After
 155     public void destroy(){
 156         if(sessionFactory!=null){
 157             sessionFactory.close();
 158         }
 159         pap.destroy();
 160     }
 161 }