2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 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.
19 * SPDX-License-Identifier: Apache-2.0
20 * ============LICENSE_END=========================================================
23 package org.onap.policy.api.main.rest.provider;
25 import static org.assertj.core.api.Assertions.assertThatCode;
26 import static org.assertj.core.api.Assertions.assertThatThrownBy;
27 import static org.junit.Assert.assertFalse;
29 import java.util.Base64;
30 import org.junit.AfterClass;
31 import org.junit.BeforeClass;
32 import org.junit.Test;
33 import org.onap.policy.api.main.parameters.ApiParameterGroup;
34 import org.onap.policy.common.parameters.ParameterService;
35 import org.onap.policy.common.utils.coder.StandardCoder;
36 import org.onap.policy.common.utils.resources.ResourceUtils;
37 import org.onap.policy.models.base.PfModelException;
38 import org.onap.policy.models.provider.PolicyModelsProviderParameters;
39 import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;
42 * This class performs unit test of {@link PolicyTypeProvider}
44 * @author Chenfei Gao (cgao@research.att.com)
46 public class TestPolicyTypeProvider {
48 private static PolicyTypeProvider policyTypeProvider;
49 private static PolicyProvider policyProvider;
50 private static PolicyModelsProviderParameters providerParams;
51 private static ApiParameterGroup apiParamGroup;
52 private static StandardCoder standardCoder;
54 private static final String POLICY_RESOURCE = "policies/vCPE.policy.monitoring.input.tosca.json";
55 private static final String POLICY_TYPE_RESOURCE = "policytypes/onap.policies.monitoring.cdap.tca.hi.lo.app.json";
58 * Initializes parameters.
60 * @throws PfModelException the PfModel parsing exception
63 public static void setupParameters() throws PfModelException {
65 standardCoder = new StandardCoder();
66 providerParams = new PolicyModelsProviderParameters();
67 providerParams.setDatabaseDriver("org.h2.Driver");
68 providerParams.setDatabaseUrl("jdbc:h2:mem:testdb");
69 providerParams.setDatabaseUser("policy");
70 providerParams.setDatabasePassword(Base64.getEncoder().encodeToString("P01icY".getBytes()));
71 providerParams.setPersistenceUnit("ToscaConceptTest");
72 apiParamGroup = new ApiParameterGroup("ApiGroup", null, providerParams);
73 ParameterService.register(apiParamGroup, true);
74 policyTypeProvider = new PolicyTypeProvider();
75 policyProvider = new PolicyProvider();
79 * Closes up DB connections and deregisters API parameter group.
81 * @throws PfModelException the PfModel parsing exception
84 public static void tearDown() throws PfModelException {
86 policyTypeProvider.close();
87 policyProvider.close();
88 ParameterService.deregister(apiParamGroup);
92 public void testFetchPolicyTypes() {
94 assertThatCode(() -> {
95 ToscaServiceTemplate serviceTemplate = policyTypeProvider.fetchPolicyTypes(null, null);
96 assertFalse(serviceTemplate.getPolicyTypes().isEmpty());
97 }).doesNotThrowAnyException();
99 assertThatThrownBy(() -> {
100 policyTypeProvider.fetchPolicyTypes("dummy", null);
101 }).hasMessage("policy type with ID dummy:null does not exist");
103 assertThatThrownBy(() -> {
104 policyTypeProvider.fetchPolicyTypes("dummy", "dummy");
105 }).hasMessage("policy type with ID dummy:dummy does not exist");
109 public void testFetchLatestPolicyTypes() {
111 assertThatThrownBy(() -> {
112 policyTypeProvider.fetchLatestPolicyTypes("dummy");
113 }).hasMessage("policy type with ID dummy:null does not exist");
117 public void testCreatePolicyType() {
119 assertThatCode(() -> {
120 String policyTypeString = ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE);
121 ToscaServiceTemplate policyTypeServiceTemplate =
122 standardCoder.decode(policyTypeString, ToscaServiceTemplate.class);
123 ToscaServiceTemplate serviceTemplate = policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
124 assertFalse(serviceTemplate.getPolicyTypes().get(0).isEmpty());
125 }).doesNotThrowAnyException();
129 public void testDeletePolicyType() {
131 assertThatCode(() -> {
132 String policyTypeString = ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE);
133 ToscaServiceTemplate policyTypeServiceTemplate =
134 standardCoder.decode(policyTypeString, ToscaServiceTemplate.class);
135 ToscaServiceTemplate serviceTemplate = policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
136 assertFalse(serviceTemplate.getPolicyTypes().get(0).isEmpty());
137 }).doesNotThrowAnyException();
139 assertThatCode(() -> {
140 String policyString = ResourceUtils.getResourceAsString(POLICY_RESOURCE);
141 ToscaServiceTemplate policyServiceTemplate =
142 standardCoder.decode(policyString, ToscaServiceTemplate.class);
143 policyProvider.createPolicy(
144 "onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0", policyServiceTemplate);
145 }).doesNotThrowAnyException();
147 String exceptionMessage = "policy type with ID onap.policies.monitoring.cdap.tca.hi.lo.app:1.0.0 "
148 + "cannot be deleted as it is parameterized by policies onap.restart.tca:1.0.0";
149 assertThatThrownBy(() -> {
150 policyTypeProvider.deletePolicyType("onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0");
151 }).hasMessage(exceptionMessage);
153 assertThatCode(() -> {
154 ToscaServiceTemplate serviceTemplate = policyProvider.deletePolicy(
155 "onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0", "onap.restart.tca", "1.0.0");
156 assertFalse(serviceTemplate.getToscaTopologyTemplate().getPolicies().get(0).isEmpty());
157 }).doesNotThrowAnyException();
159 assertThatCode(() -> {
160 ToscaServiceTemplate serviceTemplate = policyTypeProvider.deletePolicyType(
161 "onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0");
162 assertFalse(serviceTemplate.getPolicyTypes().get(0).isEmpty());
163 }).doesNotThrowAnyException();
165 assertThatThrownBy(() -> {
166 policyTypeProvider.deletePolicyType("onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0");
167 }).hasMessage("policy type with ID onap.policies.monitoring.cdap.tca.hi.lo.app:1.0.0 does not exist");