• Home
• org.umlg
• sqlg-test
• 2.1.1-1
• source code
• TestLocalVertexStepWithOrder.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

org.umlg.sqlg.test.localvertexstep.TestLocalVertexStepWithOrder Maven / Gradle / Ivy

package org.umlg.sqlg.test.localvertexstep;

import org.apache.tinkerpop.gremlin.process.traversal.Order;
import org.apache.tinkerpop.gremlin.process.traversal.Path;
import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.DefaultGraphTraversal;
import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.__;
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.List;

/**
 * @author Pieter Martin (https://github.com/pietermartin)
 *         Date: 2017/04/30
 */
public class TestLocalVertexStepWithOrder extends BaseTest {

    @Test
    public void testSqlgBranchStepKeepsIncomingOrderNotOnDb() {
        Vertex a1 = this.sqlgGraph.addVertex(T.label, "A");
        Vertex b1 = this.sqlgGraph.addVertex(T.label, "B", "order", 1);
        Vertex b2 = this.sqlgGraph.addVertex(T.label, "B", "order", 3);
        Vertex b3 = this.sqlgGraph.addVertex(T.label, "B", "order", 5);
        Vertex bb1 = this.sqlgGraph.addVertex(T.label, "BB", "order", 2);
        Vertex bb2 = this.sqlgGraph.addVertex(T.label, "BB", "order", 4);
        Vertex bb3 = this.sqlgGraph.addVertex(T.label, "BB", "order", 6);
        a1.addEdge("ab", b1);
        a1.addEdge("ab", b2);
        a1.addEdge("ab", b3);
        a1.addEdge("abb", bb1);
        a1.addEdge("abb", bb2);
        a1.addEdge("abb", bb3);
        Vertex c1 = this.sqlgGraph.addVertex(T.label, "C", "order", 2, "name", "c1");
        Vertex c2 = this.sqlgGraph.addVertex(T.label, "C", "order", 4, "name", "c2");
        Vertex c3 = this.sqlgGraph.addVertex(T.label, "C", "order", 6, "name", "c3");
        Vertex cc1 = this.sqlgGraph.addVertex(T.label, "CC", "order", 1, "name", "cc1");
        Vertex cc2 = this.sqlgGraph.addVertex(T.label, "CC", "order", 3, "name", "cc2");
        Vertex cc3 = this.sqlgGraph.addVertex(T.label, "CC", "order", 5, "name", "cc3");
        b1.addEdge("bc", c1);
        b1.addEdge("bc", c2);
        b1.addEdge("bc", c3);
        b2.addEdge("bc", c1);
        b2.addEdge("bc", c2);
        b2.addEdge("bc", c3);
        b3.addEdge("bc", c1);
        b3.addEdge("bc", c2);
        b3.addEdge("bc", c3);

        b1.addEdge("bcc", cc1);
        b1.addEdge("bcc", cc2);
        b1.addEdge("bcc", cc3);
        b2.addEdge("bcc", cc1);
        b2.addEdge("bcc", cc2);
        b2.addEdge("bcc", cc3);
        b3.addEdge("bcc", cc1);
        b3.addEdge("bcc", cc2);
        b3.addEdge("bcc", cc3);
        this.sqlgGraph.tx().commit();

        DefaultGraphTraversal traversal = (DefaultGraphTraversal) this.sqlgGraph.traversal().V(a1)
                .local(
                        __.out("ab", "abb").order().by("order", Order.decr)
                                .local(
                                        __.out("bc", "bcc").order().by("order", Order.decr)
                                )
                ).path();
        List paths = traversal.toList();
        for (Path path : paths) {
            System.out.println(path.toString() + " " + ((Vertex) path.objects().get(2)).value("name"));
        }
        Assert.assertEquals(18, paths.size());
        Assert.assertEquals(b3, paths.get(0).objects().get(1));
        Assert.assertEquals(c3, paths.get(0).objects().get(2));
        Assert.assertEquals(b3, paths.get(1).objects().get(1));
        Assert.assertEquals(cc3, paths.get(1).objects().get(2));
        Assert.assertEquals(b3, paths.get(2).objects().get(1));
        Assert.assertEquals(c2, paths.get(2).objects().get(2));
        Assert.assertEquals(b3, paths.get(3).objects().get(1));
        Assert.assertEquals(cc2, paths.get(3).objects().get(2));
        Assert.assertEquals(b3, paths.get(4).objects().get(1));
        Assert.assertEquals(c1, paths.get(4).objects().get(2));
        Assert.assertEquals(b3, paths.get(5).objects().get(1));
        Assert.assertEquals(cc1, paths.get(5).objects().get(2));

        Assert.assertEquals(b2, paths.get(6).objects().get(1));
        Assert.assertEquals(c3, paths.get(6).objects().get(2));
        Assert.assertEquals(b2, paths.get(7).objects().get(1));
        Assert.assertEquals(cc3, paths.get(7).objects().get(2));
        Assert.assertEquals(b2, paths.get(8).objects().get(1));
        Assert.assertEquals(c2, paths.get(8).objects().get(2));
        Assert.assertEquals(b2, paths.get(9).objects().get(1));
        Assert.assertEquals(cc2, paths.get(9).objects().get(2));
        Assert.assertEquals(b2, paths.get(10).objects().get(1));
        Assert.assertEquals(c1, paths.get(10).objects().get(2));
        Assert.assertEquals(b2, paths.get(11).objects().get(1));
        Assert.assertEquals(cc1, paths.get(11).objects().get(2));

        Assert.assertEquals(b1, paths.get(12).objects().get(1));
        Assert.assertEquals(c3, paths.get(12).objects().get(2));
        Assert.assertEquals(b1, paths.get(13).objects().get(1));
        Assert.assertEquals(cc3, paths.get(13).objects().get(2));
        Assert.assertEquals(b1, paths.get(14).objects().get(1));
        Assert.assertEquals(c2, paths.get(14).objects().get(2));
        Assert.assertEquals(b1, paths.get(15).objects().get(1));
        Assert.assertEquals(cc2, paths.get(15).objects().get(2));
        Assert.assertEquals(b1, paths.get(16).objects().get(1));
        Assert.assertEquals(c1, paths.get(16).objects().get(2));
        Assert.assertEquals(b1, paths.get(17).objects().get(1));
        Assert.assertEquals(cc1, paths.get(17).objects().get(2));
    }

    @Test
    public void testSqlgVertexStepOrderStartsProperly() {
        Vertex a1 = this.sqlgGraph.addVertex(T.label, "A", "name", "a1", "order", 1);
        Vertex a2 = this.sqlgGraph.addVertex(T.label, "A", "name", "a2", "order", 2);
        Vertex b1 = this.sqlgGraph.addVertex(T.label, "B", "name", "b1");
        Vertex b2 = this.sqlgGraph.addVertex(T.label, "B", "name", "b2");
        Vertex b3 = this.sqlgGraph.addVertex(T.label, "B", "name", "b3");
        a1.addEdge("ab", b1, "order", 1);
        a1.addEdge("ab", b2, "order", 2);
        a1.addEdge("ab", b3, "order", 3);
        a2.addEdge("ab", b1, "order", 1);
        a2.addEdge("ab", b2, "order", 2);
        a2.addEdge("ab", b3, "order", 3);
        Vertex c1 = this.sqlgGraph.addVertex(T.label, "C", "name", "c1", "order", 1);
        Vertex c2 = this.sqlgGraph.addVertex(T.label, "C", "name", "c2", "order", 2);
        Vertex c3 = this.sqlgGraph.addVertex(T.label, "C", "name", "c3", "order", 3);
        Vertex c4 = this.sqlgGraph.addVertex(T.label, "C", "name", "c4", "order", 1);
        Vertex c5 = this.sqlgGraph.addVertex(T.label, "C", "name", "c5", "order", 2);
        Vertex c6 = this.sqlgGraph.addVertex(T.label, "C", "name", "c6", "order", 3);
        Vertex c7 = this.sqlgGraph.addVertex(T.label, "C", "name", "c7", "order", 1);
        Vertex c8 = this.sqlgGraph.addVertex(T.label, "C", "name", "c8", "order", 2);
        Vertex c9 = this.sqlgGraph.addVertex(T.label, "C", "name", "c9", "order", 3);
        b1.addEdge("bc", c1);
        b1.addEdge("bc", c2);
        b1.addEdge("bc", c3);
        b2.addEdge("bc", c4);
        b2.addEdge("bc", c5);
        b2.addEdge("bc", c6);
        b3.addEdge("bc", c7);
        b3.addEdge("bc", c8);
        b3.addEdge("bc", c9);
        this.sqlgGraph.tx().commit();

        DefaultGraphTraversal traversal = (DefaultGraphTraversal) this.sqlgGraph.traversal()
                .V().hasLabel("A").order().by("order")
                .local(
                        __.outE("ab").order().by("order", Order.decr).inV()
                                .local(
                                        __.out("bc").order().by("order", Order.decr)
                                )
                )
                .path();
        List paths = traversal.toList();
        for (Path path : paths) {
            System.out.println(path.toString() + " " + ((Vertex) path.objects().get(3)).value("name"));
        }

        Assert.assertEquals(18, paths.size());
        Assert.assertEquals(a1, paths.get(0).objects().get(0));
        Assert.assertEquals(c9, paths.get(0).objects().get(3));
        Assert.assertEquals(a1, paths.get(1).objects().get(0));
        Assert.assertEquals(c8, paths.get(1).objects().get(3));
        Assert.assertEquals(a1, paths.get(2).objects().get(0));
        Assert.assertEquals(c7, paths.get(2).objects().get(3));

        Assert.assertEquals(a1, paths.get(3).objects().get(0));
        Assert.assertEquals(c6, paths.get(3).objects().get(3));
        Assert.assertEquals(a1, paths.get(4).objects().get(0));
        Assert.assertEquals(c5, paths.get(4).objects().get(3));
        Assert.assertEquals(a1, paths.get(5).objects().get(0));
        Assert.assertEquals(c4, paths.get(5).objects().get(3));

        Assert.assertEquals(a1, paths.get(6).objects().get(0));
        Assert.assertEquals(c3, paths.get(6).objects().get(3));
        Assert.assertEquals(a1, paths.get(7).objects().get(0));
        Assert.assertEquals(c2, paths.get(7).objects().get(3));
        Assert.assertEquals(a1, paths.get(8).objects().get(0));
        Assert.assertEquals(c1, paths.get(8).objects().get(3));



        Assert.assertEquals(a2, paths.get(9).objects().get(0));
        Assert.assertEquals(c9, paths.get(9).objects().get(3));
        Assert.assertEquals(a2, paths.get(10).objects().get(0));
        Assert.assertEquals(c8, paths.get(10).objects().get(3));
        Assert.assertEquals(a2, paths.get(11).objects().get(0));
        Assert.assertEquals(c7, paths.get(11).objects().get(3));

        Assert.assertEquals(a2, paths.get(12).objects().get(0));
        Assert.assertEquals(c6, paths.get(12).objects().get(3));
        Assert.assertEquals(a2, paths.get(13).objects().get(0));
        Assert.assertEquals(c5, paths.get(13).objects().get(3));
        Assert.assertEquals(a2, paths.get(14).objects().get(0));
        Assert.assertEquals(c4, paths.get(14).objects().get(3));

        Assert.assertEquals(a2, paths.get(15).objects().get(0));
        Assert.assertEquals(c3, paths.get(15).objects().get(3));
        Assert.assertEquals(a2, paths.get(16).objects().get(0));
        Assert.assertEquals(c2, paths.get(16).objects().get(3));
        Assert.assertEquals(a2, paths.get(17).objects().get(0));
        Assert.assertEquals(c1, paths.get(17).objects().get(3));
    }
}