package com.tinkerpop.gremlin.process.graph.step.filter;
import com.tinkerpop.gremlin.LoadGraphWith;
import com.tinkerpop.gremlin.process.AbstractGremlinProcessTest;
import com.tinkerpop.gremlin.process.Path;
import com.tinkerpop.gremlin.process.Traversal;
import com.tinkerpop.gremlin.structure.Vertex;
import org.junit.Test;
import static com.tinkerpop.gremlin.LoadGraphWith.GraphData.MODERN;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
/**
* @author Marko A. Rodriguez (http://markorodriguez.com)
*/
public abstract class CyclicPathTest extends AbstractGremlinProcessTest {
public abstract Traversal<Vertex, Vertex> get_g_v1_outXcreatedX_inXcreatedX_cyclicPath(final Object v1);
public abstract Traversal<Vertex, Path> get_g_v1_outXcreatedX_inXcreatedX_cyclicPath_path(final Object v1);
@Test
@LoadGraphWith(MODERN)
public void g_v1_outXcreatedX_inXcreatedX_cyclicPath() {
final Traversal<Vertex, Vertex> traversal = get_g_v1_outXcreatedX_inXcreatedX_cyclicPath(convertToVertexId("marko"));
printTraversalForm(traversal);
int counter = 0;
while (traversal.hasNext()) {
counter++;
Vertex vertex = traversal.next();
assertEquals("marko", vertex.<String>value("name"));
}
assertEquals(1, counter);
assertFalse(traversal.hasNext());
}
@Test
@LoadGraphWith(MODERN)
public void g_v1_outXcreatedX_inXcreatedX_cyclicPath_path() {
final Traversal<Vertex, Path> traversal = get_g_v1_outXcreatedX_inXcreatedX_cyclicPath_path(convertToVertexId("marko"));
printTraversalForm(traversal);
int counter = 0;
while (traversal.hasNext()) {
counter++;
Path path = traversal.next();
assertFalse(path.isSimple());
}
assertEquals(1, counter);
assertFalse(traversal.hasNext());
}
public static class StandardTest extends CyclicPathTest {
@Override
public Traversal<Vertex, Vertex> get_g_v1_outXcreatedX_inXcreatedX_cyclicPath(final Object v1Id) {
return g.v(v1Id).out("created").in("created").cyclicPath();
}
@Override
public Traversal<Vertex, Path> get_g_v1_outXcreatedX_inXcreatedX_cyclicPath_path(final Object v1Id) {
return g.v(v1Id).out("created").in("created").cyclicPath().path();
}
}
public static class ComputerTest extends CyclicPathTest {
public ComputerTest() {
requiresGraphComputer = true;
}
@Override
public Traversal<Vertex, Vertex> get_g_v1_outXcreatedX_inXcreatedX_cyclicPath(final Object v1Id) {
return g.v(v1Id).out("created").in("created").cyclicPath().submit(g.compute());
}
@Override
public Traversal<Vertex, Path> get_g_v1_outXcreatedX_inXcreatedX_cyclicPath_path(final Object v1Id) {
return g.v(v1Id).out("created").in("created").cyclicPath().path().submit(g.compute());
}
}
}