Add to aai-common query builder get parent step

[aai/aai-common.git] / aai-core / src / test / java / org / onap / aai / query / builder / GremlinTraversalTest.java

diff --git a/aai-core/src/test/java/org/onap/aai/query/builder/GremlinTraversalTest.java b/aai-core/src/test/java/org/onap/aai/query/builder/GremlinTraversalTest.java
index 3632543..18f4714 100644 (file)
--- a/aai-core/src/test/java/org/onap/aai/query/builder/GremlinTraversalTest.java
+++ b/aai-core/src/test/java/org/onap/aai/query/builder/GremlinTraversalTest.java
@@ -19,6 +19,7 @@
  */
 package org.onap.aai.query.builder;
 
+import org.apache.tinkerpop.gremlin.process.traversal.Path;
 import org.apache.tinkerpop.gremlin.process.traversal.step.util.Tree;
 import org.apache.tinkerpop.gremlin.structure.Edge;
 import org.apache.tinkerpop.gremlin.structure.Vertex;
@@ -55,5 +56,15 @@ public class GremlinTraversalTest extends QueryBuilderTestAbstraction {
                return new GremlinTraversal<>(loader, g, testEdgeRules);
        }
 
+       @Override
+       protected QueryBuilder<Path> getNewPathTraversalWithTestEdgeRules(Vertex v) {
+               return new GremlinTraversal<>(loader, g, v, testEdgeRules);
+       }
+
+       @Override
+       protected QueryBuilder<Path> getNewPathTraversalWithTestEdgeRules() {
+               return new GremlinTraversal<>(loader, g, testEdgeRules);
+       }
+
                
 }