X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Forg%2Fonap%2Faai%2Fmodelloader%2Fnotification%2FArtifactDeploymentManagerTest.java;fp=src%2Ftest%2Fjava%2Forg%2Fonap%2Faai%2Fmodelloader%2Fnotification%2FArtifactDeploymentManagerTest.java;h=9fc076019d52ffb5d6b246831b9b0155a9a1ea8b;hb=7b1f813441f94261f43ec4f5bb0944ad2570fbdf;hp=0000000000000000000000000000000000000000;hpb=4dd316529148d07059d844197cdb676806bdc0c6;p=aai%2Fmodel-loader.git diff --git a/src/test/java/org/onap/aai/modelloader/notification/ArtifactDeploymentManagerTest.java b/src/test/java/org/onap/aai/modelloader/notification/ArtifactDeploymentManagerTest.java new file mode 100644 index 0000000..9fc0760 --- /dev/null +++ b/src/test/java/org/onap/aai/modelloader/notification/ArtifactDeploymentManagerTest.java @@ -0,0 +1,194 @@ +/** + * ============LICENSE_START========================================== + * org.onap.aai + * =================================================================== + * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved. + * Copyright © 2017-2018 Amdocs + * =================================================================== + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + * ============LICENSE_END============================================ + */ +package org.onap.aai.modelloader.notification; + +import static org.junit.Assert.assertFalse; +import static org.mockito.Matchers.any; +import static org.mockito.Matchers.eq; +import static org.onap.aai.modelloader.fixture.NotificationDataFixtureBuilder.getNotificationDataWithCatalogFile; +import static org.onap.aai.modelloader.fixture.NotificationDataFixtureBuilder.getNotificationDataWithOneOfEach; + +import java.io.IOException; +import java.util.ArrayList; +import java.util.List; +import java.util.Properties; +import org.junit.After; +import org.junit.Before; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mockito; +import org.onap.aai.babel.service.data.BabelArtifact; +import org.onap.aai.modelloader.config.ModelLoaderConfig; +import org.onap.aai.modelloader.entity.Artifact; +import org.onap.aai.modelloader.entity.catalog.VnfCatalogArtifact; +import org.onap.aai.modelloader.entity.catalog.VnfCatalogArtifactHandler; +import org.onap.aai.modelloader.entity.model.BabelArtifactParsingException; +import org.onap.aai.modelloader.entity.model.ModelArtifactHandler; +import org.onap.aai.modelloader.extraction.InvalidArchiveException; +import org.onap.aai.modelloader.util.ArtifactTestUtils; +import org.openecomp.sdc.api.IDistributionClient; +import org.openecomp.sdc.api.notification.INotificationData; +import org.powermock.api.mockito.PowerMockito; +import org.powermock.modules.junit4.PowerMockRunner; +import org.powermock.reflect.Whitebox; + +/** + * Tests {@link ArtifactDeploymentManager } + */ +@RunWith(PowerMockRunner.class) +public class ArtifactDeploymentManagerTest { + + private static final String CONFIG_FILE = "model-loader.properties"; + private static final String SHOULD_HAVE_RETURNED_FALSE = "This should have returned false"; + + private Properties configProperties; + private ArtifactDeploymentManager manager; + + private IDistributionClient mockDistributionClient; + private ModelArtifactHandler mockModelArtifactHandler; + private NotificationPublisher mockNotificationPublisher; + private VnfCatalogArtifactHandler mockVnfCatalogArtifactHandler; + + @Before + public void setup() throws IOException { + configProperties = new Properties(); + configProperties.load(this.getClass().getClassLoader().getResourceAsStream(CONFIG_FILE)); + ModelLoaderConfig config = new ModelLoaderConfig(configProperties, null); + + mockDistributionClient = PowerMockito.mock(IDistributionClient.class); + mockModelArtifactHandler = PowerMockito.mock(ModelArtifactHandler.class); + mockNotificationPublisher = PowerMockito.mock(NotificationPublisher.class); + mockVnfCatalogArtifactHandler = PowerMockito.mock(VnfCatalogArtifactHandler.class); + + manager = new ArtifactDeploymentManager(mockDistributionClient, config); + + Whitebox.setInternalState(manager, mockModelArtifactHandler); + Whitebox.setInternalState(manager, mockNotificationPublisher); + Whitebox.setInternalState(manager, mockVnfCatalogArtifactHandler); + } + + @After + public void tearDown() { + configProperties = null; + mockDistributionClient = null; + mockModelArtifactHandler = null; + mockNotificationPublisher = null; + mockVnfCatalogArtifactHandler = null; + manager = null; + } + + + private List setupTest(byte[] xml, INotificationData data) throws IOException { + List toscaArtifacts = new ArrayList<>(); + org.openecomp.sdc.api.notification.IArtifactInfo artifactInfo = data.getServiceArtifacts().get(0); + + BabelArtifact xmlArtifact = + new BabelArtifact(artifactInfo.getArtifactName(), BabelArtifact.ArtifactType.MODEL, new String(xml)); + toscaArtifacts.add(xmlArtifact); + + return toscaArtifacts; + } + + @Test + public void deploy_catalogDeploymentsFailed() + throws IOException, BabelArtifactParsingException, InvalidArchiveException { + INotificationData data = getNotificationDataWithCatalogFile(); + + List catalogFiles = new ArrayList<>(); + catalogFiles.add(new VnfCatalogArtifact("Some catalog content")); + + PowerMockito.when(mockModelArtifactHandler.pushArtifacts(any(), any(), any(), any())).thenReturn(true); + PowerMockito.when(mockVnfCatalogArtifactHandler.pushArtifacts(eq(catalogFiles), eq(data.getDistributionID()), + any(), any())).thenReturn(false); + PowerMockito.doNothing().when(mockNotificationPublisher).publishDeployFailure(mockDistributionClient, data, + data.getServiceArtifacts().get(0)); + + assertFalse(SHOULD_HAVE_RETURNED_FALSE, + manager.deploy(data, data.getServiceArtifacts(), new ArrayList<>(), catalogFiles)); + + Mockito.verify(mockModelArtifactHandler).pushArtifacts(eq(new ArrayList()), + eq(data.getDistributionID()), any(), any()); + Mockito.verify(mockVnfCatalogArtifactHandler).pushArtifacts(eq(catalogFiles), eq(data.getDistributionID()), + any(), any()); + Mockito.verify(mockModelArtifactHandler).rollback(eq(new ArrayList()), eq(data.getDistributionID()), + any()); + Mockito.verify(mockVnfCatalogArtifactHandler).rollback(eq(new ArrayList()), + eq(data.getDistributionID()), any()); + Mockito.verify(mockNotificationPublisher).publishDeployFailure(mockDistributionClient, data, + data.getServiceArtifacts().get(0)); + } + + + private void doFailedCombinedTests(boolean modelsOK, boolean catalogsOK) + throws IOException, BabelArtifactParsingException, InvalidArchiveException { + INotificationData data = getNotificationDataWithOneOfEach(); + ArtifactTestUtils artifactTestUtils = new ArtifactTestUtils(); + byte[] xml = artifactTestUtils.loadResource("convertedYmls/AAI-SCP-Test-VSP-resource-1.0.xml"); + List toscaArtifacts = setupTest(xml, data); + List modelArtifacts = new BabelArtifactConverter().convertToModel(toscaArtifacts); + + List catalogFiles = new ArrayList<>(); + catalogFiles.add(new VnfCatalogArtifact("Some catalog content")); + + PowerMockito.when(mockVnfCatalogArtifactHandler.pushArtifacts(eq(catalogFiles), eq(data.getDistributionID()), + any(), any())).thenReturn(catalogsOK); + PowerMockito.when( + mockModelArtifactHandler.pushArtifacts(eq(modelArtifacts), eq(data.getDistributionID()), any(), any())) + .thenReturn(modelsOK); + + PowerMockito.doNothing().when(mockNotificationPublisher).publishDeploySuccess(mockDistributionClient, data, + data.getServiceArtifacts().get(0)); + PowerMockito.doNothing().when(mockNotificationPublisher).publishDeployFailure(mockDistributionClient, data, + data.getServiceArtifacts().get(0)); + + assertFalse(SHOULD_HAVE_RETURNED_FALSE, + manager.deploy(data, data.getServiceArtifacts(), modelArtifacts, catalogFiles)); + + // Catalog artifacts are only pushed if models are successful. + Mockito.verify(mockModelArtifactHandler).pushArtifacts(eq(modelArtifacts), eq(data.getDistributionID()), any(), + any()); + if (modelsOK) { + Mockito.verify(mockVnfCatalogArtifactHandler).pushArtifacts(eq(catalogFiles), eq(data.getDistributionID()), + any(), any()); + } + + if (modelsOK && catalogsOK) { + Mockito.verify(mockNotificationPublisher).publishDeploySuccess(mockDistributionClient, data, + data.getServiceArtifacts().get(0)); + Mockito.verify(mockModelArtifactHandler, Mockito.never()).rollback(any(), any(), any()); + Mockito.verify(mockVnfCatalogArtifactHandler, Mockito.never()).rollback(any(), any(), any()); + } else { + Mockito.verify(mockNotificationPublisher).publishDeployFailure(mockDistributionClient, data, + data.getServiceArtifacts().get(0)); + if (modelsOK) { + Mockito.verify(mockModelArtifactHandler).rollback(eq(new ArrayList()), + eq(data.getDistributionID()), any()); + Mockito.verify(mockVnfCatalogArtifactHandler).rollback(eq(new ArrayList()), + eq(data.getDistributionID()), any()); + } else { + Mockito.verify(mockModelArtifactHandler).rollback(eq(new ArrayList()), + eq(data.getDistributionID()), any()); + Mockito.verify(mockVnfCatalogArtifactHandler, Mockito.never()).rollback(any(), any(), any()); + } + } + } + +}