2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2019 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.assertThatCode;
27 import static org.assertj.core.api.Assertions.assertThatThrownBy;
28 import static org.junit.Assert.assertFalse;
30 import java.util.Base64;
31 import org.junit.AfterClass;
32 import org.junit.BeforeClass;
33 import org.junit.Test;
34 import org.onap.policy.api.main.parameters.ApiParameterGroup;
35 import org.onap.policy.common.parameters.ParameterService;
36 import org.onap.policy.common.utils.coder.StandardCoder;
37 import org.onap.policy.common.utils.coder.StandardYamlCoder;
38 import org.onap.policy.common.utils.resources.ResourceUtils;
39 import org.onap.policy.models.base.PfModelException;
40 import org.onap.policy.models.provider.PolicyModelsProviderParameters;
41 import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;
44 * This class performs unit test of {@link PolicyTypeProvider}.
46 * @author Chenfei Gao (cgao@research.att.com)
48 public class TestPolicyTypeProvider {
50 private static PolicyTypeProvider policyTypeProvider;
51 private static PolicyProvider policyProvider;
52 private static PolicyModelsProviderParameters providerParams;
53 private static ApiParameterGroup apiParamGroup;
54 private static StandardYamlCoder standardYamlCoder;
56 private static final String POLICY_RESOURCE = "policies/vCPE.policy.monitoring.input.tosca.yaml";
57 private static final String POLICY_TYPE_RESOURCE = "policytypes/onap.policies.monitoring.cdap.tca.hi.lo.app.yaml";
60 * Initializes parameters.
62 * @throws PfModelException the PfModel parsing exception
65 public static void setupParameters() throws PfModelException {
67 standardYamlCoder = new StandardYamlCoder();
68 providerParams = new PolicyModelsProviderParameters();
69 providerParams.setDatabaseDriver("org.h2.Driver");
70 providerParams.setDatabaseUrl("jdbc:h2:mem:testdb");
71 providerParams.setDatabaseUser("policy");
72 providerParams.setDatabasePassword(Base64.getEncoder().encodeToString("P01icY".getBytes()));
73 providerParams.setPersistenceUnit("ToscaConceptTest");
74 apiParamGroup = new ApiParameterGroup("ApiGroup", null, providerParams);
75 ParameterService.register(apiParamGroup, true);
76 policyTypeProvider = new PolicyTypeProvider();
77 policyProvider = new PolicyProvider();
81 * Closes up DB connections and deregisters API parameter group.
83 * @throws PfModelException the PfModel parsing exception
86 public static void tearDown() throws PfModelException {
88 policyTypeProvider.close();
89 policyProvider.close();
90 ParameterService.deregister(apiParamGroup);
94 public void testFetchPolicyTypes() {
96 assertThatCode(() -> {
97 ToscaServiceTemplate serviceTemplate = policyTypeProvider.fetchPolicyTypes(null, null);
98 assertFalse(serviceTemplate.getPolicyTypes().isEmpty());
99 }).doesNotThrowAnyException();
101 assertThatThrownBy(() -> {
102 policyTypeProvider.fetchPolicyTypes("dummy", null);
103 }).hasMessage("policy type with ID dummy:null does not exist");
105 assertThatThrownBy(() -> {
106 policyTypeProvider.fetchPolicyTypes("dummy", "dummy");
107 }).hasMessage("policy type with ID dummy:dummy does not exist");
111 public void testFetchLatestPolicyTypes() {
113 assertThatThrownBy(() -> {
114 policyTypeProvider.fetchLatestPolicyTypes("dummy");
115 }).hasMessage("policy type with ID dummy:null does not exist");
119 public void testCreatePolicyType() {
121 assertThatCode(() -> {
122 ToscaServiceTemplate policyTypeServiceTemplate = standardYamlCoder.decode(
123 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE), ToscaServiceTemplate.class);
124 ToscaServiceTemplate serviceTemplate = policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
125 assertFalse(serviceTemplate.getPolicyTypes().isEmpty());
126 }).doesNotThrowAnyException();
130 public void testDeletePolicyType() {
132 assertThatCode(() -> {
133 ToscaServiceTemplate policyTypeServiceTemplate = standardYamlCoder.decode(
134 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE), ToscaServiceTemplate.class);
135 ToscaServiceTemplate serviceTemplate = policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
136 assertFalse(serviceTemplate.getPolicyTypes().isEmpty());
137 }).doesNotThrowAnyException();
139 assertThatCode(() -> {
140 ToscaServiceTemplate policyServiceTemplate = standardYamlCoder.decode(
141 ResourceUtils.getResourceAsString(POLICY_RESOURCE), ToscaServiceTemplate.class);
142 policyProvider.createPolicy("onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0", policyServiceTemplate);
143 }).doesNotThrowAnyException();
145 String exceptionMessage = "policy type with ID onap.policies.monitoring.cdap.tca.hi.lo.app:1.0.0 "
146 + "cannot be deleted as it is parameterized by policies onap.restart.tca:1.0.0";
147 assertThatThrownBy(() -> {
148 policyTypeProvider.deletePolicyType("onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0");
149 }).hasMessage(exceptionMessage);
151 assertThatCode(() -> {
152 ToscaServiceTemplate serviceTemplate = policyProvider
153 .deletePolicy("onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0", "onap.restart.tca", "1.0.0");
154 assertFalse(serviceTemplate.getToscaTopologyTemplate().getPolicies().get(0).isEmpty());
155 }).doesNotThrowAnyException();
157 assertThatCode(() -> {
158 ToscaServiceTemplate serviceTemplate =
159 policyTypeProvider.deletePolicyType("onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0");
160 assertFalse(serviceTemplate.getPolicyTypes().isEmpty());
161 }).doesNotThrowAnyException();
163 assertThatThrownBy(() -> {
164 policyTypeProvider.deletePolicyType("onap.policies.monitoring.cdap.tca.hi.lo.app", "1.0.0");
165 }).hasMessage("policy type with ID onap.policies.monitoring.cdap.tca.hi.lo.app:1.0.0 does not exist");