2 * ============LICENSE_START=======================================================
3 * Copyright (C) 2021 Nordix Foundation.
4 * ================================================================================
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
9 * http://www.apache.org/licenses/LICENSE-2.0
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
17 * SPDX-License-Identifier: Apache-2.0
18 * ============LICENSE_END=========================================================
21 package org.onap.policy.clamp.controlloop.models.controlloop.persistence.provider;
23 import static org.assertj.core.api.Assertions.assertThat;
24 import static org.assertj.core.api.Assertions.assertThatThrownBy;
25 import static org.junit.Assert.assertEquals;
26 import static org.junit.Assert.assertNull;
28 import java.util.List;
29 import java.util.concurrent.atomic.AtomicInteger;
30 import org.junit.After;
31 import org.junit.Before;
32 import org.junit.Test;
33 import org.onap.policy.clamp.controlloop.models.controlloop.concepts.ControlLoop;
34 import org.onap.policy.clamp.controlloop.models.controlloop.concepts.ControlLoopOrderedState;
35 import org.onap.policy.clamp.controlloop.models.controlloop.concepts.ControlLoops;
36 import org.onap.policy.common.utils.coder.Coder;
37 import org.onap.policy.common.utils.coder.StandardCoder;
38 import org.onap.policy.common.utils.resources.ResourceUtils;
39 import org.onap.policy.models.provider.PolicyModelsProviderParameters;
40 import org.onap.policy.models.tosca.authorative.concepts.ToscaConceptIdentifier;
41 import org.onap.policy.models.tosca.authorative.concepts.ToscaTypedEntityFilter;
43 public class ControlLoopProviderTest {
45 private static final String LIST_IS_NULL = "controlLoops is marked .*ull but is null";
46 private static final Coder CODER = new StandardCoder();
47 private static final String CONTROL_LOOP_JSON = "src/test/resources/providers/TestControlLoops.json";
48 private static final String UPDATE_CL_JSON = "src/test/resources/providers/UpdateControlLoops.json";
50 private static AtomicInteger dbNameCounter = new AtomicInteger();
52 private PolicyModelsProviderParameters parameters;
53 private ControlLoopProvider controlLoopProvider;
54 private ControlLoops inputControlLoops;
55 private ControlLoops updateControlLoops;
56 private String originalJson = ResourceUtils.getResourceAsString(CONTROL_LOOP_JSON);
57 private String updateClJson = ResourceUtils.getResourceAsString(UPDATE_CL_JSON);
60 * Set up test control loop provider.
63 public void setupDao() throws Exception {
65 parameters = new PolicyModelsProviderParameters();
66 parameters.setDatabaseDriver("org.h2.Driver");
67 parameters.setName("PolicyProviderParameterGroup");
68 parameters.setImplementation("org.onap.policy.models.provider.impl.DatabasePolicyModelsProviderImpl");
69 parameters.setDatabaseUrl("jdbc:h2:mem:controlLoopProviderTestDb" + dbNameCounter.getAndDecrement());
70 parameters.setDatabaseUser("policy");
71 parameters.setDatabasePassword("P01icY");
72 parameters.setPersistenceUnit("ToscaConceptTest");
74 controlLoopProvider = new ControlLoopProvider(parameters);
76 inputControlLoops = CODER.decode(originalJson, ControlLoops.class);
77 updateControlLoops = CODER.decode(updateClJson, ControlLoops.class);
81 public void teardown() {
82 controlLoopProvider.close();
86 public void testControlLoopCreate() throws Exception {
87 assertThatThrownBy(() -> {
88 controlLoopProvider.createControlLoops(null);
89 }).hasMessageMatching(LIST_IS_NULL);
91 ControlLoops createdControlLoops = new ControlLoops();
93 .setControlLoopList(controlLoopProvider.createControlLoops(inputControlLoops.getControlLoopList()));
95 assertEquals(inputControlLoops, createdControlLoops);
99 public void testGetControlLoops() throws Exception {
101 List<ControlLoop> getResponse;
103 // Return empty list when no data present in db
104 getResponse = controlLoopProvider.getControlLoops(null, null);
105 assertThat(getResponse).isEmpty();
107 controlLoopProvider.createControlLoops(inputControlLoops.getControlLoopList());
108 String name = inputControlLoops.getControlLoopList().get(0).getName();
109 String version = inputControlLoops.getControlLoopList().get(0).getVersion();
110 assertEquals(1, controlLoopProvider.getControlLoops(name, version).size());
112 ControlLoop cl = new ControlLoop();
113 cl = controlLoopProvider.getControlLoop(new ToscaConceptIdentifier("PMSHInstance1", "1.0.1"));
114 assertEquals(inputControlLoops.getControlLoopList().get(1), cl);
116 assertNull(controlLoopProvider.getControlLoop(new ToscaConceptIdentifier("invalid_name", "1.0.1")));
118 assertThatThrownBy(() -> {
119 controlLoopProvider.getFilteredControlLoops(null);
120 }).hasMessageMatching("filter is marked .*ull but is null");
122 final ToscaTypedEntityFilter<ControlLoop> filter = ToscaTypedEntityFilter.<ControlLoop>builder()
123 .type("org.onap.domain.pmsh.PMSHControlLoopDefinition").build();
124 assertEquals(2, controlLoopProvider.getFilteredControlLoops(filter).size());
129 public void testUpdateControlLoops() throws Exception {
130 assertThatThrownBy(() -> {
131 controlLoopProvider.updateControlLoops(null);
132 }).hasMessageMatching("controlLoops is marked .*ull but is null");
134 ControlLoops existingControlLoops = new ControlLoops();
136 .setControlLoopList(controlLoopProvider.createControlLoops(inputControlLoops.getControlLoopList()));
137 ControlLoop updateResponse = new ControlLoop();
138 updateResponse = controlLoopProvider.updateControlLoop(updateControlLoops.getControlLoopList().get(0));
140 assertEquals(ControlLoopOrderedState.RUNNING, updateResponse.getOrderedState());
144 public void testDeleteControlLoop() throws Exception {
145 assertThatThrownBy(() -> {
146 controlLoopProvider.deleteControlLoop("Invalid_name", "1.0.1");
147 }).hasMessageMatching(".*.failed, control loop does not exist");
149 ControlLoop deletedCl;
150 List<ControlLoop> clList = controlLoopProvider.createControlLoops(inputControlLoops.getControlLoopList());
151 String name = inputControlLoops.getControlLoopList().get(0).getName();
152 String version = inputControlLoops.getControlLoopList().get(0).getVersion();
154 deletedCl = controlLoopProvider.deleteControlLoop(name, version);
155 assertEquals(clList.get(0), deletedCl);