org.umlg.sqlg.test.gremlincompile.TestRepeatStepWithLabels Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sqlg-test Show documentation
There is a newer version: 3.1.1
package org.umlg.sqlg.test.gremlincompile;

import org.apache.commons.lang3.tuple.Pair;
import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.DefaultGraphTraversal;
import org.apache.tinkerpop.gremlin.structure.T;
import org.apache.tinkerpop.gremlin.structure.Vertex;
import org.junit.Assert;
import org.junit.Test;
import org.umlg.sqlg.test.BaseTest;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import static org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.__.out;

/**
 * Date: 2016/10/26
 * Time: 11:53 AM
 */
public class TestRepeatStepWithLabels extends BaseTest {

    @Test
    public void testEmitWithLabel() {

        Vertex a1 = this.sqlgGraph.addVertex(T.label, "A");
        Vertex b1 = this.sqlgGraph.addVertex(T.label, "B");
        Vertex c1 = this.sqlgGraph.addVertex(T.label, "C");
        a1.addEdge("ab", b1);
        b1.addEdge("bc", c1);
        this.sqlgGraph.tx().commit();

        DefaultGraphTraversal traversal = (DefaultGraphTraversal) this.sqlgGraph.traversal()
                .V(a1).repeat(out()).times(2).emit().as("bc").select("bc");
        Assert.assertEquals(3, traversal.getSteps().size());
        List result = traversal.toList();
        Assert.assertEquals(2, traversal.getSteps().size());
        Assert.assertEquals(2, result.size());
        Assert.assertTrue(result.contains(b1));
        Assert.assertTrue(result.contains(c1));
    }

    @Test
    public void testRepeatEmitLabel1() {
        loadModern();
        DefaultGraphTraversal> traversal = (DefaultGraphTraversal>) this.sqlgGraph.traversal()
                .V().as("a")
                .repeat(out()).times(1).emit().as("b")
                .select("a", "b");
        Assert.assertEquals(3, traversal.getSteps().size());
        List> labelVertexMaps = traversal
                .toList();
        Assert.assertEquals(2, traversal.getSteps().size());
        List> testPairs = new ArrayList<>();
        testPairs.add(Pair.of(convertToVertex(this.sqlgGraph, "marko"), convertToVertex(this.sqlgGraph, "lop")));
        testPairs.add(Pair.of(convertToVertex(this.sqlgGraph, "marko"), convertToVertex(this.sqlgGraph, "vadas")));
        testPairs.add(Pair.of(convertToVertex(this.sqlgGraph, "marko"), convertToVertex(this.sqlgGraph, "josh")));
        testPairs.add(Pair.of(convertToVertex(this.sqlgGraph, "josh"), convertToVertex(this.sqlgGraph, "ripple")));
        testPairs.add(Pair.of(convertToVertex(this.sqlgGraph, "josh"), convertToVertex(this.sqlgGraph, "lop")));
        testPairs.add(Pair.of(convertToVertex(this.sqlgGraph, "peter"), convertToVertex(this.sqlgGraph, "lop")));
        Assert.assertEquals(6, labelVertexMaps.size());
        for (Map labelVertexMap : labelVertexMaps) {
            Pair pair = Pair.of(labelVertexMap.get("a"), labelVertexMap.get("b"));
            Assert.assertTrue(testPairs.remove(pair));
        }
        Assert.assertTrue(testPairs.isEmpty());
    }

    @Test
    public void testRepeatEmitLabel2() {
        loadModern();
        DefaultGraphTraversal> traversal = (DefaultGraphTraversal>) this.sqlgGraph.traversal()
                .V().as("a")
                .repeat(out()).times(2).emit().as("b")
                .select("a", "b");
        Assert.assertEquals(3, traversal.getSteps().size());
        List> labelVertexMaps = traversal.toList();
        Assert.assertEquals(2, traversal.getSteps().size());

        List> testPairs = new ArrayList<>();
        testPairs.add(Pair.of(convertToVertex(this.sqlgGraph, "marko"), convertToVertex(this.sqlgGraph, "lop")));
        testPairs.add(Pair.of(convertToVertex(this.sqlgGraph, "marko"), convertToVertex(this.sqlgGraph, "lop")));
        testPairs.add(Pair.of(convertToVertex(this.sqlgGraph, "marko"), convertToVertex(this.sqlgGraph, "vadas")));
        testPairs.add(Pair.of(convertToVertex(this.sqlgGraph, "marko"), convertToVertex(this.sqlgGraph, "josh")));
        testPairs.add(Pair.of(convertToVertex(this.sqlgGraph, "marko"), convertToVertex(this.sqlgGraph, "ripple")));
        testPairs.add(Pair.of(convertToVertex(this.sqlgGraph, "josh"), convertToVertex(this.sqlgGraph, "ripple")));
        testPairs.add(Pair.of(convertToVertex(this.sqlgGraph, "josh"), convertToVertex(this.sqlgGraph, "lop")));
        testPairs.add(Pair.of(convertToVertex(this.sqlgGraph, "peter"), convertToVertex(this.sqlgGraph, "lop")));

        Assert.assertEquals(8, labelVertexMaps.size());
        for (Map labelVertexMap : labelVertexMaps) {
            System.out.println(labelVertexMap);
            Pair pair = Pair.of(labelVertexMap.get("a"), labelVertexMap.get("b"));
            Assert.assertTrue(testPairs.remove(pair));
        }
        Assert.assertTrue(testPairs.isEmpty());
    }

}