org.umlg.sqlg.test.gremlincompile.TestGremlinCompileGraphStep Maven / Gradle / Ivy
package org.umlg.sqlg.test.gremlincompile;
import org.apache.tinkerpop.gremlin.AbstractGremlinTest;
import org.apache.tinkerpop.gremlin.process.traversal.P;
import org.apache.tinkerpop.gremlin.process.traversal.Traversal;
import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal;
import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource;
import org.apache.tinkerpop.gremlin.structure.Edge;
import org.apache.tinkerpop.gremlin.structure.Graph;
import org.apache.tinkerpop.gremlin.structure.T;
import org.apache.tinkerpop.gremlin.structure.Vertex;
import org.apache.tinkerpop.gremlin.structure.io.GraphReader;
import org.apache.tinkerpop.gremlin.structure.io.gryo.GryoIo;
import org.apache.tinkerpop.gremlin.structure.io.gryo.GryoReader;
import org.apache.tinkerpop.gremlin.util.iterator.IteratorUtils;
import org.junit.Assert;
import org.junit.Test;
import org.umlg.sqlg.test.BaseTest;
import java.io.IOException;
import java.io.InputStream;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import static org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.__.inE;
/**
* Created by pieter on 2015/07/19.
*/
public class TestGremlinCompileGraphStep extends BaseTest {
@Test
public void gg_V_asXaX_outXcreatedX_inXcreatedX_whereXneqXaXX_asXbX_selectXa_bX_addInEXa_codeveloper_b_year_2009X() throws IOException {
Graph g = this.sqlgGraph;
final GraphReader reader = GryoReader.build()
.mapper(g.io(GryoIo.build()).mapper().create())
.create();
try (final InputStream stream = AbstractGremlinTest.class.getResourceAsStream("/tinkerpop-modern.kryo")) {
reader.readGraph(stream, g);
}
assertModernGraph(g, true, false);
List> vertices = this.sqlgGraph.traversal()
.V().as("a")
.out("created")
.in("created")
.where(P.neq("a")).as("b")
.select("a", "b").toList();
Assert.assertEquals(6, vertices.size());
final GraphTraversal traversal = this.sqlgGraph.traversal()
.V().as("a")
.out("created")
.in("created")
.where(P.neq("a")).as("b")
.select("a", "b")
.addInE("a", "co-developer", "b", "year", 2009);
int count = 0;
while (traversal.hasNext()) {
final Edge edge = traversal.next();
Assert.assertEquals("co-developer", edge.label());
Assert.assertEquals(2009, (int) edge.value("year"));
Assert.assertEquals(1, IteratorUtils.count(edge.properties()));
Assert.assertEquals("person", edge.inVertex().label());
Assert.assertEquals("person", edge.outVertex().label());
Assert.assertFalse(edge.inVertex().value("name").equals("vadas"));
Assert.assertFalse(edge.outVertex().value("name").equals("vadas"));
Assert.assertFalse(edge.inVertex().equals(edge.outVertex()));
count++;
}
Assert.assertEquals(6, count);
Assert.assertEquals(12, IteratorUtils.count(this.sqlgGraph.edges()));
Assert.assertEquals(6, IteratorUtils.count(this.sqlgGraph.vertices()));
}
@Test
public void g_V_asXaX_outXcreatedX_inXcreatedX_whereXneqXaXX_asXbX_selectXa_bX_addInEXa_codeveloper_b_year_2009X() throws IOException {
Graph g = this.sqlgGraph;
final GraphReader reader = GryoReader.build()
.mapper(g.io(GryoIo.build()).mapper().create())
.create();
try (final InputStream stream = AbstractGremlinTest.class.getResourceAsStream("/tinkerpop-modern.kryo")) {
reader.readGraph(stream, g);
}
assertModernGraph(g, true, false);
final Traversal traversal = this.sqlgGraph.traversal()
.V().as("a")
.out("created")
.in("created")
.where(P.neq("a")).as("b")
.select("a", "b")
.addInE("a", "co-developer", "b", "year", 2009);
printTraversalForm(traversal);
int count = 0;
while (traversal.hasNext()) {
final Edge edge = traversal.next();
Assert.assertEquals("co-developer", edge.label());
Assert.assertEquals(2009, (int) edge.value("year"));
Assert.assertEquals(1, IteratorUtils.count(edge.properties()));
Assert.assertEquals("person", edge.inVertex().label());
Assert.assertEquals("person", edge.outVertex().label());
Assert.assertFalse(edge.inVertex().value("name").equals("vadas"));
Assert.assertFalse(edge.outVertex().value("name").equals("vadas"));
Assert.assertFalse(edge.inVertex().equals(edge.outVertex()));
count++;
}
Assert.assertEquals(6, count);
Assert.assertEquals(12, IteratorUtils.count(this.sqlgGraph.edges()));
Assert.assertEquals(6, IteratorUtils.count(this.sqlgGraph.vertices()));
}
@Test
public void g_VX1X_outEXknowsX_hasXweight_1X_asXhereX_inV_hasXname_joshX_selectXhereX() throws IOException {
Graph g = this.sqlgGraph;
final GraphReader reader = GryoReader.build()
.mapper(g.io(GryoIo.build()).mapper().create())
.create();
try (final InputStream stream = AbstractGremlinTest.class.getResourceAsStream("/tinkerpop-modern.kryo")) {
reader.readGraph(stream, g);
}
assertModernGraph(g, true, false);
Object v1Id = convertToVertexId("marko");
final List> traversals = Arrays.asList(
g.traversal().V(v1Id).outE("knows").as("here").has("weight", 1.0d).as("fake").inV().has("name", "josh").select("here")
);
traversals.forEach(traversal -> {
printTraversalForm(traversal);
Assert.assertTrue(traversal.hasNext());
Assert.assertTrue(traversal.hasNext());
final Edge edge = traversal.next();
Assert.assertEquals("knows", edge.label());
Assert.assertEquals(1.0d, edge.value("weight"), 0.00001d);
Assert.assertFalse(traversal.hasNext());
Assert.assertFalse(traversal.hasNext());
});
}
@Test
public void g_V_localXinEXknowsX_limitX2XX_outV_name() throws IOException {
Graph g = this.sqlgGraph;
final GraphReader reader = GryoReader.build()
.mapper(g.io(GryoIo.build()).mapper().create())
.create();
try (final InputStream stream = AbstractGremlinTest.class.getResourceAsStream("/tinkerpop-modern.kryo")) {
reader.readGraph(stream, g);
}
assertModernGraph(g, true, false);
final Traversal traversal = g.traversal().V().local(inE("knows").limit(2)).outV().values("name");
printTraversalForm(traversal);
int counter = 0;
while (traversal.hasNext()) {
counter++;
Assert.assertEquals(traversal.next(), "marko");
}
Assert.assertFalse(traversal.hasNext());
Assert.assertEquals(2, counter);
}
@Test
public void testCompileGraphStep() {
Vertex a1 = this.sqlgGraph.addVertex(T.label, "A");
Vertex a2 = this.sqlgGraph.addVertex(T.label, "A");
Vertex a3 = this.sqlgGraph.addVertex(T.label, "A");
Vertex b11 = this.sqlgGraph.addVertex(T.label, "B");
Vertex b12 = this.sqlgGraph.addVertex(T.label, "B");
Vertex b13 = this.sqlgGraph.addVertex(T.label, "B");
Vertex b21 = this.sqlgGraph.addVertex(T.label, "B");
Vertex b22 = this.sqlgGraph.addVertex(T.label, "B");
Vertex b23 = this.sqlgGraph.addVertex(T.label, "B");
Vertex b31 = this.sqlgGraph.addVertex(T.label, "B");
Vertex b32 = this.sqlgGraph.addVertex(T.label, "B");
Vertex b33 = this.sqlgGraph.addVertex(T.label, "B");
a1.addEdge("ab", b11);
a1.addEdge("ab", b12);
a1.addEdge("ab", b13);
a2.addEdge("ab", b21);
a2.addEdge("ab", b22);
a2.addEdge("ab", b23);
a3.addEdge("ab", b31);
a3.addEdge("ab", b32);
a3.addEdge("ab", b33);
this.sqlgGraph.tx().commit();
GraphTraversalSource g = this.sqlgGraph.traversal();
List vertexes = g.V().hasLabel("A").out("ab").toList();
Assert.assertEquals(9, vertexes.size());
Assert.assertEquals(9, vertexes.size());
Assert.assertTrue(vertexes.containsAll(Arrays.asList(b11, b12, b13, b21, b22, b23, b31, b32, b33)));
vertexes = g.V().has(T.label, "A").out("ab").toList();
Assert.assertEquals(9, vertexes.size());
Assert.assertEquals(9, vertexes.size());
Assert.assertTrue(vertexes.containsAll(Arrays.asList(b11, b12, b13, b21, b22, b23, b31, b32, b33)));
vertexes = g.V().has(T.label, "A").toList();
Assert.assertEquals(3, vertexes.size());
GraphTraversal> gt = g.V().hasLabel("A").as("a").out("ab").as("b").select("a", "b");
List> list = gt.toList();
Assert.assertEquals(9, list.size());
for (int i = 0; i < 9; i++) {
Vertex a = list.get(i).get("a");
Vertex b = list.get(i).get("b");
Assert.assertTrue(a.equals(a1) || a.equals(a2) || a.equals(a3));
if (a.equals(a1)) {
Assert.assertTrue(b.equals(b11) || b.equals(b12) || b.equals(b13));
} else if (a.equals(a2)) {
Assert.assertTrue(b.equals(b21) || b.equals(b22) || b.equals(b23));
} else if (a.equals(a3)) {
Assert.assertTrue(b.equals(b31) || b.equals(b32) || b.equals(b33));
} else {
Assert.fail();
}
}
}
@Test
public void testVWithHas() {
Vertex a1 = this.sqlgGraph.addVertex(T.label, "A", "name", "a");
Vertex a2 = this.sqlgGraph.addVertex(T.label, "A", "name", "b");
Vertex a3 = this.sqlgGraph.addVertex(T.label, "A", "name", "c");
this.sqlgGraph.tx().commit();
GraphTraversalSource g = this.sqlgGraph.traversal();
List vertexes = g.V().has(T.label, "A").has("name", "a").toList();
Assert.assertEquals(1, vertexes.size());
}
@Test
public void testE() {
Vertex a1 = this.sqlgGraph.addVertex(T.label, "A");
Vertex b1 = this.sqlgGraph.addVertex(T.label, "B");
Vertex b2 = this.sqlgGraph.addVertex(T.label, "B");
Edge e1 = a1.addEdge("ab", b1);
Edge e2 = a1.addEdge("ab", b2);
this.sqlgGraph.tx().commit();
GraphTraversalSource gts = this.sqlgGraph.traversal();
List edges = gts.E().hasLabel("ab").toList();
Assert.assertEquals(2, edges.size());
Assert.assertTrue(edges.containsAll(Arrays.asList(e1, e2)));
}
@Test
public void TestEWithLabels() {
Vertex a1 = this.sqlgGraph.addVertex(T.label, "A");
Vertex b1 = this.sqlgGraph.addVertex(T.label, "B");
Vertex b2 = this.sqlgGraph.addVertex(T.label, "B");
Vertex c11 = this.sqlgGraph.addVertex(T.label, "C");
Vertex c12 = this.sqlgGraph.addVertex(T.label, "C");
Vertex c21 = this.sqlgGraph.addVertex(T.label, "C");
Vertex c22 = this.sqlgGraph.addVertex(T.label, "C");
Edge e1 = a1.addEdge("ab", b1);
Edge e2 = a1.addEdge("ab", b2);
Edge bc1 = b1.addEdge("bc", c11);
Edge bc2 = b1.addEdge("bc", c12);
Edge bc3 = b2.addEdge("bc", c21);
Edge bc4 = b2.addEdge("bc", c22);
this.sqlgGraph.tx().commit();
GraphTraversalSource gts = this.sqlgGraph.traversal();
List> result = gts.E().hasLabel("ab").as("ab").inV().as("b").out("bc").as("c").select("ab", "b", "c").toList();
Assert.assertEquals(4, result.size());
Map c1 = result.get(0);
Assert.assertTrue(c1.containsKey("ab") && c1.containsKey("b") && c1.containsKey("c"));
Map c2 = result.get(1);
Assert.assertTrue(c2.containsKey("ab") && c2.containsKey("b") && c2.containsKey("c"));
Map c3 = result.get(2);
Assert.assertTrue(c3.containsKey("ab") && c1.containsKey("b") && c3.containsKey("c"));
Map c4 = result.get(3);
Assert.assertTrue(c4.containsKey("ab") && c1.containsKey("b") && c4.containsKey("c"));
Assert.assertTrue(c1.get("c") instanceof Vertex);
Vertex c = (Vertex) c1.get("c");
Assert.assertEquals("C", c.label());
Vertex b = (Vertex) c1.get("b");
Assert.assertEquals("B", b.label());
Edge e = (Edge) c1.get("ab");
Assert.assertEquals("ab", e.label());
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy