diff options
Diffstat (limited to 'asdctool/src/test/java/org/openecomp/sdc/asdctool/migration/tasks/mig1806/SDCInstancesMigrationTest.java')
-rw-r--r-- | asdctool/src/test/java/org/openecomp/sdc/asdctool/migration/tasks/mig1806/SDCInstancesMigrationTest.java | 69 |
1 files changed, 69 insertions, 0 deletions
diff --git a/asdctool/src/test/java/org/openecomp/sdc/asdctool/migration/tasks/mig1806/SDCInstancesMigrationTest.java b/asdctool/src/test/java/org/openecomp/sdc/asdctool/migration/tasks/mig1806/SDCInstancesMigrationTest.java new file mode 100644 index 0000000000..d040d66496 --- /dev/null +++ b/asdctool/src/test/java/org/openecomp/sdc/asdctool/migration/tasks/mig1806/SDCInstancesMigrationTest.java @@ -0,0 +1,69 @@ +package org.openecomp.sdc.asdctool.migration.tasks.mig1806; + +import fj.data.Either; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.junit.MockitoJUnitRunner; +import org.openecomp.sdc.asdctool.migration.core.task.MigrationResult; +import org.openecomp.sdc.asdctool.migration.core.task.MigrationResult.MigrationStatus; +import org.openecomp.sdc.be.dao.jsongraph.GraphVertex; +import org.openecomp.sdc.be.dao.jsongraph.TitanDao; +import org.openecomp.sdc.be.dao.titan.TitanOperationStatus; +import org.openecomp.sdc.be.datatypes.elements.ComponentInstanceDataDefinition; +import org.openecomp.sdc.be.datatypes.elements.CompositionDataDefinition; +import org.openecomp.sdc.be.datatypes.enums.ComponentTypeEnum; +import org.openecomp.sdc.be.model.jsontitan.enums.JsonConstantKeysEnum; +import org.openecomp.sdc.be.model.jsontitan.operations.NodeTemplateOperation; +import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus; + +import java.util.ArrayList; +import java.util.HashMap; +import java.util.List; +import java.util.Map; + +import static org.junit.Assert.assertEquals; +import static org.mockito.ArgumentMatchers.any; +import static org.mockito.Mockito.when; + +@RunWith(MockitoJUnitRunner.class) +public class SDCInstancesMigrationTest{ + @Mock + private TitanDao titanDao; + @Mock + private NodeTemplateOperation nodeTemplateOperation; + + + @Test + public void testFailedMigration(){ + SDCInstancesMigration instancesMigration = new SDCInstancesMigration(titanDao, nodeTemplateOperation); + when(titanDao.getByCriteria(any(), any(), any(), any() )).thenReturn(Either.right(TitanOperationStatus.GENERAL_ERROR)); + + MigrationResult migrate = instancesMigration.migrate(); + MigrationStatus migrationStatus = migrate.getMigrationStatus(); + assertEquals(MigrationStatus.FAILED, migrationStatus); + } + @Test + public void testSuccessMigration(){ + SDCInstancesMigration instancesMigration = new SDCInstancesMigration(titanDao, nodeTemplateOperation); + List<GraphVertex> list = new ArrayList<>(); + GraphVertex vertexOrig = new GraphVertex(); + Map<String, CompositionDataDefinition> jsonComposition = new HashMap<>(); + CompositionDataDefinition composition = new CompositionDataDefinition(); + Map<String, ComponentInstanceDataDefinition> componentInstances = new HashMap<>(); + ComponentInstanceDataDefinition instance = new ComponentInstanceDataDefinition(); + componentInstances.put("instanceId", instance); + composition.setComponentInstances(componentInstances); + jsonComposition.put(JsonConstantKeysEnum.COMPOSITION.getValue(), composition); + vertexOrig.setJson(jsonComposition); + vertexOrig.setType(ComponentTypeEnum.SERVICE); + list.add(vertexOrig); + + when(titanDao.getByCriteria(any(), any(), any(), any() )).thenReturn(Either.left(list)); + when(nodeTemplateOperation.createInstanceEdge(vertexOrig, instance)).thenReturn(StorageOperationStatus.OK); + + MigrationResult migrate = instancesMigration.migrate(); + MigrationStatus migrationStatus = migrate.getMigrationStatus(); + assertEquals(MigrationStatus.COMPLETED, migrationStatus); + } +} |