org.umlg.sqlg.test.vertexout.TestVertexOutWithHas Maven / Gradle / Ivy

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.0
package org.umlg.sqlg.test.vertexout;

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;

/**
 * Date: 2014/07/13
 * Time: 4:48 PM
 */
public class TestVertexOutWithHas extends BaseTest {

    @Test
    public void testVertexOutWithHas() {
        Vertex marko = this.sqlgGraph.addVertex(T.label, "Person", "name", "marko");
        Vertex bmw1 = this.sqlgGraph.addVertex(T.label, "Car", "name", "bmw", "cc", 600);
        Vertex bmw2 = this.sqlgGraph.addVertex(T.label, "Car", "name", "bmw", "cc", 800);
        Vertex ktm1 = this.sqlgGraph.addVertex(T.label, "Bike", "name", "ktm", "cc", 200);
        Vertex ktm2 = this.sqlgGraph.addVertex(T.label, "Bike", "name", "ktm", "cc", 200);
        Vertex ktm3 = this.sqlgGraph.addVertex(T.label, "Bike", "name", "ktm", "cc", 400);
        marko.addEdge("drives", bmw1);
        marko.addEdge("drives", bmw2);
        marko.addEdge("drives", ktm1);
        marko.addEdge("drives", ktm2);
        marko.addEdge("drives", ktm3);
        this.sqlgGraph.tx().commit();
        List drivesBmw = vertexTraversal(this.sqlgGraph, marko).out("drives").has("name", "bmw").toList();
        Assert.assertEquals(2L, drivesBmw.size(), 0);
        List drivesKtm = vertexTraversal(this.sqlgGraph, marko).out("drives").has("name", "ktm").toList();
        Assert.assertEquals(3L, drivesKtm.size(), 0);

        List cc600 = vertexTraversal(this.sqlgGraph, marko).out("drives").has("cc", 600).toList();
        Assert.assertEquals(1L, cc600.size(), 0);
        List cc800 = vertexTraversal(this.sqlgGraph, marko).out("drives").has("cc", 800).toList();
        Assert.assertEquals(1L, cc800.size(), 0);
        List cc200 = vertexTraversal(this.sqlgGraph, marko).out("drives").has("cc", 200).toList();
        Assert.assertEquals(2L, cc200.size(), 0);
        List cc400 = vertexTraversal(this.sqlgGraph, marko).out("drives").has("cc", 400).toList();
        Assert.assertEquals(1L, cc400.size(), 0);
    }

    @Test
    public void testVertexInWithHas() {
        Vertex marko1 = this.sqlgGraph.addVertex(T.label, "Person", "name", "marko1");
        Vertex marko2 = this.sqlgGraph.addVertex(T.label, "Person", "name", "marko2", "age", 20);
        Vertex marko3 = this.sqlgGraph.addVertex(T.label, "Person", "name", "marko3", "age", 20);
        Vertex marko4 = this.sqlgGraph.addVertex(T.label, "Person", "name", "marko4", "age", 30);
        Vertex marko5 = this.sqlgGraph.addVertex(T.label, "Person", "name", "marko5", "age", 40);

        marko2.addEdge("friend", marko1);
        marko3.addEdge("friend", marko1);
        marko4.addEdge("friend", marko1);
        marko5.addEdge("friend", marko1);

        this.sqlgGraph.tx().commit();

        List inFriendMarko2 = vertexTraversal(this.sqlgGraph, marko1).in("friend").has("name", "marko2").toList();
        Assert.assertEquals(1, inFriendMarko2.size(), 0);
        List inFriendMarko3 = vertexTraversal(this.sqlgGraph, marko1).in("friend").has("name", "marko3").toList();
        Assert.assertEquals(1, inFriendMarko3.size(), 0);
        List inFriendMarko4 = vertexTraversal(this.sqlgGraph, marko1).in("friend").has("name", "marko4").toList();
        Assert.assertEquals(1, inFriendMarko4.size(), 0);
        List inFriendMarko5 = vertexTraversal(this.sqlgGraph, marko1).in("friend").has("name", "marko5").toList();
        Assert.assertEquals(1, inFriendMarko5.size(), 0);

        List inFriendAge20 = vertexTraversal(this.sqlgGraph, marko1).in("friend").has("age", 20).toList();
        Assert.assertEquals(2, inFriendAge20.size(), 0);
        List inFriendAge30 = vertexTraversal(this.sqlgGraph, marko1).in("friend").has("age", 30).toList();
        Assert.assertEquals(1, inFriendAge30.size(), 0);
        List inFriendAge40 = vertexTraversal(this.sqlgGraph, marko1).in("friend").has("age", 40).toList();
        Assert.assertEquals(1, inFriendAge40.size(), 0);

        List inFriendNameMarko3Age20 = vertexTraversal(this.sqlgGraph, marko1).in("friend").has("age", 20).has("name", "marko3").toList();
        Assert.assertEquals(1, inFriendNameMarko3Age20.size(), 0);

        List inFriendNameMarko4Age20 = vertexTraversal(this.sqlgGraph, marko1).in("friend").has("age", 20).has("name", "marko4").toList();
        Assert.assertEquals(0, inFriendNameMarko4Age20.size(), 0);

    }

    @Test
    public void testVertexHasLabel() {
        Vertex marko = this.sqlgGraph.addVertex(T.label, "Person", "name", "marko");
        Vertex bmw = this.sqlgGraph.addVertex(T.label, "Car", "name", "bmw");
        Vertex vw = this.sqlgGraph.addVertex(T.label, "Car", "name", "vw");
        marko.addEdge("drives", bmw);
        marko.addEdge("drives", vw);
        this.sqlgGraph.tx().commit();
        Assert.assertEquals(1,vertexTraversal(this.sqlgGraph, marko).out("drives").has(T.label, "Car").has("name", "bmw").count().next(), 0);
        Assert.assertEquals(0, vertexTraversal(this.sqlgGraph, marko).out("drives").has(T.label, "Person").has("name", "vw").count().next(), 0);
    }
    
    /**
     * we reference labels that have schemas
     */
    @Test
    public void testVertexHasLabelInSchema() {
        Vertex marko = this.sqlgGraph.addVertex(T.label, "TestSchema.Person", "name", "marko");
        Vertex bmw = this.sqlgGraph.addVertex(T.label, "TestSchema.Car", "name", "bmw");
        Vertex vw = this.sqlgGraph.addVertex(T.label, "TestSchema.Car", "name", "vw");
        marko.addEdge("drives", bmw);
        marko.addEdge("drives", vw);
        this.sqlgGraph.tx().commit();
        Assert.assertEquals(1,vertexTraversal(this.sqlgGraph, marko).out("drives").has(T.label, "TestSchema.Car").has("name", "bmw").count().next(), 0);
        Assert.assertEquals(0, vertexTraversal(this.sqlgGraph, marko).out("drives").has(T.label, "TestSchema.Person").has("name", "vw").count().next(), 0);
    }

    @Test
    public void testHasWithPercentageInIt() {
        this.sqlgGraph.addVertex(T.label, "Person", "uid", "\"BLRNC5->CXC4030052~%%%~FAJ1211373~%%%~2015-07-19~%%%~9999-12-31~%%%~Enabled~%%%~Licensed~%%%~Improved~%%%~compressed~%%%~mode~%%%~handling.~%%%~Restricted:~%%%~\"\"Partial.~%%%~Feature~%%%~is~%%%~restricted~%%%~in~%%%~RNC~%%%~W12B~%%%~SW.~%%%~RNC~%%%~W13.0.1.1~%%%~or~%%%~later~%%%~SW~%%%~is~%%%~required~%%%~in~%%%~order~%%%~to~%%%~run~%%%~this~%%%~feature.~%%%~For~%%%~RBS~%%%~W12.1.2.2/~%%%~W13.0.0.0~%%%~or~%%%~later~%%%~is~%%%~required.~%%%~OSS-RC~%%%~12.2~%%%~or~%%%~later~%%%~is~%%%~required.\"\".~%%%~GA:~%%%~W13A\"");
        this.sqlgGraph.tx().commit();
        List persons = this.sqlgGraph.traversal().V().hasLabel("Person").has("uid", "\"BLRNC5->CXC4030052~%%%~FAJ1211373~%%%~2015-07-19~%%%~9999-12-31~%%%~Enabled~%%%~Licensed~%%%~Improved~%%%~compressed~%%%~mode~%%%~handling.~%%%~Restricted:~%%%~\"\"Partial.~%%%~Feature~%%%~is~%%%~restricted~%%%~in~%%%~RNC~%%%~W12B~%%%~SW.~%%%~RNC~%%%~W13.0.1.1~%%%~or~%%%~later~%%%~SW~%%%~is~%%%~required~%%%~in~%%%~order~%%%~to~%%%~run~%%%~this~%%%~feature.~%%%~For~%%%~RBS~%%%~W12.1.2.2/~%%%~W13.0.0.0~%%%~or~%%%~later~%%%~is~%%%~required.~%%%~OSS-RC~%%%~12.2~%%%~or~%%%~later~%%%~is~%%%~required.\"\".~%%%~GA:~%%%~W13A\"").toList();
        Assert.assertEquals(1, persons.size());
    }

}