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

import org.apache.tinkerpop.gremlin.structure.T;
import org.apache.tinkerpop.gremlin.structure.Vertex;
import org.junit.Assert;
import org.junit.Test;

import java.util.HashSet;
import java.util.Set;

/**
 * Date: 2014/07/22
 * Time: 7:53 PM
 */
public class TestEdgeToDifferentLabeledVertexes extends BaseTest {

    @Test
    public void testEdgeIsToDifferentLabeledVertexes() {
        Vertex v1 = this.sqlgGraph.addVertex(T.label, "Person", "name", "a");
        Vertex v2 = this.sqlgGraph.addVertex(T.label, "Person", "name", "b");
        Vertex v3 = this.sqlgGraph.addVertex(T.label, "Product", "name", "c");
        v1.addEdge("label1", v2);
        v1.addEdge("label1", v3);
        this.sqlgGraph.tx().commit();
        Assert.assertEquals(2, vertexTraversal(this.sqlgGraph, v1).out().count().next(), 0);
        Set names = new HashSet<>();
        vertexTraversal(this.sqlgGraph, v1).out().forEachRemaining(
               v -> names.add(v.property("name").value())
        );
        Assert.assertTrue(names.contains("b"));
        Assert.assertTrue(names.contains("c"));
    }
}