org.umlg.sqlg.test.union.TestUnion 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.union;

import org.apache.tinkerpop.gremlin.process.traversal.Traversal;
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;
import java.util.Map;

import static org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.__.*;
import static org.junit.Assert.assertEquals;

/**
 * Date: 2016/05/30
 * Time: 9:01 PM
 */
public class TestUnion extends BaseTest {

    @Test
    public void testUnionAsPerUMLG() {
        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 bb1 = this.sqlgGraph.addVertex(T.label, "BB");
        Vertex bb2 = this.sqlgGraph.addVertex(T.label, "BB");
        a1.addEdge("ab", b1);
        a1.addEdge("ab", b2);
        bb1.addEdge("ab", a1);
        bb2.addEdge("ab", a1);
        this.sqlgGraph.tx().commit();
        List vertices = this.sqlgGraph.traversal().V(a1).in("ab").toList();
        assertEquals(2, vertices.size());
        vertices = this.sqlgGraph.traversal().V(a1).union(optional(out("ab")), optional(in("ab"))).toList();
        assertEquals(4, vertices.size());
        vertices = this.sqlgGraph.traversal().V(a1).union(out("ab"), in("ab")).toList();
        assertEquals(4, vertices.size());
        vertices = this.sqlgGraph.traversal().V(a1).union(out("ab"), in("ab")).toList();
        assertEquals(4, vertices.size());
    }

    @Test
    public void testUnionFailure() {
        loadModern();


        Traversal> traversal = (Traversal) this.sqlgGraph.traversal().V().union(
                repeat(union(
                        out("created"),
                        in("created"))).times(2),
                repeat(union(
                        in("created"),
                        out("created"))).times(2)).label().groupCount();
        printTraversalForm(traversal);
        final Map groupCount = traversal.next();
        Assert.assertFalse(traversal.hasNext());
        Assert.assertEquals(12l, groupCount.get("software").longValue());
        Assert.assertEquals(20l, groupCount.get("person").longValue());
        Assert.assertEquals(2, groupCount.size());
    }

}