X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Forg%2Fonap%2Faai%2Fmodelloader%2Fnotification%2FArtifactDeploymentManagerMockTest.java;fp=src%2Ftest%2Fjava%2Forg%2Fonap%2Faai%2Fmodelloader%2Fnotification%2FArtifactDeploymentManagerMockTest.java;h=396cc03b1962cea537e8d2e593c0023a317ed8cc;hb=7b1f813441f94261f43ec4f5bb0944ad2570fbdf;hp=0000000000000000000000000000000000000000;hpb=4dd316529148d07059d844197cdb676806bdc0c6;p=aai%2Fmodel-loader.git diff --git a/src/test/java/org/onap/aai/modelloader/notification/ArtifactDeploymentManagerMockTest.java b/src/test/java/org/onap/aai/modelloader/notification/ArtifactDeploymentManagerMockTest.java new file mode 100644 index 0000000..396cc03 --- /dev/null +++ b/src/test/java/org/onap/aai/modelloader/notification/ArtifactDeploymentManagerMockTest.java @@ -0,0 +1,274 @@ +/** + * ============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.junit.Assert.assertTrue; +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.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.fixture.NotificationDataFixtureBuilder; +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.reflect.Whitebox; + +/** + * Tests {@link ArtifactDeploymentManager } + */ +public class ArtifactDeploymentManagerMockTest { + + 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; + } +/* + @Test + public void deploy_csarDeploymentsFailed() throws IOException, BabelArtifactParsingException { + ArtifactTestUtils artifactTestUtils = new ArtifactTestUtils(); + INotificationData data = NotificationDataFixtureBuilder.getNotificationDataWithToscaCsarFile(); + byte[] xml = artifactTestUtils.loadResource("convertedYmls/AAI-SCP-Test-VSP-resource-1.0.xml"); + List toscaArtifacts = setupTest(xml, data); + List modelArtifacts = new BabelArtifactConverter().convertToModel(toscaArtifacts); + + PowerMockito.when( + mockModelArtifactHandler.pushArtifacts(eq(modelArtifacts), 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(), modelArtifacts, new ArrayList<>())); + + Mockito.verify(mockModelArtifactHandler).pushArtifacts(eq(modelArtifacts), eq(data.getDistributionID()), any(), + any()); + Mockito.verify(mockVnfCatalogArtifactHandler, Mockito.never()).pushArtifacts(eq(modelArtifacts), + eq(data.getDistributionID()), any(), any()); + Mockito.verify(mockModelArtifactHandler).rollback(eq(new ArrayList()), eq(data.getDistributionID()), + any()); + Mockito.verify(mockVnfCatalogArtifactHandler, Mockito.never()).rollback(eq(new ArrayList()), + eq(data.getDistributionID()), any()); + Mockito.verify(mockNotificationPublisher).publishDeployFailure(mockDistributionClient, data, + data.getServiceArtifacts().get(0)); + } +*/ + 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)); + } + +/* + @Test + public void deploy_bothDeploymentsFailed() + throws IOException, BabelArtifactParsingException, InvalidArchiveException { + doFailedCombinedTests(false, false); + } +*/ +/* + @Test + public void deploy_modelsFailedCatalogsOK() + throws IOException, BabelArtifactParsingException, InvalidArchiveException { + doFailedCombinedTests(false, true); + } +*/ +/* + @Test + public void deploy_catalogsFailedModelsOK() + throws IOException, BabelArtifactParsingException, InvalidArchiveException { + doFailedCombinedTests(true, false); + } +*/ + 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()); + } + } + } + +/* + @Test + public void deploy_bothOK() 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(true); + PowerMockito.when( + mockModelArtifactHandler.pushArtifacts(eq(modelArtifacts), eq(data.getDistributionID()), any(), any())) + .thenReturn(true); + PowerMockito.doNothing().when(mockNotificationPublisher).publishDeploySuccess(mockDistributionClient, data, + data.getServiceArtifacts().get(0)); + + assertTrue("This should have returned true", + manager.deploy(data, data.getServiceArtifacts(), modelArtifacts, catalogFiles)); + + Mockito.verify(mockVnfCatalogArtifactHandler).pushArtifacts(eq(catalogFiles), eq(data.getDistributionID()), + any(), any()); + Mockito.verify(mockModelArtifactHandler).pushArtifacts(eq(modelArtifacts), eq(data.getDistributionID()), any(), + any()); + 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()); + } +*/ +}