2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2018-2019 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.xacml.rest.elk;
23 import static org.assertj.core.api.Assertions.assertThatCode;
24 import static org.assertj.core.api.Assertions.assertThatThrownBy;
25 import static org.junit.Assert.assertEquals;
26 import static org.junit.Assert.assertFalse;
27 import static org.mockito.Mockito.when;
29 import java.io.BufferedReader;
30 import java.io.IOException;
31 import java.io.StringReader;
32 import java.util.ArrayList;
33 import java.util.HashMap;
34 import java.util.List;
36 import javax.servlet.http.HttpServletRequest;
37 import javax.servlet.http.HttpServletResponse;
38 import org.junit.Before;
39 import org.junit.Test;
40 import org.mockito.Mockito;
41 import org.onap.policy.pap.xacml.rest.elk.client.ElkConnector.PolicyIndexType;
42 import org.onap.policy.pap.xacml.rest.elk.client.PolicyElasticSearchController;
43 import org.onap.policy.rest.adapter.PolicyRestAdapter;
44 import org.onap.policy.rest.dao.CommonClassDao;
45 import org.springframework.mock.web.MockHttpServletResponse;
47 public class PolicyElasticSearchControllerTest {
49 private PolicyElasticSearchController controller;
50 private HttpServletRequest request = null;
51 private HttpServletResponse response = null;
55 controller = new PolicyElasticSearchController();
56 request = Mockito.mock(HttpServletRequest.class);
57 response = new MockHttpServletResponse();
61 public void testSearchDictionary() throws IOException {
62 List<String> jsonString = new ArrayList<>();
63 jsonString.add("{\"type\":\"attribute\",\"data\":{\"xacmlId\":\"Test\"}}");
64 jsonString.add("{\"type\":\"onapName\",\"data\":{\"onapName\":\"Test\"}}");
65 jsonString.add("{\"type\":\"actionPolicy\",\"data\":{\"attributeName\":\"Test\"}}");
66 jsonString.add("{\"type\":\"brmsParam\",\"data\":{\"ruleName\":\"Test\"}}");
67 jsonString.add("{\"type\":\"pepOptions\",\"data\":{\"pepName\":\"Test\"}}");
68 jsonString.add("{\"type\":\"clSite\",\"data\":{\"siteName\":\"Test\"}}");
69 jsonString.add("{\"type\":\"clService\",\"data\":{\"serviceName\":\"Test\"}}");
70 jsonString.add("{\"type\":\"clVarbind\",\"data\":{\"varbindName\":\"Test\"}}");
71 jsonString.add("{\"type\":\"clVnf\",\"data\":{\"vnftype\":\"Test\"}}");
72 jsonString.add("{\"type\":\"clVSCL\",\"data\":{\"vsclaction\":\"Test\"}}");
73 jsonString.add("{\"type\":\"decision\",\"data\":{\"xacmlId\":\"Test\"}}");
74 jsonString.add("{\"type\":\"fwTerm\",\"data\":{\"termName\":\"Test\"}}");
75 jsonString.add("{\"type\":\"msDCAEUUID\",\"data\":{\"name\":\"Test\"}}");
76 jsonString.add("{\"type\":\"msLocation\",\"data\":{\"name\":\"Test\"}}");
77 jsonString.add("{\"type\":\"msModels\",\"data\":{\"modelName\":\"Test\"}}");
78 jsonString.add("{\"type\":\"psGroupPolicy\",\"data\":{\"name\":\"Test\"}}");
79 jsonString.add("{\"type\":\"safeRisk\",\"data\":{\"name\":\"Test\"}}");
80 jsonString.add("{\"type\":\"safePolicyWarning\",\"data\":{\"name\":\"Test\"}}");
81 for (int i = 0; i < jsonString.size(); i++) {
82 BufferedReader br = new BufferedReader(new StringReader(jsonString.get(i)));
83 when(request.getReader()).thenReturn(br);
84 assertThatCode(() -> controller.searchDictionary(request, response)).doesNotThrowAnyException();
89 public void testController() throws IOException {
90 CommonClassDao dao = Mockito.mock(CommonClassDao.class);
91 PolicyElasticSearchController controller = new PolicyElasticSearchController(dao);
92 assertEquals(PolicyIndexType.all, controller.toPolicyIndexType(null));
93 assertEquals(PolicyIndexType.config, controller.toPolicyIndexType("config"));
95 Map<String, String> searchKeys = new HashMap<String, String>();
96 searchKeys.put("key", "value");
97 assertThatThrownBy(() -> controller.search(PolicyIndexType.config, "text", searchKeys))
98 .isInstanceOf(Exception.class);
100 when(request.getParameter("policyName")).thenReturn("policyName");
101 when(request.getParameter("action")).thenReturn("search");
102 when(request.getReader())
103 .thenReturn(new BufferedReader(new StringReader("{\"searchdata\": { \"query\": \"value space\", "
104 + "\"policyType\": \"all\", " + "\"closedLooppolicyType\": \"type\", " + "\"onapName\": \"pef\", "
105 + "\"vnfType\": \"vnf\", " + "\"policyStatus\": \"active\", " + "\"vproAction\": \"reboot\", "
106 + "\"serviceType\": \"type\", " + "\"bindTextSearch\": \"pef\", " + "\"d2Service\": \"vDNS\"} }")));
107 controller.searchPolicy(request, response);
108 assertEquals(HttpServletResponse.SC_OK, response.getStatus());
110 PolicyRestAdapter policyData = new PolicyRestAdapter();
111 assertFalse(controller.deleteElk(policyData));