2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2019-2021 AT&T Intellectual Property. All rights reserved.
6 * Modifications Copyright (C) 2019-2021 Nordix Foundation.
7 * Modifications Copyright (C) 2020 Bell Canada. All rights reserved.
8 * ================================================================================
9 * Licensed under the Apache License, Version 2.0 (the "License");
10 * you may not use this file except in compliance with the License.
11 * You may obtain a copy of the License at
13 * http://www.apache.org/licenses/LICENSE-2.0
15 * Unless required by applicable law or agreed to in writing, software
16 * distributed under the License is distributed on an "AS IS" BASIS,
17 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
18 * See the License for the specific language governing permissions and
19 * limitations under the License.
21 * SPDX-License-Identifier: Apache-2.0
22 * ============LICENSE_END=========================================================
25 package org.onap.policy.api.main.rest.provider;
27 import static org.assertj.core.api.Assertions.assertThatCode;
28 import static org.assertj.core.api.Assertions.assertThatThrownBy;
29 import static org.junit.Assert.assertFalse;
30 import static org.junit.Assert.assertNotNull;
32 import java.util.Collections;
33 import org.junit.AfterClass;
34 import org.junit.BeforeClass;
35 import org.junit.Test;
36 import org.onap.policy.api.main.parameters.ApiParameterGroup;
37 import org.onap.policy.common.parameters.ParameterService;
38 import org.onap.policy.common.utils.coder.CoderException;
39 import org.onap.policy.common.utils.coder.StandardYamlCoder;
40 import org.onap.policy.common.utils.resources.ResourceUtils;
41 import org.onap.policy.models.base.PfModelException;
42 import org.onap.policy.models.provider.PolicyModelsProviderParameters;
43 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicyType;
44 import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;
47 * This class performs unit test of {@link PolicyTypeProvider}.
49 * @author Chenfei Gao (cgao@research.att.com)
51 public class TestPolicyTypeProvider {
53 private static PolicyTypeProvider policyTypeProvider;
54 private static PolicyProvider policyProvider;
55 private static PolicyModelsProviderParameters providerParams;
56 private static ApiParameterGroup apiParamGroup;
57 private static StandardYamlCoder standardYamlCoder;
59 private static final String POLICY_TYPE_VERSION = "1.0.0";
61 private static final String POLICY_RESOURCE_MONITORING = "policies/vCPE.policy.monitoring.input.tosca.yaml";
62 private static final String POLICY_TYPE_RESOURCE_MONITORING = "policytypes/onap.policies.monitoring.tcagen2.yaml";
63 private static final String POLICY_TYPE_RESOURCE_WITH_NO_VERSION =
64 "policytypes/onap.policies.optimization.Resource.no.version.yaml";
65 private static final String POLICY_TYPE_NAME_MONITORING = "onap.policies.monitoring.tcagen2";
67 private static final String POLICY_TYPE_RESOURCE_OPERATIONAL_COMMON =
68 "policytypes/onap.policies.controlloop.operational.Common.yaml";
69 private static final String POLICY_TYPE_RESOURCE_OPERATIONAL_DROOLS =
70 "policytypes/onap.policies.controlloop.operational.common.Drools.yaml";
71 private static final String POLICY_TYPE_RESOURCE_OPERATIONAL_APEX =
72 "policytypes/onap.policies.controlloop.operational.common.Apex.yaml";
73 private static final String POLICY_TYPE_OPERATIONAL_COMMON = "onap.policies.controlloop.operational.Common";
74 private static final String POLICY_TYPE_OPERATIONAL_APEX = "onap.policies.controlloop.operational.common.Apex";
75 private static final String POLICY_TYPE_OPERATIONAL_DROOLS = "onap.policies.controlloop.operational.common.Drools";
78 * Initializes parameters.
80 * @throws PfModelException the PfModel parsing exception
83 public static void setupParameters() throws PfModelException {
85 standardYamlCoder = new StandardYamlCoder();
86 providerParams = new PolicyModelsProviderParameters();
87 providerParams.setDatabaseDriver("org.h2.Driver");
88 providerParams.setDatabaseUrl("jdbc:h2:mem:testdb");
89 providerParams.setDatabaseUser("policy");
90 providerParams.setDatabasePassword("P01icY");
91 providerParams.setPersistenceUnit("ToscaConceptTest");
92 apiParamGroup = new ApiParameterGroup("ApiGroup", null, providerParams, Collections.emptyList(),
93 Collections.emptyList());
94 ParameterService.register(apiParamGroup, true);
95 policyTypeProvider = new PolicyTypeProvider();
96 policyProvider = new PolicyProvider();
100 * Closes up DB connections and deregisters API parameter group.
102 * @throws PfModelException the PfModel parsing exception
105 public static void tearDown() throws PfModelException {
107 policyTypeProvider.close();
108 policyProvider.close();
109 ParameterService.deregister(apiParamGroup);
113 public void testFetchPolicyTypes() throws Exception {
115 ToscaServiceTemplate serviceTemplate = policyTypeProvider.fetchPolicyTypes(null, null);
116 assertFalse(serviceTemplate.getPolicyTypes().isEmpty());
118 assertThatThrownBy(() -> {
119 policyTypeProvider.fetchPolicyTypes("dummy", null);
120 }).hasMessage("policy types for filter ToscaEntityFilter(name=dummy, version=null) do not exist");
122 assertThatThrownBy(() -> {
123 policyTypeProvider.fetchPolicyTypes("dummy", "dummy");
124 }).hasMessage("policy types for filter ToscaEntityFilter(name=dummy, version=dummy) do not exist");
128 public void testFetchLatestPolicyTypes() {
130 assertThatThrownBy(() -> {
131 policyTypeProvider.fetchLatestPolicyTypes("dummy");
132 }).hasMessage("policy types for filter ToscaEntityFilter(name=dummy, version=LATEST) do not exist");
136 public void testCreatePolicyType() throws Exception {
138 ToscaServiceTemplate policyTypeServiceTemplate = standardYamlCoder
139 .decode(ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE_MONITORING), ToscaServiceTemplate.class);
140 ToscaServiceTemplate serviceTemplate = policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
141 assertFalse(serviceTemplate.getPolicyTypes().isEmpty());
143 assertThatCode(() -> {
144 policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
145 }).doesNotThrowAnyException();
147 ToscaPolicyType policyType = policyTypeServiceTemplate.getPolicyTypes().get("onap.policies.monitoring.tcagen2");
148 policyType.setDescription("Some other description");
150 assertThatThrownBy(() -> {
151 policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
152 }).hasMessageContaining("item \"entity\" value \"onap.policies.monitoring.tcagen2:1.0.0\" INVALID, "
153 + "does not equal existing entity");
155 assertThatThrownBy(() -> {
156 ToscaServiceTemplate badPolicyType =
157 standardYamlCoder.decode(ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE_WITH_NO_VERSION),
158 ToscaServiceTemplate.class);
159 policyTypeProvider.createPolicyType(badPolicyType);
160 }).hasMessageContaining("item \"version\" value \"0.0.0\" INVALID, is null");
162 policyTypeProvider.deletePolicyType(POLICY_TYPE_NAME_MONITORING, POLICY_TYPE_VERSION);
166 public void testCreateOperationalPolicyTypes() throws CoderException, PfModelException {
167 ToscaServiceTemplate policyTypeServiceTemplate = standardYamlCoder.decode(
168 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE_OPERATIONAL_COMMON), ToscaServiceTemplate.class);
169 ToscaServiceTemplate serviceTemplate = policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
171 assertNotNull(serviceTemplate.getPolicyTypes().get(POLICY_TYPE_OPERATIONAL_COMMON));
173 policyTypeServiceTemplate = standardYamlCoder.decode(
174 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE_OPERATIONAL_DROOLS), ToscaServiceTemplate.class);
175 serviceTemplate = policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
176 assertNotNull(serviceTemplate.getPolicyTypes().get(POLICY_TYPE_OPERATIONAL_DROOLS));
178 policyTypeProvider.deletePolicyType(POLICY_TYPE_OPERATIONAL_DROOLS, POLICY_TYPE_VERSION);
179 policyTypeProvider.deletePolicyType(POLICY_TYPE_OPERATIONAL_COMMON, POLICY_TYPE_VERSION);
183 public void testCreateApexOperationalPolicyTypes() throws CoderException, PfModelException {
184 ToscaServiceTemplate policyTypeServiceTemplate = standardYamlCoder.decode(
185 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE_OPERATIONAL_COMMON), ToscaServiceTemplate.class);
186 ToscaServiceTemplate serviceTemplate = policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
187 policyTypeServiceTemplate = standardYamlCoder.decode(
188 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE_OPERATIONAL_APEX), ToscaServiceTemplate.class);
189 serviceTemplate = policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
190 assertNotNull(serviceTemplate.getPolicyTypes().get(POLICY_TYPE_OPERATIONAL_APEX));
191 policyTypeProvider.deletePolicyType(POLICY_TYPE_OPERATIONAL_APEX, POLICY_TYPE_VERSION);
195 public void testDeletePolicyType() throws Exception {
197 ToscaServiceTemplate policyTypeServiceTemplate = standardYamlCoder
198 .decode(ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE_MONITORING), ToscaServiceTemplate.class);
199 ToscaServiceTemplate serviceTemplate = policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
200 assertFalse(serviceTemplate.getPolicyTypes().isEmpty());
202 ToscaServiceTemplate policyServiceTemplate = standardYamlCoder
203 .decode(ResourceUtils.getResourceAsString(POLICY_RESOURCE_MONITORING), ToscaServiceTemplate.class);
204 policyProvider.createPolicy("onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0", policyServiceTemplate);
206 String exceptionMessage = "policy type onap.policies.monitoring.tcagen2:1.0.0 is in use, "
207 + "it is referenced in policy onap.restart.tca:1.0.0";
208 assertThatThrownBy(() -> {
209 policyTypeProvider.deletePolicyType("onap.policies.monitoring.tcagen2", "1.0.0");
210 }).hasMessage(exceptionMessage);
213 policyProvider.deletePolicy("onap.policies.monitoring.tcagen2", "1.0.0", "onap.restart.tca", "1.0.0");
214 assertFalse(serviceTemplate.getToscaTopologyTemplate().getPolicies().get(0).isEmpty());
216 serviceTemplate = policyTypeProvider.deletePolicyType("onap.policies.monitoring.tcagen2", "1.0.0");
217 assertFalse(serviceTemplate.getPolicyTypes().isEmpty());
219 assertThatThrownBy(() -> {
220 policyTypeProvider.deletePolicyType("onap.policies.monitoring.tcagen2", "1.0.0");
221 }).hasMessage("policy type onap.policies.monitoring.tcagen2:1.0.0 not found");