diff options
Diffstat (limited to 'asdc-tests/src/main/java/org/openecomp/sdc/ci/tests/migration/v1707/postupgrade/AttKeyPropertiesRenameTest.java')
-rw-r--r-- | asdc-tests/src/main/java/org/openecomp/sdc/ci/tests/migration/v1707/postupgrade/AttKeyPropertiesRenameTest.java | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/asdc-tests/src/main/java/org/openecomp/sdc/ci/tests/migration/v1707/postupgrade/AttKeyPropertiesRenameTest.java b/asdc-tests/src/main/java/org/openecomp/sdc/ci/tests/migration/v1707/postupgrade/AttKeyPropertiesRenameTest.java new file mode 100644 index 0000000000..95fc068dd3 --- /dev/null +++ b/asdc-tests/src/main/java/org/openecomp/sdc/ci/tests/migration/v1707/postupgrade/AttKeyPropertiesRenameTest.java @@ -0,0 +1,71 @@ +package org.openecomp.sdc.ci.tests.migration.v1707.postupgrade; + +import com.thinkaurelius.titan.core.TitanVertex; +import org.junit.Rule; +import org.junit.rules.TestName; +import org.openecomp.sdc.ci.tests.api.ComponentBaseTest; +import org.openecomp.sdc.ci.tests.migration.v1707.CommonMigrationUtils; +import org.openecomp.sdc.ci.tests.utils.graph.GraphFileUtils; +import org.testng.annotations.Test; + +import java.io.IOException; +import java.util.ArrayList; +import java.util.Collections; +import java.util.List; + +import static org.testng.Assert.*; + +public class AttKeyPropertiesRenameTest extends ComponentBaseTest { + + @Rule + public static TestName name = new TestName(); + + + public AttKeyPropertiesRenameTest() { + super(name, AttKeyPropertiesRenameTest.class.getName()); + } + + @Test + public void verifyAttPropertyKeys() throws Exception { + initGraph(); + CommonMigrationUtils.assertKeyNotExist(titanGraph, "attContact"); + CommonMigrationUtils.assertKeyNotExist(titanGraph, "attCreator"); + CommonMigrationUtils.assertKeyNotExist(titanGraph, "attuid"); + CommonMigrationUtils.assertKeyNotExist(titanGraph, "pmatt"); + + CommonMigrationUtils.assertKeyExists(titanGraph, "userId"); + CommonMigrationUtils.assertKeyExists(titanGraph, "projectCode"); + CommonMigrationUtils.assertKeyExists(titanGraph, "contactId"); + CommonMigrationUtils.assertKeyExists(titanGraph, "creatorId"); + + verifyPropertyKeysVerticesSameAsPreUpgrade("attuid", "userId"); + verifyPropertyKeysVerticesSameAsPreUpgrade("pmatt", "projectCode"); + verifyPropertyKeysVerticesSameAsPreUpgrade("attContact", "contactId"); + verifyPropertyKeysVerticesSameAsPreUpgrade("attCreator", "creatorId"); + + } + + private void assertKeyNotExist(String key) { + assertNotNull(titanGraph.getPropertyKey(key)); + } + + private void assertKeyExists(String key) { + assertNull(titanGraph.getPropertyKey(key)); + } + + private void verifyPropertyKeysVerticesSameAsPreUpgrade(String oldPropertyKEyName, String newPropertyKeyName) throws IOException { + List<String> verticesIdsFromGraph = getVerticesIdsFromGRaph(newPropertyKeyName); + List<String> verticesIdsFromFile = GraphFileUtils.getVerticesIdsFromFile(oldPropertyKEyName); + Collections.sort(verticesIdsFromFile); + Collections.sort(verticesIdsFromGraph); + assertEquals(verticesIdsFromFile, verticesIdsFromGraph); + } + + private List<String> getVerticesIdsFromGRaph(String newPropertyKeyName) { + Iterable<TitanVertex> vertices = titanGraph.query().has(newPropertyKeyName).vertices(); + assertTrue(vertices.iterator().hasNext()); + List<String> verticesIdsFromGraph = new ArrayList<>(); + vertices.forEach(vertex -> verticesIdsFromGraph.add(vertex.id().toString())); + return verticesIdsFromGraph; + } +} |