2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2019-2020 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.policycontroller;
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.mockito.ArgumentMatchers.anyString;
27 import static org.mockito.ArgumentMatchers.eq;
29 import com.mockrunner.mock.web.MockHttpServletRequest;
30 import java.util.ArrayList;
31 import java.util.LinkedHashMap;
32 import java.util.LinkedList;
33 import java.util.List;
34 import javax.servlet.http.HttpServletRequest;
35 import javax.servlet.http.HttpServletResponse;
37 import org.hibernate.Criteria;
38 import org.hibernate.Session;
39 import org.hibernate.SessionFactory;
40 import org.junit.Test;
41 import org.mockito.Mockito;
42 import org.onap.policy.pap.xacml.rest.components.PolicyDbDao;
43 import org.onap.policy.pap.xacml.rest.components.PolicyDbDaoTransaction;
44 import org.onap.policy.rest.adapter.PolicyRestAdapter;
45 import org.onap.policy.rest.dao.CommonClassDao;
46 import org.onap.policy.rest.jpa.PolicyDbDaoEntity;
47 import org.onap.policy.rest.jpa.PolicyVersion;
48 import org.powermock.reflect.Whitebox;
49 import org.springframework.http.HttpStatus;
50 import org.springframework.http.converter.HttpMessageNotReadableException;
51 import org.springframework.mock.web.MockHttpServletResponse;
53 public class PolicyCreationTest {
55 public void testCreation() {
56 CommonClassDao dao = Mockito.mock(CommonClassDao.class);
57 PolicyVersion version = new PolicyVersion();
58 version.setPolicyName("policyname");
59 Mockito.when(dao.getEntityItem(eq(PolicyVersion.class), eq("policyName"), anyString())).thenReturn(version);
60 PolicyCreation.setCommonClassDao(dao);
61 assertEquals(dao, PolicyCreation.getCommonClassDao());
62 PolicyCreation creation = new PolicyCreation(dao);
63 assertEquals(dao, PolicyCreation.getCommonClassDao());
65 HttpServletRequest req = new MockHttpServletRequest();
66 Exception cause = new Exception("oops");
67 HttpMessageNotReadableException exception = new HttpMessageNotReadableException("oops", cause);
68 assertEquals(HttpStatus.BAD_REQUEST,
69 creation.messageNotReadableExceptionHandler(req, exception).getStatusCode());
71 HttpServletResponse response = new MockHttpServletResponse();
72 PolicyRestAdapter policyData = new PolicyRestAdapter();
73 policyData.setPolicyType("Config");
74 policyData.setConfigPolicyType("ClosedLoop_Fault");
75 policyData.setDomainDir("domain");
76 policyData.setPolicyName("policyname");
77 assertThatCode(() -> creation.savePolicy(policyData, response)).doesNotThrowAnyException();
79 version.setHigherVersion(1);
80 assertThatCode(() -> creation.savePolicy(policyData, response)).doesNotThrowAnyException();
82 policyData.setEditPolicy(true);
83 assertThatCode(() -> creation.savePolicy(policyData, response)).doesNotThrowAnyException();
85 policyData.setEditPolicy(false);
86 version.setHigherVersion(0);
87 assertThatCode(() -> creation.savePolicy(policyData, response)).doesNotThrowAnyException();
89 policyData.setEditPolicy(true);
90 assertThatCode(() -> creation.savePolicy(policyData, response)).doesNotThrowAnyException();
92 version.setHigherVersion(1);
93 policyData.setConfigPolicyType("Firewall Config");
94 assertThatThrownBy(() -> creation.savePolicy(policyData, response))
95 .isInstanceOf(IllegalArgumentException.class);
97 policyData.setConfigPolicyType("BRMS_Raw");
98 assertThatCode(() -> creation.savePolicy(policyData, response)).doesNotThrowAnyException();
99 policyData.setConfigPolicyType("BRMS_Param");
100 assertThatThrownBy(() -> creation.savePolicy(policyData, response))
101 .isInstanceOf(IllegalArgumentException.class);
102 policyData.setConfigPolicyType("Base");
103 Mockito.when(policyData.getRuleData()).thenReturn(new LinkedHashMap<>());
105 SessionFactory mockSessionFactory = Mockito.mock(SessionFactory.class);
106 Session mockSession = Mockito.mock(Session.class);
107 Criteria mockCriteria = Mockito.mock(Criteria.class);
108 List<?> policyDbDaoEntityList = new LinkedList<>();
110 Mockito.when(mockSessionFactory.openSession()).thenReturn(mockSession);
111 Mockito.when(mockSession.createCriteria(PolicyDbDaoEntity.class)).thenReturn(mockCriteria);
112 Mockito.when(mockCriteria.list()).thenReturn(policyDbDaoEntityList);
113 Whitebox.setInternalState(PolicyDbDao.class, "sessionfactory", mockSessionFactory);
115 PolicyDbDao mockPolicyDbDao = Mockito.mock(PolicyDbDao.class);
116 PolicyDbDaoTransaction mockTransaction = Mockito.mock(PolicyDbDaoTransaction.class);
117 Mockito.when(mockPolicyDbDao.getNewTransaction()).thenReturn(mockTransaction);
119 assertThatCode(() -> creation.savePolicy(policyData, response)).doesNotThrowAnyException();
120 policyData.setConfigPolicyType("ClosedLoop_PM");
121 assertThatThrownBy(() -> creation.savePolicy(policyData, response))
122 .isInstanceOf(IllegalArgumentException.class);
123 policyData.setConfigPolicyType("Micro Service");
124 assertThatThrownBy(() -> creation.savePolicy(policyData, response))
125 .isInstanceOf(IllegalArgumentException.class);
126 policyData.setConfigPolicyType("Optimization");
127 assertThatThrownBy(() -> creation.savePolicy(policyData, response))
128 .isInstanceOf(IllegalArgumentException.class);
130 policyData.setPolicyType("Action");
131 List<Object> choices = new ArrayList<Object>();
132 policyData.setRuleAlgorithmschoices(choices);
133 assertThatCode(() -> creation.savePolicy(policyData, response)).doesNotThrowAnyException();
135 policyData.setPolicyType("Decision");
136 List<Object> settings = new ArrayList<Object>();
137 policyData.setSettings(settings);
138 assertThatThrownBy(() -> creation.savePolicy(policyData, response))
139 .isInstanceOf(IllegalArgumentException.class);