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.assertEquals;
29 import static org.junit.Assert.assertFalse;
30 import static org.junit.Assert.assertNotNull;
31 import static org.junit.Assert.assertTrue;
32 import static org.junit.Assert.fail;
34 import java.util.ArrayList;
35 import java.util.Base64;
36 import java.util.List;
39 import org.junit.After;
40 import org.junit.Before;
41 import org.junit.Test;
42 import org.onap.policy.api.main.parameters.ApiParameterGroup;
43 import org.onap.policy.common.parameters.ParameterService;
44 import org.onap.policy.common.utils.coder.StandardCoder;
45 import org.onap.policy.common.utils.coder.StandardYamlCoder;
46 import org.onap.policy.common.utils.resources.ResourceUtils;
47 import org.onap.policy.models.base.PfModelException;
48 import org.onap.policy.models.pdp.concepts.Pdp;
49 import org.onap.policy.models.pdp.concepts.PdpGroup;
50 import org.onap.policy.models.pdp.concepts.PdpGroupFilter;
51 import org.onap.policy.models.pdp.concepts.PdpSubGroup;
52 import org.onap.policy.models.pdp.enums.PdpHealthStatus;
53 import org.onap.policy.models.pdp.enums.PdpState;
54 import org.onap.policy.models.provider.PolicyModelsProvider;
55 import org.onap.policy.models.provider.PolicyModelsProviderFactory;
56 import org.onap.policy.models.provider.PolicyModelsProviderParameters;
57 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicyIdentifier;
58 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicyTypeIdentifier;
59 import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;
60 import org.onap.policy.models.tosca.legacy.concepts.LegacyGuardPolicyInput;
61 import org.onap.policy.models.tosca.legacy.concepts.LegacyGuardPolicyOutput;
64 * This class performs unit test of {@link LegacyGuardPolicyProvider}.
66 * @author Chenfei Gao (cgao@research.att.com)
68 public class TestLegacyGuardPolicyProvider {
70 private static LegacyGuardPolicyProvider guardPolicyProvider;
71 private static PolicyTypeProvider policyTypeProvider;
72 private static PolicyModelsProviderParameters providerParams;
73 private static ApiParameterGroup apiParamGroup;
74 private static StandardCoder standardCoder;
75 private static StandardYamlCoder standardYamlCoder;
77 private static final String POLICY_RESOURCE = "policies/vDNS.policy.guard.frequency.input.json";
78 private static final String POLICY_RESOURCE_VER1 = "policies/vDNS.policy.guard.frequency.input.ver1.json";
79 private static final String POLICY_RESOURCE_VER2 = "policies/vDNS.policy.guard.frequency.input.ver2.json";
80 private static final String POLICY_TYPE_RESOURCE =
81 "policytypes/onap.policies.controlloop.guard.FrequencyLimiter.yaml";
82 private static final String POLICY_TYPE_ID = "onap.policies.controlloop.guard.FrequencyLimiter:1.0.0";
83 private static final String POLICY_TYPE_NAME = "onap.policies.controlloop.guard.FrequencyLimiter";
84 private static final String POLICY_TYPE_VERSION = "1.0.0";
85 private static final String POLICY_ID = "guard.frequency.scaleout:1.0.0";
86 private static final String POLICY_NAME = "guard.frequency.scaleout";
87 private static final String POLICY_VERSION = "1";
88 private static final String LEGACY_MINOR_PATCH_SUFFIX = ".0.0";
91 * Initializes parameters.
93 * @throws PfModelException the PfModel parsing exception
96 public void setupParameters() throws PfModelException {
98 standardCoder = new StandardCoder();
99 standardYamlCoder = new StandardYamlCoder();
100 providerParams = new PolicyModelsProviderParameters();
101 providerParams.setDatabaseDriver("org.h2.Driver");
102 providerParams.setDatabaseUrl("jdbc:h2:mem:testdb");
103 providerParams.setDatabaseUser("policy");
104 providerParams.setDatabasePassword(Base64.getEncoder().encodeToString("P01icY".getBytes()));
105 providerParams.setPersistenceUnit("ToscaConceptTest");
106 apiParamGroup = new ApiParameterGroup("ApiGroup", null, providerParams);
107 ParameterService.register(apiParamGroup, true);
108 guardPolicyProvider = new LegacyGuardPolicyProvider();
109 policyTypeProvider = new PolicyTypeProvider();
113 * Closes up DB connections and deregisters API parameter group.
115 * @throws PfModelException the PfModel parsing exception
118 public void tearDown() throws PfModelException {
120 guardPolicyProvider.close();
121 policyTypeProvider.close();
122 ParameterService.deregister(apiParamGroup);
127 public void testFetchGuardPolicy() {
129 assertThatThrownBy(() -> {
130 guardPolicyProvider.fetchGuardPolicy("dummy", null);
131 }).hasMessage("no policy found for policy: dummy:null");
133 assertThatThrownBy(() -> {
134 guardPolicyProvider.fetchGuardPolicy("dummy", "dummy");
135 }).hasMessage("legacy policy version is not an integer");
137 assertThatCode(() -> {
138 ToscaServiceTemplate policyTypeServiceTemplate = standardYamlCoder.decode(
139 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE), ToscaServiceTemplate.class);
140 policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
142 String policyString = ResourceUtils.getResourceAsString(POLICY_RESOURCE_VER1);
143 LegacyGuardPolicyInput policyToCreate = standardCoder.decode(policyString, LegacyGuardPolicyInput.class);
144 Map<String, LegacyGuardPolicyOutput> createdPolicy = guardPolicyProvider.createGuardPolicy(policyToCreate);
145 assertNotNull(createdPolicy);
146 assertFalse(createdPolicy.isEmpty());
148 policyString = ResourceUtils.getResourceAsString(POLICY_RESOURCE_VER2);
149 policyToCreate = standardCoder.decode(policyString, LegacyGuardPolicyInput.class);
150 createdPolicy = guardPolicyProvider.createGuardPolicy(policyToCreate);
151 assertNotNull(createdPolicy);
152 assertFalse(createdPolicy.isEmpty());
154 Map<String, LegacyGuardPolicyOutput> firstVersion =
155 guardPolicyProvider.fetchGuardPolicy("guard.frequency.scaleout", "1");
156 assertNotNull(firstVersion);
158 firstVersion.get("guard.frequency.scaleout").getMetadata().get("policy-version").toString());
160 Map<String, LegacyGuardPolicyOutput> latestVersion =
161 guardPolicyProvider.fetchGuardPolicy("guard.frequency.scaleout", null);
162 assertNotNull(latestVersion);
164 latestVersion.get("guard.frequency.scaleout").getMetadata().get("policy-version").toString());
165 }).doesNotThrowAnyException();
167 assertThatThrownBy(() -> {
168 guardPolicyProvider.fetchGuardPolicy("guard.frequency.scaleout", "1.0.0");
169 }).hasMessage("legacy policy version is not an integer");
171 assertThatThrownBy(() -> {
172 guardPolicyProvider.fetchGuardPolicy("guard.frequency.scaleout", "latest");
173 }).hasMessage("legacy policy version is not an integer");
175 assertThatCode(() -> {
176 guardPolicyProvider.deleteGuardPolicy("guard.frequency.scaleout", "1");
177 guardPolicyProvider.deleteGuardPolicy("guard.frequency.scaleout", "2");
178 policyTypeProvider.deletePolicyType("onap.policies.controlloop.guard.FrequencyLimiter", "1.0.0");
179 }).doesNotThrowAnyException();
183 public void testFetchDeployedGuardPolicies() {
185 assertThatThrownBy(() -> {
186 guardPolicyProvider.fetchDeployedGuardPolicies("dummy");
187 }).hasMessage("No policy type defined for dummy");
189 try (PolicyModelsProvider databaseProvider =
190 new PolicyModelsProviderFactory().createPolicyModelsProvider(providerParams)) {
191 assertEquals(0, databaseProvider.getPdpGroups("name").size());
192 assertEquals(0, databaseProvider.getFilteredPdpGroups(PdpGroupFilter.builder().build()).size());
194 assertNotNull(databaseProvider.createPdpGroups(new ArrayList<>()));
195 assertNotNull(databaseProvider.updatePdpGroups(new ArrayList<>()));
197 PdpGroup pdpGroup = new PdpGroup();
198 pdpGroup.setName("group");
199 pdpGroup.setVersion("1.2.3");
200 pdpGroup.setPdpGroupState(PdpState.ACTIVE);
201 pdpGroup.setPdpSubgroups(new ArrayList<>());
202 List<PdpGroup> groupList = new ArrayList<>();
203 groupList.add(pdpGroup);
205 PdpSubGroup pdpSubGroup = new PdpSubGroup();
206 pdpSubGroup.setPdpType("type");
207 pdpSubGroup.setDesiredInstanceCount(123);
208 pdpSubGroup.setSupportedPolicyTypes(new ArrayList<>());
209 pdpSubGroup.getSupportedPolicyTypes()
210 .add(new ToscaPolicyTypeIdentifier(POLICY_TYPE_NAME, POLICY_TYPE_VERSION));
211 pdpGroup.getPdpSubgroups().add(pdpSubGroup);
214 pdp.setInstanceId("type-0");
215 pdp.setMessage("Hello");
216 pdp.setPdpState(PdpState.ACTIVE);
217 pdp.setHealthy(PdpHealthStatus.UNKNOWN);
218 pdpSubGroup.setPdpInstances(new ArrayList<>());
219 pdpSubGroup.getPdpInstances().add(pdp);
222 assertEquals(123, databaseProvider.createPdpGroups(groupList).get(0).getPdpSubgroups().get(0)
223 .getDesiredInstanceCount());
224 assertEquals(1, databaseProvider.getPdpGroups("group").size());
226 // Create Policy Type
227 assertThatCode(() -> {
228 ToscaServiceTemplate policyTypeServiceTemplate = standardYamlCoder.decode(
229 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE), ToscaServiceTemplate.class);
230 policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
231 }).doesNotThrowAnyException();
234 assertThatCode(() -> {
235 String policyString = ResourceUtils.getResourceAsString(POLICY_RESOURCE);
236 LegacyGuardPolicyInput policyToCreate =
237 standardCoder.decode(policyString, LegacyGuardPolicyInput.class);
238 Map<String, LegacyGuardPolicyOutput> policyCreated =
239 guardPolicyProvider.createGuardPolicy(policyToCreate);
240 assertFalse(policyCreated.isEmpty());
241 }).doesNotThrowAnyException();
243 // Test fetchDeployedPolicies (deployedPolicyMap.isEmpty())==true
244 assertThatThrownBy(() -> {
245 guardPolicyProvider.fetchDeployedGuardPolicies(POLICY_NAME);
246 }) .hasMessage("could not find policy with ID " + POLICY_NAME + " and type " + POLICY_TYPE_ID
247 + " deployed in any pdp group");
250 // Update pdpSubGroup
251 pdpSubGroup.setPolicies(new ArrayList<>());
252 pdpSubGroup.getPolicies()
253 .add(new ToscaPolicyIdentifier(POLICY_NAME, POLICY_VERSION + LEGACY_MINOR_PATCH_SUFFIX));
255 databaseProvider.createPdpGroups(groupList).get(0).getPdpSubgroups().get(0).getPolicies().size());
257 // Test fetchDeployedPolicies
258 assertThatCode(() -> {
259 guardPolicyProvider.fetchDeployedGuardPolicies(POLICY_NAME);
260 }).doesNotThrowAnyException();
262 // Test validateDeleteEligibility exception path(!pdpGroups.isEmpty())
263 assertThatThrownBy(() -> {
264 guardPolicyProvider.deleteGuardPolicy(POLICY_NAME, POLICY_VERSION);
265 }) .hasMessageContaining("policy with ID " + POLICY_NAME + ":" + POLICY_VERSION
266 + " cannot be deleted as it is deployed in pdp groups");
267 } catch (Exception exc) {
268 fail("Test should not throw an exception");
273 public void testCreateGuardPolicy() {
275 assertThatThrownBy(() -> {
276 String policyString = ResourceUtils.getResourceAsString(POLICY_RESOURCE);
277 LegacyGuardPolicyInput policyToCreate = standardCoder.decode(policyString, LegacyGuardPolicyInput.class);
278 guardPolicyProvider.createGuardPolicy(policyToCreate);
279 }).hasMessage("policy type " + POLICY_TYPE_ID + " for policy " + POLICY_ID + " does not exist");
281 assertThatCode(() -> {
282 ToscaServiceTemplate policyTypeServiceTemplate = standardYamlCoder.decode(
283 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE), ToscaServiceTemplate.class);
284 policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
286 String policyString = ResourceUtils.getResourceAsString(POLICY_RESOURCE);
287 LegacyGuardPolicyInput policyToCreate = standardCoder.decode(policyString, LegacyGuardPolicyInput.class);
288 Map<String, LegacyGuardPolicyOutput> createdPolicy = guardPolicyProvider.createGuardPolicy(policyToCreate);
289 assertNotNull(createdPolicy);
290 assertFalse(createdPolicy.isEmpty());
291 assertTrue(createdPolicy.containsKey("guard.frequency.scaleout"));
292 assertEquals("onap.policies.controlloop.guard.FrequencyLimiter",
293 createdPolicy.get("guard.frequency.scaleout").getType());
294 assertEquals("1.0.0", createdPolicy.get("guard.frequency.scaleout").getVersion());
295 }).doesNotThrowAnyException();
299 public void testDeleteGuardPolicyException() {
300 String policyId = "guard.frequency.scaleout";
301 String policyVersion = "1";
302 String policyTypeVersion = "1.0.0";
303 String policyTypeId = "onap.policies.controlloop.guard.FrequencyLimiter";
304 String legacyMinorPatchSuffix = ".0.0";
306 try (PolicyModelsProvider databaseProvider =
307 new PolicyModelsProviderFactory().createPolicyModelsProvider(providerParams)) {
308 assertEquals(0, databaseProvider.getPdpGroups("name").size());
309 assertEquals(0, databaseProvider.getFilteredPdpGroups(PdpGroupFilter.builder().build()).size());
311 assertNotNull(databaseProvider.createPdpGroups(new ArrayList<>()));
312 assertNotNull(databaseProvider.updatePdpGroups(new ArrayList<>()));
314 PdpGroup pdpGroup = new PdpGroup();
315 pdpGroup.setName("group");
316 pdpGroup.setVersion("1.2.3");
317 pdpGroup.setPdpGroupState(PdpState.ACTIVE);
318 pdpGroup.setPdpSubgroups(new ArrayList<>());
319 List<PdpGroup> groupList = new ArrayList<>();
320 groupList.add(pdpGroup);
322 PdpSubGroup pdpSubGroup = new PdpSubGroup();
323 pdpSubGroup.setPdpType("type");
324 pdpSubGroup.setDesiredInstanceCount(123);
325 pdpSubGroup.setSupportedPolicyTypes(new ArrayList<>());
326 pdpSubGroup.getSupportedPolicyTypes().add(new ToscaPolicyTypeIdentifier(policyTypeId, policyTypeVersion));
327 pdpGroup.getPdpSubgroups().add(pdpSubGroup);
330 pdp.setInstanceId("type-0");
331 pdp.setMessage("Hello");
332 pdp.setPdpState(PdpState.ACTIVE);
333 pdp.setHealthy(PdpHealthStatus.UNKNOWN);
334 pdpSubGroup.setPdpInstances(new ArrayList<>());
335 pdpSubGroup.getPdpInstances().add(pdp);
338 assertEquals(123, databaseProvider.createPdpGroups(groupList).get(0).getPdpSubgroups().get(0)
339 .getDesiredInstanceCount());
340 assertEquals(1, databaseProvider.getPdpGroups("group").size());
342 // Create Policy Type
343 assertThatCode(() -> {
344 ToscaServiceTemplate policyTypeServiceTemplate = standardYamlCoder.decode(
345 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE), ToscaServiceTemplate.class);
346 policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
347 }).doesNotThrowAnyException();
350 assertThatCode(() -> {
351 String policyString = ResourceUtils.getResourceAsString(POLICY_RESOURCE);
352 LegacyGuardPolicyInput policyToCreate =
353 standardCoder.decode(policyString, LegacyGuardPolicyInput.class);
354 Map<String, LegacyGuardPolicyOutput> createdPolicy =
355 guardPolicyProvider.createGuardPolicy(policyToCreate);
356 assertNotNull(createdPolicy);
357 assertFalse(createdPolicy.isEmpty());
358 }).doesNotThrowAnyException();
360 // Update pdpSubGroup
361 pdpSubGroup.setPolicies(new ArrayList<>());
362 pdpSubGroup.getPolicies().add(new ToscaPolicyIdentifier(policyId, policyVersion + legacyMinorPatchSuffix));
364 databaseProvider.createPdpGroups(groupList).get(0).getPdpSubgroups().get(0).getPolicies().size());
365 assertThatThrownBy(() -> {
366 guardPolicyProvider.deleteGuardPolicy("guard.frequency.scaleout", "1");
367 }).hasMessageContaining("cannot be deleted as it is deployed in pdp groups");
368 } catch (Exception exc) {
369 fail("Test should not throw an exception");
374 public void testDeleteGuardPolicy() {
375 assertThatThrownBy(() -> {
376 guardPolicyProvider.deleteGuardPolicy("dummy", null);
377 }).hasMessage("legacy policy version is not an integer");
379 assertThatThrownBy(() -> {
380 guardPolicyProvider.deleteGuardPolicy("dummy", "1.0.0");
381 }).hasMessage("legacy policy version is not an integer");
383 assertThatCode(() -> {
384 ToscaServiceTemplate policyTypeServiceTemplate = standardYamlCoder.decode(
385 ResourceUtils.getResourceAsString(POLICY_TYPE_RESOURCE), ToscaServiceTemplate.class);
386 policyTypeProvider.createPolicyType(policyTypeServiceTemplate);
388 String policyString = ResourceUtils.getResourceAsString(POLICY_RESOURCE);
389 LegacyGuardPolicyInput policyToCreate = standardCoder.decode(policyString, LegacyGuardPolicyInput.class);
390 Map<String, LegacyGuardPolicyOutput> createdPolicy = guardPolicyProvider.createGuardPolicy(policyToCreate);
391 assertNotNull(createdPolicy);
392 assertFalse(createdPolicy.isEmpty());
394 Map<String, LegacyGuardPolicyOutput> deletedPolicy =
395 guardPolicyProvider.deleteGuardPolicy("guard.frequency.scaleout", "1");
396 assertNotNull(deletedPolicy);
397 assertFalse(deletedPolicy.isEmpty());
398 assertTrue(deletedPolicy.containsKey("guard.frequency.scaleout"));
399 assertEquals("onap.policies.controlloop.guard.FrequencyLimiter",
400 deletedPolicy.get("guard.frequency.scaleout").getType());
402 deletedPolicy.get("guard.frequency.scaleout").getMetadata().get("policy-version").toString());
404 }).doesNotThrowAnyException();
406 assertThatThrownBy(() -> {
407 guardPolicyProvider.deleteGuardPolicy("guard.frequency.scaleout", "1");
408 }).hasMessage("no policy found for policy: guard.frequency.scaleout:1");
410 assertThatCode(() -> {
411 policyTypeProvider.deletePolicyType("onap.policies.controlloop.guard.FrequencyLimiter", "1.0.0");
412 }).doesNotThrowAnyException();