2 * ============LICENSE_START=======================================================
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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
21 package org.onap.policy.pap.test;
23 import static org.junit.Assert.assertTrue;
24 import static org.junit.Assert.fail;
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;
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;
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;
54 import com.mockrunner.mock.web.MockServletInputStream;
57 public class XACMLPAPTest {
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;
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));
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);
85 public void testFirwallCreatePolicy() throws IOException, ServletException, SQLException {
86 httpServletRequest = Mockito.mock(HttpServletRequest.class);
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);
100 pap.service(httpServletRequest, httpServletResponse);
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");
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.*");
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");
126 sessionBuilder.addProperties(properties);
127 sessionFactory = sessionBuilder.buildSessionFactory();
129 // Set up dao with SessionFactory
130 CommonClassDaoImpl.setSessionfactory(sessionFactory);
131 PolicyCreation.setCommonClassDao(new CommonClassDaoImpl());
135 public void testDummy() throws ServletException, IOException {
137 Mockito.when(httpServletRequest.getMethod()).thenReturn("POST");
138 mockOutput = Mockito.mock(ServletOutputStream.class);
141 Mockito.when(httpServletResponse.getOutputStream()).thenReturn(mockOutput);
142 } catch (IOException e) {
147 pap.service(httpServletRequest, httpServletResponse);
149 } catch (Exception e) {
155 public void destroy(){
156 if(sessionFactory!=null){
157 sessionFactory.close();