2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
6 * Copyright © 2017-2018 European Software Marketing Ltd.
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.
19 * ============LICENSE_END=========================================================
21 package org.onap.aai.modelloader.notification;
23 import static org.hamcrest.CoreMatchers.is;
24 import static org.hamcrest.CoreMatchers.not;
25 import static org.hamcrest.collection.IsEmptyCollection.empty;
26 import static org.junit.Assert.assertThat;
27 import static org.mockito.Mockito.doNothing;
28 import static org.mockito.Mockito.mock;
29 import static org.mockito.Mockito.when;
30 import static org.onap.aai.modelloader.fixture.NotificationDataFixtureBuilder.getNotificationDataWithInvalidType;
31 import static org.onap.aai.modelloader.fixture.NotificationDataFixtureBuilder.getNotificationDataWithModelQuerySpec;
32 import static org.onap.aai.modelloader.fixture.NotificationDataFixtureBuilder.getNotificationDataWithOneOfEach;
33 import static org.onap.aai.modelloader.fixture.NotificationDataFixtureBuilder.getNotificationDataWithOneService;
34 import static org.onap.aai.modelloader.fixture.NotificationDataFixtureBuilder.getNotificationDataWithToscaCsarFile;
36 import java.io.IOException;
37 import java.util.ArrayList;
38 import java.util.List;
39 import java.util.Properties;
40 import org.hamcrest.collection.IsEmptyCollection;
41 import org.junit.After;
42 import org.junit.Before;
43 import org.junit.Test;
44 import org.mockito.Matchers;
45 import org.mockito.Mockito;
46 import org.mockito.internal.util.reflection.Whitebox;
47 import org.onap.aai.babel.service.data.BabelArtifact;
48 import org.onap.aai.modelloader.config.ModelLoaderConfig;
49 import org.onap.aai.modelloader.entity.Artifact;
50 import org.onap.aai.modelloader.entity.model.BabelArtifactParsingException;
51 import org.onap.aai.modelloader.restclient.BabelServiceClient;
52 import org.onap.aai.modelloader.restclient.BabelServiceClientException;
53 import org.onap.aai.modelloader.service.BabelServiceClientFactory;
54 import org.onap.aai.modelloader.service.HttpsBabelServiceClientFactory;
55 import org.onap.aai.modelloader.util.ArtifactTestUtils;
56 import org.onap.sdc.api.IDistributionClient;
57 import org.onap.sdc.api.notification.IArtifactInfo;
58 import org.onap.sdc.api.notification.INotificationData;
59 import org.onap.sdc.api.results.IDistributionClientDownloadResult;
60 import org.onap.sdc.impl.DistributionClientDownloadResultImpl;
61 import org.onap.sdc.utils.DistributionActionResultEnum;
64 * Tests {@link ArtifactDownloadManager}.
66 public class TestArtifactDownloadManager {
68 private ArtifactDownloadManager downloadManager;
69 private BabelServiceClient mockBabelClient;
70 private IDistributionClient mockDistributionClient;
71 private NotificationPublisher mockNotificationPublisher;
72 private BabelArtifactConverter mockBabelArtifactConverter;
73 private BabelServiceClientFactory mockClientFactory;
76 public void setup() throws Exception {
77 mockBabelClient = mock(BabelServiceClient.class);
78 mockDistributionClient = mock(IDistributionClient.class);
79 mockNotificationPublisher = mock(NotificationPublisher.class);
80 mockBabelArtifactConverter = mock(BabelArtifactConverter.class);
81 mockClientFactory = mock(HttpsBabelServiceClientFactory.class);
82 when(mockClientFactory.create(Mockito.any())).thenReturn(mockBabelClient);
84 Properties configProperties = new Properties();
85 configProperties.load(this.getClass().getClassLoader().getResourceAsStream("model-loader.properties"));
86 downloadManager = new ArtifactDownloadManager(mockDistributionClient,
87 new ModelLoaderConfig(configProperties, "."), mockClientFactory);
89 Whitebox.setInternalState(downloadManager, "notificationPublisher", mockNotificationPublisher);
90 Whitebox.setInternalState(downloadManager, "babelArtifactConverter", mockBabelArtifactConverter);
94 public void tearDown() {
95 downloadManager = null;
96 mockDistributionClient = null;
97 mockNotificationPublisher = null;
101 * Test downloading zero artifacts from SDC.
104 public void testDownloadWithZeroArtifacts() {
105 List<Artifact> modelFiles = new ArrayList<>();
106 List<Artifact> catalogFiles = new ArrayList<>();
107 assertThat(downloadManager.downloadArtifacts(getNotificationDataWithOneService(), new ArrayList<>(), modelFiles,
108 catalogFiles), is(true));
109 assertThat(modelFiles, is(empty()));
110 assertThat(catalogFiles, is(empty()));
111 Mockito.verifyZeroInteractions(mockBabelClient, mockDistributionClient, mockNotificationPublisher,
112 mockBabelArtifactConverter);
116 public void testArtifactDownloadFails() {
117 INotificationData data = getNotificationDataWithOneService();
118 IArtifactInfo artifact = data.getServiceArtifacts().get(0);
119 String errorMessage = "error msg";
120 when(mockDistributionClient.download(artifact)).thenReturn(
121 createDistributionClientDownloadResult(DistributionActionResultEnum.FAIL, errorMessage, null));
122 doNothing().when(mockNotificationPublisher).publishDownloadFailure(mockDistributionClient, data, artifact,
125 assertThat(downloadManager.downloadArtifacts(data, data.getServiceArtifacts(), null, null), is(false));
127 Mockito.verify(mockDistributionClient).download(artifact);
128 Mockito.verify(mockNotificationPublisher).publishDownloadFailure(mockDistributionClient, data, artifact,
131 Mockito.verifyZeroInteractions(mockBabelClient, mockBabelArtifactConverter);
135 public void testErrorCreatingBabelClient() throws Exception {
136 when(mockClientFactory.create(Mockito.any())).thenThrow(new BabelServiceClientException(new Exception()));
138 INotificationData data = getNotificationDataWithToscaCsarFile();
139 IArtifactInfo artifactInfo = data.getServiceArtifacts().get(0);
140 setupValidDownloadCsarMocks(data, artifactInfo, new ArtifactTestUtils());
141 doNothing().when(mockNotificationPublisher).publishDeployFailure(mockDistributionClient, data, artifactInfo);
143 assertThat(downloadManager.downloadArtifacts(data, data.getServiceArtifacts(), null, null), is(false));
145 Mockito.verify(mockDistributionClient).download(artifactInfo);
146 Mockito.verify(mockNotificationPublisher).publishDownloadSuccess(mockDistributionClient, data, artifactInfo);
147 Mockito.verify(mockNotificationPublisher).publishDeployFailure(mockDistributionClient, data, artifactInfo);
149 Mockito.verifyZeroInteractions(mockBabelArtifactConverter);
153 public void downloadArtifacts_invalidToscaCsarFile() throws IOException, BabelServiceClientException {
154 INotificationData data = getNotificationDataWithToscaCsarFile();
155 IArtifactInfo artifact = data.getServiceArtifacts().get(0);
156 when(mockDistributionClient.download(artifact)).thenReturn(createDistributionClientDownloadResult(
157 DistributionActionResultEnum.SUCCESS, null, "This is not a valid Tosca CSAR File".getBytes()));
158 doNothing().when(mockNotificationPublisher).publishDownloadSuccess(mockDistributionClient, data, artifact);
159 when(mockBabelClient.postArtifact(Matchers.any(), Matchers.anyString(), Matchers.anyString(),
160 Matchers.anyString())).thenThrow(new BabelServiceClientException(""));
161 doNothing().when(mockNotificationPublisher).publishDeployFailure(mockDistributionClient, data, artifact);
163 assertThat(downloadManager.downloadArtifacts(data, data.getServiceArtifacts(), null, null), is(false));
165 Mockito.verify(mockDistributionClient).download(artifact);
166 Mockito.verify(mockNotificationPublisher).publishDownloadSuccess(mockDistributionClient, data, artifact);
167 Mockito.verify(mockBabelClient).postArtifact(Matchers.any(), Matchers.anyString(), Matchers.anyString(),
168 Matchers.anyString());
169 Mockito.verify(mockNotificationPublisher).publishDeployFailure(mockDistributionClient, data, artifact);
171 Mockito.verifyZeroInteractions(mockBabelArtifactConverter);
176 public void downloadArtifacts_invalidModelQuerySpec() {
177 INotificationData data = getNotificationDataWithModelQuerySpec();
178 IArtifactInfo artifact = data.getServiceArtifacts().get(0);
180 List<org.onap.aai.modelloader.entity.Artifact> modelArtifacts = new ArrayList<>();
182 when(mockDistributionClient.download(artifact)).thenReturn(createDistributionClientDownloadResult(
183 DistributionActionResultEnum.SUCCESS, null, "This is not a valid Model Query Spec".getBytes()));
184 doNothing().when(mockNotificationPublisher).publishDownloadSuccess(mockDistributionClient, data, artifact);
186 assertThat(downloadManager.downloadArtifacts(data, data.getServiceArtifacts(), modelArtifacts, null),
189 Mockito.verify(mockDistributionClient).download(artifact);
190 Mockito.verify(mockNotificationPublisher).publishDownloadSuccess(mockDistributionClient, data, artifact);
191 Mockito.verify(mockNotificationPublisher).publishDeployFailure(mockDistributionClient, data, artifact);
193 Mockito.verifyZeroInteractions(mockBabelClient, mockBabelArtifactConverter);
197 public void downloadArtifacts_validToscaCsarFile()
198 throws IOException, BabelServiceClientException, BabelArtifactParsingException {
199 INotificationData data = getNotificationDataWithToscaCsarFile();
200 IArtifactInfo artifactInfo = data.getServiceArtifacts().get(0);
202 setupValidDownloadCsarMocks(data, artifactInfo, new ArtifactTestUtils());
204 List<org.onap.aai.modelloader.entity.Artifact> modelArtifacts = new ArrayList<>();
205 List<org.onap.aai.modelloader.entity.Artifact> catalogFiles = new ArrayList<>();
206 assertThat(downloadManager.downloadArtifacts(data, data.getServiceArtifacts(), modelArtifacts, catalogFiles),
208 assertThat(catalogFiles.size(), is(0));
210 Mockito.verify(mockDistributionClient).download(artifactInfo);
211 Mockito.verify(mockNotificationPublisher).publishDownloadSuccess(mockDistributionClient, data, artifactInfo);
212 Mockito.verify(mockBabelClient).postArtifact(Matchers.any(), Matchers.anyString(), Matchers.anyString(),
213 Matchers.anyString());
214 Mockito.verify(mockBabelArtifactConverter).convertToModel(Matchers.any());
215 Mockito.verify(mockBabelArtifactConverter).convertToCatalog(Matchers.any());
218 private void setupValidDownloadCsarMocks(INotificationData data, IArtifactInfo artifactInfo,
219 ArtifactTestUtils artifactTestUtils) throws IOException, BabelServiceClientException {
220 when(mockDistributionClient.download(artifactInfo))
221 .thenReturn(createDistributionClientDownloadResult(DistributionActionResultEnum.SUCCESS, null,
222 artifactTestUtils.loadResource("compressedArtifacts/service-VscpaasTest-csar.csar")));
223 when(mockBabelClient.postArtifact(Matchers.any(), Matchers.anyString(), Matchers.anyString(),
224 Matchers.anyString())).thenReturn(createBabelArtifacts());
227 private List<BabelArtifact> createBabelArtifacts() {
228 List<BabelArtifact> artifactList = new ArrayList<>();
229 artifactList.add(new BabelArtifact("ModelArtifact", BabelArtifact.ArtifactType.MODEL, "Some model payload"));
230 artifactList.add(new BabelArtifact("VNFCArtifact", BabelArtifact.ArtifactType.VNFCATALOG, "Some VNFC payload"));
235 public void downloadArtifactsWithValidModelQuerySpec()
236 throws IOException, BabelServiceClientException, BabelArtifactParsingException {
237 INotificationData data = getNotificationDataWithModelQuerySpec();
238 IArtifactInfo artifact = data.getServiceArtifacts().get(0);
239 setupValidModelQuerySpecMocks(new ArtifactTestUtils(), data, artifact);
241 List<org.onap.aai.modelloader.entity.Artifact> modelFiles = new ArrayList<>();
242 List<org.onap.aai.modelloader.entity.Artifact> catalogFiles = new ArrayList<>();
243 assertThat(downloadManager.downloadArtifacts(data, data.getServiceArtifacts(), modelFiles, catalogFiles),
246 assertThat(modelFiles, is(not(IsEmptyCollection.empty())));
247 assertThat(catalogFiles, is(empty()));
249 Mockito.verify(mockDistributionClient).download(artifact);
250 Mockito.verify(mockNotificationPublisher).publishDownloadSuccess(mockDistributionClient, data, artifact);
252 Mockito.verifyZeroInteractions(mockBabelClient, mockBabelArtifactConverter);
255 private void setupValidModelQuerySpecMocks(ArtifactTestUtils artifactTestUtils, INotificationData data,
256 IArtifactInfo artifact) throws IOException {
257 when(mockDistributionClient.download(artifact))
258 .thenReturn(createDistributionClientDownloadResult(DistributionActionResultEnum.SUCCESS, null,
259 artifactTestUtils.loadResource("models/named-query-wan-connector.xml")));
260 doNothing().when(mockNotificationPublisher).publishDownloadSuccess(mockDistributionClient, data, artifact);
264 public void downloadArtifacts_validCsarAndModelFiles()
265 throws IOException, BabelServiceClientException, BabelArtifactParsingException {
266 ArtifactTestUtils artifactTestUtils = new ArtifactTestUtils();
267 INotificationData data = getNotificationDataWithOneOfEach();
268 List<IArtifactInfo> artifacts = new ArrayList<>();
270 IArtifactInfo serviceArtifact = data.getServiceArtifacts().get(0);
271 IArtifactInfo modelSpecArtifact = data.getResources().get(1).getArtifacts().get(0);
273 artifacts.add(serviceArtifact);
274 artifacts.add(modelSpecArtifact);
276 setupValidDownloadCsarMocks(data, serviceArtifact, artifactTestUtils);
277 setupValidModelQuerySpecMocks(artifactTestUtils, data, modelSpecArtifact);
279 List<org.onap.aai.modelloader.entity.Artifact> modelFiles = new ArrayList<>();
280 List<org.onap.aai.modelloader.entity.Artifact> catalogFiles = new ArrayList<>();
281 assertThat(downloadManager.downloadArtifacts(data, artifacts, modelFiles, catalogFiles), is(true));
283 Mockito.verify(mockDistributionClient).download(serviceArtifact);
284 Mockito.verify(mockNotificationPublisher).publishDownloadSuccess(mockDistributionClient, data, serviceArtifact);
285 Mockito.verify(mockBabelClient).postArtifact(Matchers.any(), Matchers.anyString(), Matchers.anyString(),
286 Matchers.anyString());
287 Mockito.verify(mockBabelArtifactConverter).convertToModel(Matchers.any());
288 Mockito.verify(mockBabelArtifactConverter).convertToCatalog(Matchers.any());
290 Mockito.verify(mockDistributionClient).download(modelSpecArtifact);
291 Mockito.verify(mockNotificationPublisher).publishDownloadSuccess(mockDistributionClient, data,
294 Mockito.verifyNoMoreInteractions(mockBabelClient, mockBabelArtifactConverter);
298 @SuppressWarnings("unchecked")
299 public void activateCallback_toscaToModelConverterHasProcessToscaArtifactsException() throws Exception {
300 ArtifactTestUtils artifactTestUtils = new ArtifactTestUtils();
301 INotificationData data = getNotificationDataWithToscaCsarFile();
302 IArtifactInfo artifactInfo = data.getServiceArtifacts().get(0);
304 when(mockDistributionClient.download(artifactInfo))
305 .thenReturn(createDistributionClientDownloadResult(DistributionActionResultEnum.SUCCESS, null,
306 artifactTestUtils.loadResource("compressedArtifacts/service-VscpaasTest-csar.csar")));
307 when(mockBabelArtifactConverter.convertToModel(Mockito.anyList()))
308 .thenThrow(BabelArtifactParsingException.class);
309 doNothing().when(mockNotificationPublisher).publishDeployFailure(mockDistributionClient, data, artifactInfo);
310 when(mockBabelClient.postArtifact(Matchers.any(), Matchers.anyString(), Matchers.anyString(),
311 Matchers.anyString())).thenReturn(createBabelArtifacts());
313 List<Artifact> modelArtifacts = new ArrayList<>();
314 List<Artifact> catalogFiles = new ArrayList<>();
315 assertThat(downloadManager.downloadArtifacts(data, data.getServiceArtifacts(), modelArtifacts, catalogFiles),
317 assertThat(modelArtifacts, is(empty()));
318 assertThat(catalogFiles, is(empty()));
320 Mockito.verify(mockDistributionClient).download(artifactInfo);
321 Mockito.verify(mockNotificationPublisher).publishDeployFailure(mockDistributionClient, data, artifactInfo);
322 Mockito.verify(mockBabelClient).postArtifact(Matchers.any(), Matchers.anyString(), Matchers.anyString(),
323 Matchers.anyString());
324 Mockito.verify(mockBabelArtifactConverter).convertToModel(Matchers.any());
328 public void downloadArtifactsWithInvalidType()
329 throws IOException, BabelServiceClientException, BabelArtifactParsingException {
330 INotificationData data = getNotificationDataWithInvalidType();
331 IArtifactInfo artifact = data.getServiceArtifacts().get(0);
333 when(mockDistributionClient.download(artifact)).thenReturn(createDistributionClientDownloadResult(
334 DistributionActionResultEnum.SUCCESS, null, "This content does not matter.".getBytes()));
335 doNothing().when(mockNotificationPublisher).publishDownloadSuccess(mockDistributionClient, data, artifact);
337 assertThat(downloadManager.downloadArtifacts(data, data.getServiceArtifacts(), null, new ArrayList<>()),
339 Mockito.verify(mockDistributionClient).download(artifact);
340 Mockito.verify(mockNotificationPublisher).publishDownloadSuccess(mockDistributionClient, data, artifact);
341 Mockito.verify(mockNotificationPublisher).publishDeployFailure(mockDistributionClient, data, artifact);
342 Mockito.verifyZeroInteractions(mockBabelClient, mockBabelArtifactConverter);
345 private IDistributionClientDownloadResult createDistributionClientDownloadResult(
346 DistributionActionResultEnum status, String message, byte[] payload) {
347 DistributionClientDownloadResultImpl downloadResult = new DistributionClientDownloadResultImpl(status, message);
348 downloadResult.setArtifactPayload(payload);
349 return downloadResult;