2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2019-2020 AT&T Intellectual Property. All rights reserved.
6 * Modifications Copyright (C) 2019 Nordix Foundation.
7 * ================================================================================
8 * Licensed under the Apache License, Version 2.0 (the "License");
9 * you may not use this file except in compliance with the License.
10 * You may obtain a copy of the License at
12 * http://www.apache.org/licenses/LICENSE-2.0
14 * Unless required by applicable law or agreed to in writing, software
15 * distributed under the License is distributed on an "AS IS" BASIS,
16 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17 * See the License for the specific language governing permissions and
18 * limitations under the License.
20 * SPDX-License-Identifier: Apache-2.0
21 * ============LICENSE_END=========================================================
24 package org.onap.policy.api.main.rest.provider;
26 import static org.assertj.core.api.Assertions.assertThatThrownBy;
27 import static org.junit.Assert.assertFalse;
28 import static org.junit.Assert.assertNotNull;
30 import java.util.Base64;
31 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.ToscaServiceTemplate;
46 * This class performs unit test of {@link PolicyTypeProvider}.
48 * @author Chenfei Gao (cgao@research.att.com)
50 public class TestPolicyTypeProvider {
52 private static PolicyTypeProvider policyTypeProvider;
53 private static PolicyProvider policyProvider;
54 private static PolicyModelsProviderParameters providerParams;
55 private static ApiParameterGroup apiParamGroup;
56 private static StandardYamlCoder standardYamlCoder;
58 private static final String POLICY_TYPE_VERSION = "1.0.0";
60 private static final String POLICY_RESOURCE_MONITORING = "policies/vCPE.policy.monitoring.input.tosca.yaml";
61 private static final String POLICY_TYPE_RESOURCE_MONITORING =
62 "policytypes/onap.policies.monitoring.cdap.tca.hi.lo.app.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.cdap.tca.hi.lo.app";
67 public static final String POLICY_TYPE_RESOURCE_OPERATIONAL =
68 "policytypes/onap.policies.controlloop.operational.Common.yaml";
69 public static final String POLICY_TYPE_OPERATIONAL_COMMON = "onap.policies.controlloop.operational.Common";
70 public static final String POLICY_TYPE_OPERATIONAL_APEX = "onap.policies.controlloop.operational.common.Apex";
71 public static final String POLICY_TYPE_OPERATIONAL_DROOLS = "onap.policies.controlloop.operational.common.Drools";
74 * Initializes parameters.
76 * @throws PfModelException the PfModel parsing exception
79 public static void setupParameters() throws PfModelException {
81 standardYamlCoder = new StandardYamlCoder();
82 providerParams = new PolicyModelsProviderParameters();
83 providerParams.setDatabaseDriver("org.h2.Driver");
84 providerParams.setDatabaseUrl("jdbc:h2:mem:testdb");
85 providerParams.setDatabaseUser("policy");
86 providerParams.setDatabasePassword(Base64.getEncoder().encodeToString("P01icY".getBytes()));
87 providerParams.setPersistenceUnit("ToscaConceptTest");
88 apiParamGroup = new ApiParameterGroup("ApiGroup", null, providerParams, Collections.emptyList());
89 ParameterService.register(apiParamGroup, true);
90 policyTypeProvider = new PolicyTypeProvider();
91 policyProvider = new PolicyProvider();
95 * Closes up DB connections and deregisters API parameter group.
97 * @throws PfModelException the PfModel parsing exception
100 public static void tearDown() throws PfModelException {
102 policyTypeProvider.close();
103 policyProvider.close();
104 ParameterService.deregister(apiParamGroup);
108 public void testFetchPolicyTypes() throws Exception {
110 ToscaServiceTemplate serviceTemplate = policyTypeProvider.fetchPolicyTypes(null, null);
111 assertFalse(serviceTemplate.getPolicyTypes().isEmpty());
113 assertThatThrownBy(() -> {
114 policyTypeProvider.fetchPolicyTypes("dummy", null);
115 }).hasMessage("policy type with ID dummy:null does not exist");
117 assertThatThrownBy(() -> {
118 policyTypeProvider.fetchPolicyTypes("dummy", "dummy");
119 }).hasMessage("policy type with ID dummy:dummy does not exist");
123 public void testFetchLatestPolicyTypes() {
125 assertThatThrownBy(() -> {
126 policyTypeProvider.fetchLatestPolicyTypes("dummy");
127 }).hasMessage("policy type with ID dummy:null does not exist");
131 public void testCreatePolicyType() throws Exception {
133 ToscaServiceTemplate policyTypeServiceTemplate = standardYamlCoder.decode(
134 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE_MONITORING), ToscaServiceTemplate.class);
135 ToscaServiceTemplate serviceTemplate = policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
136 assertFalse(serviceTemplate.getPolicyTypes().isEmpty());
138 String errorMessage = "policy type onap.policies.monitoring.cdap.tca.hi.lo.app:1.0.0 already exists; "
139 + "its latest version is 1.0.0";
140 assertThatThrownBy(() -> {
141 ToscaServiceTemplate duplicatePolicyType = standardYamlCoder.decode(
142 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE_MONITORING), ToscaServiceTemplate.class);
143 policyTypeProvider.createPolicyType(duplicatePolicyType);
144 }).hasMessage(errorMessage);
146 assertThatThrownBy(() -> {
147 ToscaServiceTemplate badPolicyType = standardYamlCoder.decode(ResourceUtils.getResourceAsString(
148 POLICY_TYPE_RESOURCE_WITH_NO_VERSION), ToscaServiceTemplate.class);
149 policyTypeProvider.createPolicyType(badPolicyType);
150 }).hasMessage("mandatory 'version' field is missing in policy types: onap.policies.optimization.Resource");
152 policyTypeProvider.deletePolicyType(POLICY_TYPE_NAME_MONITORING, POLICY_TYPE_VERSION);
156 public void testCreateOperationalPolicyTypes() throws CoderException, PfModelException {
157 ToscaServiceTemplate policyTypeServiceTemplate = standardYamlCoder.decode(
158 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE_OPERATIONAL), ToscaServiceTemplate.class);
159 ToscaServiceTemplate serviceTemplate = policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
161 assertNotNull(serviceTemplate.getPolicyTypes().get(POLICY_TYPE_OPERATIONAL_COMMON));
162 assertNotNull(serviceTemplate.getPolicyTypes().get(POLICY_TYPE_OPERATIONAL_APEX));
163 assertNotNull(serviceTemplate.getPolicyTypes().get(POLICY_TYPE_OPERATIONAL_DROOLS));
165 policyTypeProvider.deletePolicyType(POLICY_TYPE_OPERATIONAL_COMMON, POLICY_TYPE_VERSION);
166 policyTypeProvider.deletePolicyType(POLICY_TYPE_OPERATIONAL_APEX, POLICY_TYPE_VERSION);
167 policyTypeProvider.deletePolicyType(POLICY_TYPE_OPERATIONAL_DROOLS, POLICY_TYPE_VERSION);
171 public void testDeletePolicyType() throws Exception {
173 ToscaServiceTemplate policyTypeServiceTemplate = standardYamlCoder.decode(
174 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE_MONITORING), ToscaServiceTemplate.class);
175 ToscaServiceTemplate serviceTemplate = policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
176 assertFalse(serviceTemplate.getPolicyTypes().isEmpty());
178 ToscaServiceTemplate policyServiceTemplate = standardYamlCoder.decode(
179 ResourceUtils.getResourceAsString(POLICY_RESOURCE_MONITORING), ToscaServiceTemplate.class);
180 policyProvider.createPolicy("onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0", policyServiceTemplate);
182 String exceptionMessage = "policy type with ID onap.policies.monitoring.cdap.tca.hi.lo.app:1.0.0 "
183 + "cannot be deleted as it is parameterized by policies onap.restart.tca:1.0.0";
184 assertThatThrownBy(() -> {
185 policyTypeProvider.deletePolicyType("onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0");
186 }).hasMessage(exceptionMessage);
188 serviceTemplate = policyProvider
189 .deletePolicy("onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0", "onap.restart.tca", "1.0.0");
190 assertFalse(serviceTemplate.getToscaTopologyTemplate().getPolicies().get(0).isEmpty());
193 policyTypeProvider.deletePolicyType("onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0");
194 assertFalse(serviceTemplate.getPolicyTypes().isEmpty());
196 assertThatThrownBy(() -> {
197 policyTypeProvider.deletePolicyType("onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0");
198 }).hasMessage("policy type with ID onap.policies.monitoring.cdap.tca.hi.lo.app:1.0.0 does not exist");