1 files changed, 39 insertions, 0 deletions
diff --git a/asdc-tests/src/main/java/org/openecomp/sdc/ci/tests/utils/graph/GraphFileUtils.java b/asdc-tests/src/main/java/org/openecomp/sdc/ci/tests/utils/graph/GraphFileUtils.java
new file mode 100644
index 0000000000..2fa5807aff
--- /dev/null
+++ b/asdc-tests/src/main/java/org/openecomp/sdc/ci/tests/utils/graph/GraphFileUtils.java
@@ -0,0 +1,39 @@
+package org.openecomp.sdc.ci.tests.utils.graph;
+
+import com.thinkaurelius.titan.core.TitanVertex;
+import org.apache.commons.io.FileUtils;
+import org.openecomp.sdc.be.dao.neo4j.GraphPropertiesDictionary;
+
+import java.io.IOException;
+import java.nio.charset.Charset;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.ArrayList;
+import java.util.List;
+
+public class GraphFileUtils {
+
+    public static final String TEMP_FILES_PATH = "src/main/resources/ci/tempFiles/%s.txt";
+
+    public static void writeVerticesUIDToFile(String fileName, Iterable<TitanVertex> vertices) throws IOException {
+        Path path = Paths.get(String.format(TEMP_FILES_PATH, fileName));
+        Files.deleteIfExists(path);
+        Path file = Files.createFile(path);
+        final String newLine = System.getProperty("line.separator");
+        for (TitanVertex vertex : vertices) {
+            FileUtils.writeStringToFile(file.toFile(), String.valueOf(vertex.id()) + newLine, Charset.defaultCharset(), true);
+        }
+    }
+
+    public static List<String> getVerticesIdsFromFile(String fileName) throws IOException {
+        List<String> verticesUids = new ArrayList<>();
+        Files.lines(Paths.get(String.format(TEMP_FILES_PATH, fileName))).forEach(verticesUids::add);
+        return verticesUids;
+    }
+
+    private static String getUid(TitanVertex titanVertex) {
+        return (String )titanVertex.value(GraphPropertiesDictionary.UNIQUE_ID.getProperty());
+    }
+
+}