Add to aai-common query builder get parent step

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

diff --git a/aai-core/src/test/java/org/onap/aai/query/builder/TraversalQueryTest.java b/aai-core/src/test/java/org/onap/aai/query/builder/TraversalQueryTest.java
index 1eeca6f..b38d13d 100644 (file)
--- a/aai-core/src/test/java/org/onap/aai/query/builder/TraversalQueryTest.java
+++ b/aai-core/src/test/java/org/onap/aai/query/builder/TraversalQueryTest.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.dsl.graph.GraphTraversal;
 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.__;
 import org.apache.tinkerpop.gremlin.process.traversal.step.util.Tree;
@@ -74,6 +75,16 @@ public class TraversalQueryTest extends QueryBuilderTestAbstraction {
        protected QueryBuilder<Tree> getNewTreeTraversalWithTestEdgeRules() {
                return new TraversalQuery<>(loader, g, testEdgeRules);
        }
+
+       @Override
+       protected QueryBuilder<Path> getNewPathTraversalWithTestEdgeRules(Vertex v) {
+               return new TraversalQuery<>(loader, g, v, testEdgeRules);
+       }
+
+       @Override
+       protected QueryBuilder<Path> getNewPathTraversalWithTestEdgeRules() {
+               return new TraversalQuery<>(loader, g, testEdgeRules);
+       }
        
        @Test
        public void unionQuery() {