2 * ============LICENSE_START=======================================================
3 * Copyright (C) 2021-2022 Nordix Foundation.
4 * Modifications Copyright (C) 2021 AT&T Intellectual Property. All rights reserved.
5 * ================================================================================
6 * Licensed under the Apache License, Version 2.0 (the "License");
7 * you may not use this file except in compliance with the License.
8 * You may obtain a copy of the License at
10 * http://www.apache.org/licenses/LICENSE-2.0
12 * Unless required by applicable law or agreed to in writing, software
13 * distributed under the License is distributed on an "AS IS" BASIS,
14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 * See the License for the specific language governing permissions and
16 * limitations under the License.
18 * SPDX-License-Identifier: Apache-2.0
19 * ============LICENSE_END=========================================================
22 package org.onap.policy.clamp.acm.participant.kubernetes.rest;
24 import static org.hamcrest.CoreMatchers.is;
25 import static org.mockito.ArgumentMatchers.any;
26 import static org.mockito.Mockito.doNothing;
27 import static org.mockito.Mockito.doReturn;
28 import static org.mockito.Mockito.when;
29 import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.content;
30 import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
31 import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
34 import java.io.IOException;
35 import java.nio.charset.StandardCharsets;
36 import java.util.List;
37 import org.apache.commons.io.FileUtils;
38 import org.json.JSONObject;
39 import org.junit.jupiter.api.BeforeAll;
40 import org.junit.jupiter.api.BeforeEach;
41 import org.junit.jupiter.api.Test;
42 import org.junit.jupiter.api.extension.ExtendWith;
43 import org.onap.policy.clamp.acm.participant.kubernetes.controller.ChartController;
44 import org.onap.policy.clamp.acm.participant.kubernetes.models.ChartInfo;
45 import org.onap.policy.clamp.acm.participant.kubernetes.models.ChartList;
46 import org.onap.policy.clamp.acm.participant.kubernetes.parameters.ParticipantK8sParameters;
47 import org.onap.policy.clamp.acm.participant.kubernetes.service.ChartService;
48 import org.onap.policy.common.utils.coder.Coder;
49 import org.onap.policy.common.utils.coder.CoderException;
50 import org.onap.policy.common.utils.coder.StandardCoder;
51 import org.springframework.beans.factory.annotation.Autowired;
52 import org.springframework.boot.actuate.autoconfigure.metrics.CompositeMeterRegistryAutoConfiguration;
53 import org.springframework.boot.actuate.autoconfigure.metrics.MetricsAutoConfiguration;
54 import org.springframework.boot.context.properties.EnableConfigurationProperties;
55 import org.springframework.boot.test.autoconfigure.web.servlet.WebMvcTest;
56 import org.springframework.boot.test.mock.mockito.MockBean;
57 import org.springframework.context.annotation.Import;
58 import org.springframework.http.MediaType;
59 import org.springframework.mock.web.MockMultipartFile;
60 import org.springframework.test.context.junit.jupiter.SpringExtension;
61 import org.springframework.test.web.servlet.MockMvc;
62 import org.springframework.test.web.servlet.RequestBuilder;
63 import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
64 import org.springframework.test.web.servlet.setup.MockMvcBuilders;
65 import org.springframework.web.context.WebApplicationContext;
68 @ExtendWith(SpringExtension.class)
69 @WebMvcTest(value = ChartController.class, properties = "chart.api.enabled=true")
70 @Import({MetricsAutoConfiguration.class, CompositeMeterRegistryAutoConfiguration.class})
71 @EnableConfigurationProperties(value = ParticipantK8sParameters.class)
72 class ChartControllerTest {
74 private static final Coder CODER = new StandardCoder();
75 private static final String CHART_INFO_YAML = "src/test/resources/ChartList.json";
76 private static List<ChartInfo> charts;
77 private static final String RETRIEVE_CHART_URL = "/helm/charts";
78 private static final String INSTALL_CHART_URL = "/helm/install";
79 private static final String UNINSTALL_CHART_URL = "/helm/uninstall/";
80 private static final String ONBOARD_CHART_URL = "/helm/onboard/chart";
81 private static final String DELETE_CHART_URL = "/helm/chart";
82 private static final String CONFIGURE_REPO_URL = "/helm/repo";
85 private MockMvc mockMvc;
88 private ChartService chartService;
91 private WebApplicationContext context;
94 * Read input chart info json.
95 * @throws CoderException in case of error.
98 static void setupParams() throws CoderException {
99 charts = CODER.decode(new File(CHART_INFO_YAML), ChartList.class).getCharts();
103 * Mock service layer in Controller.
106 void mockServiceClass() {
107 when(chartService.getAllCharts()).thenReturn(charts);
108 when(chartService.getChart(charts.get(0).getChartId().getName(), charts.get(0).getChartId().getVersion()))
109 .thenReturn(charts.get(0));
111 this.mockMvc = MockMvcBuilders.webAppContextSetup(this.context).build();
115 * Test endpoint for retrieving all charts.
116 * @throws Exception in case of error.
119 void retrieveAllCharts() throws Exception {
120 RequestBuilder requestBuilder;
121 requestBuilder = MockMvcRequestBuilders.get(RETRIEVE_CHART_URL).accept(MediaType.APPLICATION_JSON_VALUE);
123 mockMvc.perform(requestBuilder).andExpect(status().isOk())
124 .andExpect(content().contentTypeCompatibleWith(MediaType.APPLICATION_JSON))
125 .andExpect(jsonPath("$.charts.[0].chartId.name", is("HelloWorld")));
129 * Test endpoint for installing a chart.
130 * @throws Exception in case of error.
133 void installChart() throws Exception {
134 RequestBuilder requestBuilder;
136 //Mocking successful installation for void install method
137 doReturn(true).when(chartService).installChart(charts.get(0));
139 requestBuilder = MockMvcRequestBuilders.post(INSTALL_CHART_URL).accept(MediaType.APPLICATION_JSON_VALUE)
140 .content(getInstallationJson(charts.get(0).getChartId().getName(), charts.get(0).getChartId().getVersion()))
141 .contentType(MediaType.APPLICATION_JSON_VALUE);
143 mockMvc.perform(requestBuilder).andExpect(status().isCreated());
145 //Install Invalid chart, expects HTTP status NOT_FOUND
146 requestBuilder = MockMvcRequestBuilders.post(INSTALL_CHART_URL).accept(MediaType.APPLICATION_JSON_VALUE)
147 .content(getInstallationJson("invalidName", "invalidVersion"))
148 .contentType(MediaType.APPLICATION_JSON_VALUE);
150 mockMvc.perform(requestBuilder).andExpect(status().isNotFound());
154 * Test endpoint for uninstalling a chart.
155 * @throws Exception in case of error.
158 void uninstallChart() throws Exception {
159 RequestBuilder requestBuilder;
161 //Mocking successful scenario for void uninstall method
162 doNothing().when(chartService).uninstallChart(charts.get(0));
164 requestBuilder = MockMvcRequestBuilders.delete(UNINSTALL_CHART_URL + charts.get(0)
165 .getChartId().getName() + "/" + charts.get(0).getChartId().getVersion())
166 .accept(MediaType.APPLICATION_JSON_VALUE).contentType(MediaType.APPLICATION_JSON_VALUE);
168 mockMvc.perform(requestBuilder).andExpect(status().isNoContent());
171 requestBuilder = MockMvcRequestBuilders.delete(UNINSTALL_CHART_URL + "invalidName"
172 + "/" + "invalidVersion").accept(MediaType.APPLICATION_JSON_VALUE)
173 .contentType(MediaType.APPLICATION_JSON_VALUE);
175 mockMvc.perform(requestBuilder).andExpect(status().isNotFound());
179 * Test endpoint for chart onboarding.
180 * @throws Exception in case of error.
183 void onboardChart() throws Exception {
184 RequestBuilder requestBuilder;
185 MockMultipartFile chartFile = new MockMultipartFile("chart", "hello.tgz",
186 MediaType.TEXT_PLAIN_VALUE, "Dummy data".getBytes());
188 MockMultipartFile overrideFile = new MockMultipartFile("values", "values.yaml",
189 MediaType.TEXT_PLAIN_VALUE, "Dummy data".getBytes());
191 //Mocking successful scenario for void uninstall method
192 when(chartService.saveChart(charts.get(0), chartFile, null)).thenReturn(charts.get(0));
194 requestBuilder = MockMvcRequestBuilders.multipart(ONBOARD_CHART_URL)
195 .file(chartFile).file(overrideFile).param("info", getChartInfoJson());
197 mockMvc.perform(requestBuilder).andExpect(status().isOk());
201 * Test endpoint for deleting a chart.
202 * @throws Exception in case of error.
205 void deleteChart() throws Exception {
206 RequestBuilder requestBuilder;
208 //Mocking successful scenario for void uninstall method
209 doNothing().when(chartService).deleteChart(charts.get(0));
211 requestBuilder = MockMvcRequestBuilders.delete(DELETE_CHART_URL + "/" + charts.get(0)
212 .getChartId().getName() + "/" + charts.get(0).getChartId().getVersion())
213 .accept(MediaType.APPLICATION_JSON_VALUE)
214 .contentType(MediaType.APPLICATION_JSON_VALUE);
216 mockMvc.perform(requestBuilder).andExpect(status().isNoContent());
218 requestBuilder = MockMvcRequestBuilders.delete(UNINSTALL_CHART_URL + "invalidName"
219 + "/" + "invalidVersion").accept(MediaType.APPLICATION_JSON_VALUE)
220 .contentType(MediaType.APPLICATION_JSON_VALUE);
222 mockMvc.perform(requestBuilder).andExpect(status().isNotFound());
227 * Test endpoint for configuring a helm repository.
228 * @throws Exception in case of error.
231 void testConfigureRepo() throws Exception {
232 RequestBuilder requestBuilder;
233 when(chartService.configureRepository(any())).thenReturn(true);
235 requestBuilder = MockMvcRequestBuilders.post(CONFIGURE_REPO_URL).accept(MediaType.APPLICATION_JSON_VALUE)
236 .content(getInstallationJson(charts.get(0).getChartId().getName(), charts.get(0).getChartId().getVersion()))
237 .contentType(MediaType.APPLICATION_JSON_VALUE);
239 mockMvc.perform(requestBuilder).andExpect(status().isCreated());
244 void testConfigureRepoAlreadyExist() throws Exception {
245 RequestBuilder requestBuilder;
246 when(chartService.configureRepository(any())).thenReturn(false);
248 requestBuilder = MockMvcRequestBuilders.post(CONFIGURE_REPO_URL).accept(MediaType.APPLICATION_JSON_VALUE)
249 .content(getInstallationJson(charts.get(0).getChartId().getName(), charts.get(0).getChartId().getVersion()))
250 .contentType(MediaType.APPLICATION_JSON_VALUE);
252 mockMvc.perform(requestBuilder).andExpect(status().isConflict());
256 private String getInstallationJson(String name, String version) {
257 JSONObject jsonObj = new JSONObject();
258 jsonObj.put("name", name);
259 jsonObj.put("version", version);
260 return jsonObj.toString();
263 private String getChartInfoJson() throws IOException {
264 return FileUtils.readFileToString(new File(CHART_INFO_YAML), StandardCharsets.UTF_8);