Fix the bugs in loop state logic
[clamp.git] / src / test / java / org / onap / clamp / loop / LoopOperationTestItCase.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * ONAP CLAMP
4  * ================================================================================
5  * Copyright (C) 2019 Nokia Intellectual Property. All rights
6  *                             reserved.
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
11  *
12  * http://www.apache.org/licenses/LICENSE-2.0
13  *
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.
19  * ============LICENSE_END============================================
20  * ===================================================================
21  *
22  */
23
24 package org.onap.clamp.loop;
25
26 import static org.assertj.core.api.Assertions.assertThat;
27
28 import com.google.gson.Gson;
29 import com.google.gson.GsonBuilder;
30 import com.google.gson.JsonObject;
31
32 import java.io.IOException;
33 import java.util.HashSet;
34
35 import org.apache.camel.Exchange;
36 import org.apache.camel.Message;
37 import org.json.simple.parser.ParseException;
38 import org.junit.Test;
39 import org.junit.runner.RunWith;
40 import org.mockito.Mockito;
41 import org.onap.clamp.clds.Application;
42 import org.onap.clamp.clds.config.ClampProperties;
43 import org.onap.clamp.loop.LoopOperation.TempLoopState;
44 import org.onap.clamp.policy.microservice.MicroServicePolicy;
45 import org.onap.clamp.policy.operational.OperationalPolicy;
46 import org.springframework.beans.factory.annotation.Autowired;
47 import org.springframework.boot.test.context.SpringBootTest;
48 import org.springframework.test.context.junit4.SpringRunner;
49
50 @RunWith(SpringRunner.class)
51 @SpringBootTest(classes = Application.class)
52 public class LoopOperationTestItCase {
53
54     private Gson gson = new GsonBuilder().setPrettyPrinting().excludeFieldsWithoutExposeAnnotation().create();
55     @Autowired
56     LoopService loopService;
57
58     @Autowired
59     ClampProperties property;
60
61     private Loop createTestLoop() {
62         String yaml = "imports:\n"
63             + "  - \"http://www.getcloudify.org/spec/cloudify/3.4/types.yaml\"\n"
64             + "node_templates:\n"
65             + "  docker_service_host:\n"
66             + "    type: dcae.nodes.SelectedDockerHost";
67
68         Loop loopTest = new Loop("ControlLoopTest", yaml, "<xml></xml>");
69         loopTest.setGlobalPropertiesJson(new Gson().fromJson("{\"testname\":\"testvalue\"}", JsonObject.class));
70         loopTest.setLastComputedState(LoopState.DESIGN);
71         loopTest.setDcaeDeploymentId("123456789");
72         loopTest.setDcaeDeploymentStatusUrl("http4://localhost:8085");
73         loopTest.setDcaeBlueprintId("UUID-blueprint");
74
75         MicroServicePolicy microServicePolicy = new MicroServicePolicy("configPolicyTest", "",
76             "tosca_definitions_version: tosca_simple_yaml_1_0_0", true,
77              gson.fromJson("{\"configtype\":\"json\"}", JsonObject.class), new HashSet<>());
78         microServicePolicy.setProperties(new Gson().fromJson("{\"param1\":\"value1\"}", JsonObject.class));
79
80         loopTest.addMicroServicePolicy(microServicePolicy);
81         return loopTest;
82     }
83
84
85     @Test
86     public void testAnalysePolicyResponse() {
87         LoopOperation loopOp = new LoopOperation(loopService, property);
88         String status1 = loopOp.analysePolicyResponse(200);
89         String status2 = loopOp.analysePolicyResponse(404);
90         String status3 = loopOp.analysePolicyResponse(500);
91         String status4 = loopOp.analysePolicyResponse(503);
92
93         // then
94         assertThat(status1).isEqualTo("SUBMITTED");
95         assertThat(status2).isEqualTo("NOT_SUBMITTED");
96         assertThat(status3).isEqualTo("IN_ERROR");
97         assertThat(status4).isEqualTo("IN_ERROR");
98     }
99
100     @Test
101     public void testGetOperationalPolicyName() {
102         LoopOperation loopOp = new LoopOperation(loopService, property);
103         Loop loop = this.createTestLoop();
104         String opName1 = loopOp.getOperationalPolicyName(loop);
105         assertThat(opName1).isNull();
106
107         OperationalPolicy opPolicy1 = new OperationalPolicy("OperationalPolicyTest1", null,
108                 gson.fromJson("{\"type\":\"Operational\"}", JsonObject.class));
109         loop.addOperationalPolicy(opPolicy1);
110         String opName2 = loopOp.getOperationalPolicyName(loop);
111         assertThat(opName2).isEqualTo("OperationalPolicyTest1");
112     }
113
114     @Test
115     public void testAnalyseDcaeResponse() throws ParseException {
116         LoopOperation loopOp = new LoopOperation(loopService, property);
117         String dcaeStatus1 = loopOp.analyseDcaeResponse(null, null);
118         assertThat(dcaeStatus1).isEqualTo("NOT_DEPLOYED");
119
120         String dcaeStatus2 = loopOp.analyseDcaeResponse(null, 500);
121         assertThat(dcaeStatus2).isEqualTo("IN_ERROR");
122
123         String dcaeStatus3 = loopOp.analyseDcaeResponse(null, 404);
124         assertThat(dcaeStatus3).isEqualTo("NOT_DEPLOYED");
125
126         Exchange camelExchange = Mockito.mock(Exchange.class);
127         Message mockMessage = Mockito.mock(Message.class);
128         Mockito.when(camelExchange.getIn()).thenReturn(mockMessage);
129         Mockito.when(mockMessage.getBody(String.class))
130             .thenReturn("{\"operationType\":\"install\",\"status\":\"succeeded\"}");
131         String dcaeStatus4 = loopOp.analyseDcaeResponse(camelExchange, 200);
132         assertThat(dcaeStatus4).isEqualTo("DEPLOYED");
133
134         Mockito.when(mockMessage.getBody(String.class))
135             .thenReturn("{\"operationType\":\"install\",\"status\":\"processing\"}");
136         String dcaeStatus5 = loopOp.analyseDcaeResponse(camelExchange, 200);
137         assertThat(dcaeStatus5).isEqualTo("PROCESSING");
138
139         Mockito.when(mockMessage.getBody(String.class))
140             .thenReturn("{\"operationType\":\"install\",\"status\":\"failed\"}");
141         String dcaeStatus6 = loopOp.analyseDcaeResponse(camelExchange, 200);
142         assertThat(dcaeStatus6).isEqualTo("IN_ERROR");
143
144         Mockito.when(mockMessage.getBody(String.class))
145             .thenReturn("{\"operationType\":\"uninstall\",\"status\":\"succeeded\"}");
146         String dcaeStatus7 = loopOp.analyseDcaeResponse(camelExchange, 200);
147         assertThat(dcaeStatus7).isEqualTo("NOT_DEPLOYED");
148
149         Mockito.when(mockMessage.getBody(String.class))
150             .thenReturn("{\"operationType\":\"uninstall\",\"status\":\"processing\"}");
151         String dcaeStatus8 = loopOp.analyseDcaeResponse(camelExchange, 200);
152         assertThat(dcaeStatus8).isEqualTo("PROCESSING");
153
154         Mockito.when(mockMessage.getBody(String.class))
155             .thenReturn("{\"operationType\":\"uninstall\",\"status\":\"failed\"}");
156         String dcaeStatus9 = loopOp.analyseDcaeResponse(camelExchange, 200);
157         assertThat(dcaeStatus9).isEqualTo("IN_ERROR");
158     }
159
160     @Test
161     public void testUpdateLoopStatus() {
162         LoopOperation loopOp = new LoopOperation(loopService, property);
163         Loop loop = this.createTestLoop();
164         loopService.saveOrUpdateLoop(loop);
165         LoopState newState1 = loopOp.updateLoopStatus(loop, TempLoopState.SUBMITTED, TempLoopState.DEPLOYED);
166         LoopState dbState1 = loopService.getLoop(loop.getName()).getLastComputedState();
167         assertThat(newState1).isEqualTo(LoopState.DEPLOYED);
168         assertThat(dbState1).isEqualTo(LoopState.DEPLOYED);
169
170         LoopState newState2 = loopOp.updateLoopStatus(loop, TempLoopState.SUBMITTED, TempLoopState.NOT_DEPLOYED);
171         LoopState dbState2 = loopService.getLoop(loop.getName()).getLastComputedState();
172         assertThat(newState2).isEqualTo(LoopState.SUBMITTED);
173         assertThat(dbState2).isEqualTo(LoopState.SUBMITTED);
174
175         LoopState newState3 = loopOp.updateLoopStatus(loop, TempLoopState.SUBMITTED, TempLoopState.PROCESSING);
176         assertThat(newState3).isEqualTo(LoopState.WAITING);
177
178         LoopState newState4 = loopOp.updateLoopStatus(loop, TempLoopState.SUBMITTED, TempLoopState.IN_ERROR);
179         assertThat(newState4).isEqualTo(LoopState.IN_ERROR);
180
181         LoopState newState5 = loopOp.updateLoopStatus(loop, TempLoopState.NOT_SUBMITTED, TempLoopState.DEPLOYED);
182         assertThat(newState5).isEqualTo(LoopState.IN_ERROR);
183
184         LoopState newState6 = loopOp.updateLoopStatus(loop, TempLoopState.NOT_SUBMITTED, TempLoopState.PROCESSING);
185         assertThat(newState6).isEqualTo(LoopState.IN_ERROR);
186
187         LoopState newState7 = loopOp.updateLoopStatus(loop, TempLoopState.NOT_SUBMITTED, TempLoopState.NOT_DEPLOYED);
188         assertThat(newState7).isEqualTo(LoopState.DESIGN);
189
190         LoopState newState8 = loopOp.updateLoopStatus(loop, TempLoopState.IN_ERROR, TempLoopState.DEPLOYED);
191         assertThat(newState8).isEqualTo(LoopState.IN_ERROR);
192
193         LoopState newState9 = loopOp.updateLoopStatus(loop, TempLoopState.IN_ERROR, TempLoopState.NOT_DEPLOYED);
194         assertThat(newState9).isEqualTo(LoopState.IN_ERROR);
195
196         LoopState newState10 = loopOp.updateLoopStatus(loop, TempLoopState.IN_ERROR, TempLoopState.PROCESSING);
197         assertThat(newState10).isEqualTo(LoopState.IN_ERROR);
198
199         LoopState newState11 = loopOp.updateLoopStatus(loop, TempLoopState.IN_ERROR, TempLoopState.IN_ERROR);
200         assertThat(newState11).isEqualTo(LoopState.IN_ERROR);
201     }
202
203     @Test
204     public void testUpdateLoopInfo() throws ParseException {
205         Loop loop = this.createTestLoop();
206         loopService.saveOrUpdateLoop(loop);
207
208         Exchange camelExchange = Mockito.mock(Exchange.class);
209         Message mockMessage = Mockito.mock(Message.class);
210         Mockito.when(camelExchange.getIn()).thenReturn(mockMessage);
211         Mockito.when(mockMessage.getBody(String.class))
212             .thenReturn("{\"links\":{\"status\":\"http://testhost/dcae-operationstatus\",\"test2\":\"test2\"}}");
213
214         LoopOperation loopOp = new LoopOperation(loopService, property);
215         loopOp.updateLoopInfo(camelExchange, loop, "testNewId");
216
217         Loop newLoop = loopService.getLoop(loop.getName());
218         String newDeployId =  newLoop.getDcaeDeploymentId();
219         String newDeploymentStatusUrl = newLoop.getDcaeDeploymentStatusUrl();
220
221         assertThat(newDeployId).isEqualTo("testNewId");
222         assertThat(newDeploymentStatusUrl).isEqualTo("http4://testhost/dcae-operationstatus");
223     }
224
225     @Test
226     public void testGetDeploymentId() {
227         Loop loop = this.createTestLoop();
228         LoopOperation loopOp = new LoopOperation(loopService, property);
229         String deploymentId1 = loopOp.getDeploymentId(loop);
230         assertThat(deploymentId1).isEqualTo("123456789");
231
232         loop.setDcaeDeploymentId(null);
233         String deploymentId2 = loopOp.getDeploymentId(loop);
234         assertThat(deploymentId2).isEqualTo("closedLoop_ControlLoopTest_deploymentId");
235
236         loop.setDcaeDeploymentId("");
237         String deploymentId3 = loopOp.getDeploymentId(loop);
238         assertThat(deploymentId3).isEqualTo("closedLoop_ControlLoopTest_deploymentId");
239     }
240
241     @Test
242     public void testGetDeployPayload() throws IOException {
243         Loop loop = this.createTestLoop();
244         LoopOperation loopOp = new LoopOperation(loopService, property);
245         String deploymentPayload = loopOp.getDeployPayload(loop);
246
247         String expectedPayload = "{\"serviceTypeId\":\"UUID-blueprint\",\"inputs\":{\"imports\":[\"http://www.getcloudify.org/spec/cloudify/3.4/types.yaml\"],\"node_templates\":{\"docker_service_host\":{\"type\":\"dcae.nodes.SelectedDockerHost\"}}}}";
248         assertThat(deploymentPayload).isEqualTo(expectedPayload);
249     }
250 }