Java AbstractGremlinTest类代码示例

import org.apache.tinkerpop.gremlin.AbstractGremlinTest; //导入依赖的package包/类
@Test
public void testTail() throws IOException {
    Graph graph = this.sqlgGraph;
    final GraphReader reader = GryoReader.build()
            .mapper(graph.io(GryoIo.build()).mapper().create())
            .create();
    try (final InputStream stream = AbstractGremlinTest.class.getResourceAsStream("/tinkerpop-modern.kryo")) {
        reader.readGraph(stream, graph);
    }
    assertModernGraph(graph, true, false);
    GraphTraversalSource g = graph.traversal();
    int tail = 0;
    for (int i = 1; i < 72; i++) {
        tail = i;
        List<Vertex> vertices = g.V().repeat(both()).times(3).tail(tail).toList();
        if (tail != vertices.size()) {
            System.out.println("expected " + tail + " found " + vertices.size());
        }
    }
}
 

import org.apache.tinkerpop.gremlin.AbstractGremlinTest; //导入依赖的package包/类
@Test
public void shouldConstructDetachedEdge() throws IOException {
    Graph g = this.sqlgGraph;
    final GraphReader reader = GryoReader.build()
            .mapper(g.io(GryoIo.build()).mapper().create())
            .create();
    try (final InputStream stream = AbstractGremlinTest.class.getResourceAsStream("/tinkerpop-modern.kryo")) {
        reader.readGraph(stream, g);
    }
    assertModernGraph(g, true, false);
    Edge e = g.traversal().E(convertToEdgeId("marko", "knows", "vadas")).next();
    e.property("year", 2002);
    g.tx().commit();
    e = g.traversal().E(convertToEdgeId("marko", "knows", "vadas")).next();
    final DetachedEdge detachedEdge = DetachedFactory.detach(e, true);
    Assert.assertEquals(convertToEdgeId("marko", "knows", "vadas"), detachedEdge.id());

    Assert.assertEquals("knows", detachedEdge.label());
    Assert.assertEquals(DetachedVertex.class, detachedEdge.vertices(Direction.OUT).next().getClass());
    Assert.assertEquals(convertToVertexId("marko"), detachedEdge.vertices(Direction.OUT).next().id());
    Assert.assertEquals("person", detachedEdge.vertices(Direction.IN).next().label());
    Assert.assertEquals(DetachedVertex.class, detachedEdge.vertices(Direction.IN).next().getClass());
    Assert.assertEquals(convertToVertexId("vadas"), detachedEdge.vertices(Direction.IN).next().id());
    Assert.assertEquals("person", detachedEdge.vertices(Direction.IN).next().label());

    Assert.assertEquals(2, IteratorUtils.count(detachedEdge.properties()));
    Assert.assertEquals(1, IteratorUtils.count(detachedEdge.properties("year")));
    Assert.assertEquals(0.5d, detachedEdge.properties("weight").next().value());
}
 

import org.apache.tinkerpop.gremlin.AbstractGremlinTest; //导入依赖的package包/类
@Test
public void shouldReferenceVertexWhenRemoved() {
    final AtomicBoolean triggered = new AtomicBoolean(false);
    final Vertex v = this.sqlgGraph.addVertex();
    final String label = v.label();
    final Object id = v.id();

    final MutationListener listener = new AbstractMutationListener() {
        @Override
        public void vertexRemoved(final Vertex element) {
            Assert.assertThat(element, IsInstanceOf.instanceOf(ReferenceVertex.class));
            Assert.assertEquals(id, element.id());
            Assert.assertEquals(label, element.label());
            triggered.set(true);
        }
    };
    final EventStrategy.Builder builder = EventStrategy.build().addListener(listener).detach(ReferenceFactory.class);

    if (this.sqlgGraph.features().graph().supportsTransactions())
        builder.eventQueue(new EventStrategy.TransactionalEventQueue(this.sqlgGraph));

    final EventStrategy eventStrategy = builder.create();
    final GraphTraversalSource gts = this.sqlgGraph.traversal().withStrategies(eventStrategy);

    gts.V(v).drop().iterate();
    this.sqlgGraph.tx().commit();

    AbstractGremlinTest.assertVertexEdgeCounts(this.sqlgGraph, 0, 0);
    Assert.assertThat(triggered.get(), CoreMatchers.is(true));
}
 

import org.apache.tinkerpop.gremlin.AbstractGremlinTest; //导入依赖的package包/类
public void g_V_chooseXlabel_eq_person__unionX__out_lang__out_nameX__in_labelX() throws IOException {
    Graph graph = this.sqlgGraph;
    final GraphReader reader = GryoReader.build()
            .mapper(graph.io(GryoIo.build()).mapper().create())
            .create();
    try (final InputStream stream = AbstractGremlinTest.class.getResourceAsStream("/tinkerpop-modern.kryo")) {
        reader.readGraph(stream, graph);
    }
    assertModernGraph(graph, true, false);
    GraphTraversalSource g = graph.traversal();
    List<Vertex> traversala2 =  g.V().hasId(convertToVertexId("marko")).toList();
    Assert.assertEquals(1, traversala2.size());
    Assert.assertEquals(convertToVertex(graph, "marko"), traversala2.get(0));

}
 

import org.apache.tinkerpop.gremlin.AbstractGremlinTest; //导入依赖的package包/类
protected void loadModern(SqlgGraph sqlgGraph) {
    Io.Builder<GraphSONIo> builder = GraphSONIo.build(GraphSONVersion.V3_0);
    final GraphReader reader = sqlgGraph.io(builder).reader().create();
    try (final InputStream stream = AbstractGremlinTest.class.getResourceAsStream("/tinkerpop-modern-v3d0.json")) {
        reader.readGraph(stream, sqlgGraph);
    } catch (IOException e) {
        Assert.fail(e.getMessage());
    }
}
 

import org.apache.tinkerpop.gremlin.AbstractGremlinTest; //导入依赖的package包/类
protected void loadGratefulDead(SqlgGraph sqlgGraph) {
    Io.Builder<GraphSONIo> builder = GraphSONIo.build(GraphSONVersion.V3_0);
    final GraphReader reader = sqlgGraph.io(builder).reader().create();
    try (final InputStream stream = AbstractGremlinTest.class.getResourceAsStream("/grateful-dead-v3d0.json")) {
        reader.readGraph(stream, sqlgGraph);
    } catch (IOException e) {
        Assert.fail(e.getMessage());
    }
}
 

import org.apache.tinkerpop.gremlin.AbstractGremlinTest; //导入依赖的package包/类
@Override
public boolean beforeTestExecution(final Class<? extends AbstractGremlinTest> testClass) {
    unloadSugar();
    SugarLoader.load();
    return true;
}
 

import org.apache.tinkerpop.gremlin.AbstractGremlinTest; //导入依赖的package包/类
@Override
public void afterTestExecution(final Class<? extends AbstractGremlinTest> testClass) {
    unloadSugar();
}
 

import org.apache.tinkerpop.gremlin.AbstractGremlinTest; //导入依赖的package包/类
/**
 * This test helps with data conversions on Grateful Dead.  No Assertions...run as needed. Never read from the
 * GraphML source as it will always use a String identifier.
 */
@Test
public void shouldWriteGratefulDead() throws IOException {
    final Graph g = TinkerGraph.open();

    final GraphReader reader = GryoReader.build().create();
    try (final InputStream stream = AbstractGremlinTest.class.getResourceAsStream("/org/apache/tinkerpop/gremlin/structure/io/gryo/grateful-dead.kryo")) {
        reader.readGraph(stream, g);
    }

    /* keep this hanging around because changes to gryo format will need grateful dead generated from json so you can generate the gio
    final GraphSONMapper mapper = GraphSONMapper.build().embedTypes(true).create();
    final GraphReader reader = org.apache.tinkerpop.gremlin.structure.io.graphson.GraphSONReader.build().mapper(mapper).create();
    try (final InputStream stream = AbstractGremlinTest.class.getResourceAsStream("/org/apache/tinkerpop/gremlin/structure/io/graphson/grateful-dead-typed.json")) {
        reader.readGraph(stream, g);
    }
    */

    final Graph ng = TinkerGraph.open();
    g.traversal().V().sideEffect(ov -> {
        final Vertex v = ov.get();
        if (v.label().equals("song"))
            ng.addVertex(T.id, Integer.parseInt(v.id().toString()), T.label, "song", "name", v.value("name"), "performances", v.property("performances").orElse(0), "songType", v.property("songType").orElse(""));
        else if (v.label().equals("artist"))
            ng.addVertex(T.id, Integer.parseInt(v.id().toString()), T.label, "artist", "name", v.value("name"));
        else
            throw new RuntimeException("damn");
    }).iterate();

    g.traversal().E().sideEffect(oe -> {
        final Edge e = oe.get();
        final Vertex v2 = ng.traversal().V(Integer.parseInt(e.inVertex().id().toString())).next();
        final Vertex v1 = ng.traversal().V(Integer.parseInt(e.outVertex().id().toString())).next();

        if (e.label().equals("followedBy"))
            v1.addEdge("followedBy", v2, T.id, Integer.parseInt(e.id().toString()), "weight", e.value("weight"));
        else if (e.label().equals("sungBy"))
            v1.addEdge("sungBy", v2, T.id, Integer.parseInt(e.id().toString()));
        else if (e.label().equals("writtenBy"))
            v1.addEdge("writtenBy", v2, T.id, Integer.parseInt(e.id().toString()));
        else
            throw new RuntimeException("bah");

    }).iterate();

    final OutputStream os = new FileOutputStream(tempPath + "grateful-dead.kryo");
    GryoWriter.build().create().writeGraph(os, ng);
    os.close();

    final OutputStream os2 = new FileOutputStream(tempPath + "grateful-dead.json");
    GraphSONWriter.build().mapper(GraphSONMapper.build().create()).create().writeGraph(os2, g);
    os2.close();

    final OutputStream os3 = new FileOutputStream(tempPath + "grateful-dead.xml");
    GraphMLWriter.build().create().writeGraph(os3, g);
    os3.close();

    final OutputStream os4 = new FileOutputStream(tempPath + "grateful-dead-typed.json");
    GraphSONWriter.build().mapper(GraphSONMapper.build().embedTypes(true).create()).create().writeGraph(os4, g);
    os4.close();
}