org.umlg.sqlg.test.gremlincompile.TestPropertyReference 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 static org.junit.Assert.assertEquals;
import static org.umlg.sqlg.predicate.PropertyReference.propertyRef;

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

import org.apache.tinkerpop.gremlin.process.traversal.Compare;
import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.DefaultGraphTraversal;
import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal;
import org.apache.tinkerpop.gremlin.structure.T;
import org.apache.tinkerpop.gremlin.structure.Vertex;
import org.junit.BeforeClass;
import org.junit.Test;
import org.umlg.sqlg.test.BaseTest;

/**
 * Test column references
 * @author JP Moresmau
 *
 */
public class TestPropertyReference extends BaseTest {

	@BeforeClass
    public static void beforeClass() {
        BaseTest.beforeClass();
        if (isPostgres()) {
            configuration.addProperty("distributed", true);
        }
    }
 
	public TestPropertyReference() {
		
	}
	
	 @Test
	 public void testInt() {
		 Vertex v1 = this.sqlgGraph.addVertex(T.label, "Person", "name", "johnny", "score",2,"experience",3);
	     Vertex v2 = this.sqlgGraph.addVertex(T.label, "Person", "name", "pietie", "score",2,"experience",2);
	     Vertex v3 = this.sqlgGraph.addVertex(T.label, "Person", "name", "koosie", "score",2,"experience",1);
	       
	     this.sqlgGraph.tx().commit();
	     
	     assertOneStepOneVertex(this.sqlgGraph.traversal().V().hasLabel("Person").has("score",propertyRef(Compare.eq, "experience"))
	    		 ,v2);
	     assertOneStepOneVertex(this.sqlgGraph.traversal().V().hasLabel("Person").has("score",propertyRef(Compare.lt, "experience"))
	    		 ,v1);
	     assertOneStepOneVertex(this.sqlgGraph.traversal().V().hasLabel("Person").has("score",propertyRef(Compare.gt, "experience"))
	    		 ,v3);
	 }
	 
	 private void assertOneStepOneVertex( GraphTraversal gt,Vertex v){
		 DefaultGraphTraversal traversal = (DefaultGraphTraversal)gt;
	     assertEquals(2, traversal.getSteps().size());
	     List vertices = traversal.toList();
	     assertEquals(1, traversal.getSteps().size());
	     assertEquals(1, vertices.size());
	     assertEquals(v, vertices.get(0));
	 }
	 
	 @Test
	 public void testMultiplePath(){
		 Vertex v1 = this.sqlgGraph.addVertex(T.label, "Person", "name", "johnny", "score",2,"experience",3);
	     Vertex v2 = this.sqlgGraph.addVertex(T.label, "Person", "name", "pietie", "score",2,"experience",2);
	     Vertex v3 = this.sqlgGraph.addVertex(T.label, "Person", "name", "koosie", "score",2,"experience",1);
	       
	     Vertex v4 = this.sqlgGraph.addVertex(T.label, "Group", "name", "Friends");
	     v4.addEdge("contains", v1);
	     v4.addEdge("contains", v2);
	     v4.addEdge("contains", v3);
	     
	     Vertex v5 = this.sqlgGraph.addVertex(T.label, "Company", "name", "Acme");
	     v5.addEdge("groups", v4);
	     
	     this.sqlgGraph.tx().commit();
	    
	     GraphTraversal> traversal =sqlgGraph.traversal()
	    		 .V().hasLabel("Company").as("c").out("groups")
	    		 .as("g").out("contains").has("score",propertyRef(Compare.eq, "experience")).as("p")
	    		 .select("c","p");
	     List> l =traversal.toList();
	     assertEquals(1,l.size());
	     assertEquals(v2,l.get(0).get("p"));
	 }
}