org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal#Admin

Source File: CoreTraversalTest.java From tinkerpop with Apache License 2.0

6 votes

@Test
@LoadGraphWith(MODERN)
public void shouldThrowFastNoSuchElementExceptionInNestedTraversals() {
    //The nested traversal should throw a regular FastNoSuchElementException

    final GraphTraversal<Object, Object> nestedTraversal = __.has("name", "foo");
    final GraphTraversal<Vertex, Object> traversal = g.V().has("name", "marko").branch(nestedTraversal);

    final GraphTraversal.Admin<Object, Object> nestedTraversalAdmin = nestedTraversal.asAdmin();
    nestedTraversalAdmin.reset();
    nestedTraversalAdmin.addStart(nestedTraversalAdmin.getTraverserGenerator().generate(g.V().has("name", "marko").next(), (Step) traversal.asAdmin().getStartStep(), 1l));

    try {
        nestedTraversal.next();
    } catch (NoSuchElementException e) {
        assertEquals(FastNoSuchElementException.class, e.getClass());
    }
}

Source File: InlineFilterStrategyTest.java From tinkerpop with Apache License 2.0

5 votes

private static GraphTraversal.Admin<?, ?> addHas(final GraphTraversal<?, ?> traversal, final Object... hasKeyValues) {
    final HasStep<?> hasStep = new HasStep<>((Traversal.Admin) traversal);
    for (int i = 0; i < hasKeyValues.length; i = i + 2) {
        hasStep.addHasContainer(new HasContainer((String) hasKeyValues[i], (P) hasKeyValues[i + 1]));
    }
    return traversal.asAdmin().addStep(hasStep);
}

Source File: JavaTranslatorBenchmark.java From tinkerpop with Apache License 2.0

5 votes

@Benchmark
public GraphTraversal.Admin<Vertex,Vertex> testTranslationLong() {
    return translator.translate(g.V().match(
            as("a").has("song", "name", "HERE COMES SUNSHINE"),
            as("a").map(inE("followedBy").values("weight").mean()).as("b"),
            as("a").inE("followedBy").as("c"),
            as("c").filter(values("weight").where(P.gte("b"))).outV().as("d")).
            <String>select("d").by("name").asAdmin().getBytecode());
}

Source File: IncidentToAdjacentStrategyProcessTest.java From tinkerpop with Apache License 2.0

5 votes

@Test
@LoadGraphWith(MODERN)
public void shouldGenerateCorrectTraversers() throws Exception {

    final GraphTraversalSource itag = g.withStrategies(IncidentToAdjacentStrategy.instance());
    
    GraphTraversal.Admin traversal;

    traversal = itag.V().outE().iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);
    traversal = itag.V().outE().inV().iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);
    traversal = itag.V().outE().otherV().iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);
    traversal = itag.V().out().iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);

    traversal = itag.V().bothE().iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);
    traversal = itag.V().bothE().otherV().iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);
    traversal = itag.V().both().iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);

    traversal = itag.V().flatMap(bothE()).iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);
    traversal = itag.V().flatMap(bothE().otherV()).iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);
    traversal = itag.V().flatMap(both()).iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);
}

Source File: SubgraphStrategyProcessTest.java From tinkerpop with Apache License 2.0

5 votes

@Test
@LoadGraphWith(MODERN)
public void shouldGenerateCorrectTraversers() throws Exception {

    assumeThat(graph, Matchers.not(Matchers.instanceOf(RemoteGraph.class)));

    GraphTraversalSource sg;
    GraphTraversal.Admin traversal;
    SubgraphStrategy strategy;

    strategy = SubgraphStrategy.build().vertices(has("name", P.within("josh", "lop", "ripple"))).create();
    sg = g.withStrategies(strategy);

    traversal = sg.V().outE().iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);
    traversal = sg.V().outE().inV().iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);
    traversal = sg.V().out().iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);

    traversal = sg.V().bothE().iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);
    traversal = sg.V().bothE().otherV().iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_LP_O_P_S_SE_SL_TraverserGenerator);
    traversal = sg.V().both().iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_LP_O_P_S_SE_SL_TraverserGenerator);

    traversal = sg.V().flatMap(bothE()).iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_O_TraverserGenerator);
    traversal = sg.V().flatMap(bothE().otherV()).iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_LP_O_P_S_SE_SL_TraverserGenerator);
    traversal = sg.V().flatMap(both()).iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_LP_O_P_S_SE_SL_TraverserGenerator);

    strategy = SubgraphStrategy.build().vertices(__.filter(__.simplePath())).create();
    sg = g.withStrategies(strategy);

    traversal = sg.V().out().iterate().asAdmin();
    assertTrue(traversal.getTraverserGenerator() instanceof B_LP_O_P_S_SE_SL_TraverserGenerator);
}

Source File: TinkerGraphStepStrategyTest.java From tinkerpop with Apache License 2.0

5 votes

private static GraphTraversal.Admin<?, ?> g_V(final Object... hasKeyValues) {
    final GraphTraversal.Admin<?, ?> traversal = new DefaultGraphTraversal<>();
    final TinkerGraphStep<Vertex, Vertex> graphStep = new TinkerGraphStep<>(new GraphStep<>(traversal, Vertex.class, true));
    for (int i = 0; i < hasKeyValues.length; i = i + 2) {
        graphStep.addHasContainer(new HasContainer((String) hasKeyValues[i], (P) hasKeyValues[i + 1]));
    }
    return traversal.addStep(graphStep);
}

Source File: TinkerGraphStepStrategyTest.java From tinkergraph-gremlin with Apache License 2.0

5 votes

private static GraphTraversal.Admin<?, ?> g_V(final Object... hasKeyValues) {
    final GraphTraversal.Admin<?, ?> traversal = new DefaultGraphTraversal<>();
    final TinkerGraphStep<Vertex, Vertex> graphStep = new TinkerGraphStep<>(new GraphStep<>(traversal, Vertex.class, true));
    for (int i = 0; i < hasKeyValues.length; i = i + 2) {
        graphStep.addHasContainer(new HasContainer((String) hasKeyValues[i], (P) hasKeyValues[i + 1]));
    }
    return traversal.addStep(graphStep);
}

Source File: SparqlTraversalSource.java From tinkerpop with Apache License 2.0

5 votes

/**
 * The start step for a SPARQL based traversal that accepts a string representation of the query to execute.
 */
public <S> GraphTraversal<S,?> sparql(final String query) {
    final SparqlTraversalSource clone = this.withStrategies(SparqlStrategy.instance()).clone();

    // the inject() holds the sparql which the SparqlStrategy then detects and converts to a traversal
    clone.bytecode.addStep(GraphTraversal.Symbols.inject, query);
    final GraphTraversal.Admin<S, S> traversal = new DefaultGraphTraversal(clone);
    return traversal.addStep(new InjectStep<>(traversal, query));
}

Source File: Neo4jGraphStepStrategyTest.java From tinkerpop with Apache License 2.0

5 votes

private static GraphTraversal.Admin<?, ?> g_V(final Object... hasKeyValues) {
    final GraphTraversal.Admin<?, ?> traversal = new DefaultGraphTraversal<>();
    final Neo4jGraphStep<Vertex, Vertex> graphStep = new Neo4jGraphStep<>(new GraphStep<>(traversal, Vertex.class, true));
    for (int i = 0; i < hasKeyValues.length; i = i + 2) {
        graphStep.addHasContainer(new HasContainer((String) hasKeyValues[i], (P) hasKeyValues[i + 1]));
    }
    return traversal.addStep(graphStep);
}

Source File: GroovyTranslatorTest.java From tinkerpop with Apache License 2.0

4 votes

@Test
public void shouldSupportStringSupplierLambdas() {
    final TinkerGraph graph = TinkerFactory.createModern();
    GraphTraversalSource g = graph.traversal();
    g = g.withStrategies(new TranslationStrategy(g, GroovyTranslator.of("g"), false));
    final GraphTraversal.Admin<Vertex, Integer> t = g.withSideEffect("lengthSum", 0).withSack(1)
            .V()
            .filter(Lambda.predicate("it.get().label().equals('person')"))
            .flatMap(Lambda.function("it.get().vertices(Direction.OUT)"))
            .map(Lambda.<Traverser<Object>, Integer>function("it.get().value('name').length()"))
            .sideEffect(Lambda.consumer("{ x -> x.sideEffects(\"lengthSum\", x.<Integer>sideEffects('lengthSum') + x.get()) }"))
            .order().by(Lambda.comparator("a,b -> a <=> b"))
            .sack(Lambda.biFunction("{ a,b -> a + b }"))
            .asAdmin();
    final List<Integer> sacks = new ArrayList<>();
    final List<Integer> lengths = new ArrayList<>();
    while (t.hasNext()) {
        final Traverser.Admin<Integer> traverser = t.nextTraverser();
        sacks.add(traverser.sack());
        lengths.add(traverser.get());
    }
    assertFalse(t.hasNext());
    //
    assertEquals(6, lengths.size());
    assertEquals(3, lengths.get(0).intValue());
    assertEquals(3, lengths.get(1).intValue());
    assertEquals(3, lengths.get(2).intValue());
    assertEquals(4, lengths.get(3).intValue());
    assertEquals(5, lengths.get(4).intValue());
    assertEquals(6, lengths.get(5).intValue());
    ///
    assertEquals(6, sacks.size());
    assertEquals(4, sacks.get(0).intValue());
    assertEquals(4, sacks.get(1).intValue());
    assertEquals(4, sacks.get(2).intValue());
    assertEquals(5, sacks.get(3).intValue());
    assertEquals(6, sacks.get(4).intValue());
    assertEquals(7, sacks.get(5).intValue());
    //
    assertEquals(24, t.getSideEffects().<Number>get("lengthSum").intValue());

    final String script = GroovyTranslator.of("g").translate(t.getBytecode()).getScript();
    assertEquals("g.withStrategies(org.apache.tinkerpop.gremlin.process.traversal.strategy.decoration.TranslationStrategy.instance())" +
                    ".withSideEffect(\"lengthSum\",(int) 0).withSack((int) 1)" +
                    ".V()" +
                    ".filter({it.get().label().equals('person')})" +
                    ".flatMap({it.get().vertices(Direction.OUT)})" +
                    ".map({it.get().value('name').length()})" +
                    ".sideEffect({ x -> x.sideEffects(\"lengthSum\", x.<Integer>sideEffects('lengthSum') + x.get()) })" +
                    ".order().by({a,b -> a <=> b})" +
                    ".sack({ a,b -> a + b })",
            script);
}

Source File: JavaTranslatorBenchmark.java From tinkerpop with Apache License 2.0

4 votes

@Benchmark
public GraphTraversal.Admin<Vertex,Vertex> testTranslationWithStrategy() {
    return translator.translate(g.withStrategies(ReadOnlyStrategy.instance())
            .withStrategies(SubgraphStrategy.build().vertices(hasLabel("person")).create())
            .V().out().in("link").out().in("link").asAdmin().getBytecode());
}

Source File: JavaTranslatorBenchmark.java From tinkerpop with Apache License 2.0

4 votes

@Benchmark
public GraphTraversal.Admin<Vertex,Vertex> testTranslationMedium() {
    return translator.translate(g.V().out().in("link").out().in("link").asAdmin().getBytecode());
}

Source File: JavaTranslatorBenchmark.java From tinkerpop with Apache License 2.0

4 votes

@Benchmark
public GraphTraversal.Admin<Vertex,Vertex> testTranslationShort() {
    return translator.translate(g.V().asAdmin().getBytecode());
}

Source File: GryoSerializerIntegrateTest.java From tinkerpop with Apache License 2.0

4 votes

@Test
public void shouldHaveAllRegisteredGryoSerializerClasses() throws Exception {
    // this is a stress test that ensures that when data is spilling to disk, persisted to an RDD, etc. the correct classes are registered with GryoSerializer.
    final TinkerGraph randomGraph = TinkerGraph.open();
    int totalVertices = 200000;
    TestHelper.createRandomGraph(randomGraph, totalVertices, 100);
    final String inputLocation = TestHelper.makeTestDataFile(GryoSerializerIntegrateTest.class,
                                                             UUID.randomUUID().toString(),
                                                             "random-graph.kryo");
    randomGraph.io(IoCore.gryo()).writeGraph(inputLocation);
    randomGraph.clear();
    randomGraph.close();

    final String outputLocation = TestHelper.makeTestDataDirectory(GryoSerializerIntegrateTest.class, UUID.randomUUID().toString());
    Configuration configuration = getBaseConfiguration();
    configuration.clearProperty(Constants.SPARK_SERIALIZER); // ensure proper default to GryoSerializer
    configuration.setProperty(Constants.GREMLIN_HADOOP_INPUT_LOCATION, inputLocation);
    configuration.setProperty(Constants.GREMLIN_HADOOP_GRAPH_READER, GryoInputFormat.class.getCanonicalName());
    configuration.setProperty(Constants.GREMLIN_HADOOP_GRAPH_WRITER, GryoOutputFormat.class.getCanonicalName());
    configuration.setProperty(Constants.GREMLIN_HADOOP_OUTPUT_LOCATION, outputLocation);
    configuration.setProperty(Constants.GREMLIN_SPARK_PERSIST_CONTEXT, false);
    Graph graph = GraphFactory.open(configuration);
    final GraphTraversal.Admin<Vertex, Map<Vertex, Collection<Vertex>>> traversal = graph.traversal().withComputer(SparkGraphComputer.class).V().group("m").<Map<Vertex, Collection<Vertex>>>cap("m").asAdmin();
    assertTrue(traversal.hasNext());
    assertEquals(traversal.next(), traversal.getSideEffects().get("m"));
    assertFalse(traversal.hasNext());
    assertTrue(traversal.getSideEffects().exists("m"));
    assertTrue(traversal.getSideEffects().get("m") instanceof Map);
    assertEquals(totalVertices, traversal.getSideEffects().<Map>get("m").size());

    configuration = getBaseConfiguration();
    configuration.clearProperty(Constants.SPARK_SERIALIZER); // ensure proper default to GryoSerializer
    configuration.setProperty(Constants.GREMLIN_HADOOP_INPUT_LOCATION, inputLocation);
    configuration.setProperty(Constants.GREMLIN_HADOOP_GRAPH_READER, GryoInputFormat.class.getCanonicalName());
    configuration.setProperty(Constants.GREMLIN_HADOOP_GRAPH_WRITER, PersistedOutputRDD.class.getCanonicalName());
    configuration.setProperty(Constants.GREMLIN_SPARK_PERSIST_STORAGE_LEVEL, "DISK_ONLY");
    configuration.setProperty(Constants.GREMLIN_HADOOP_OUTPUT_LOCATION, "persisted-rdd");
    configuration.setProperty(Constants.GREMLIN_SPARK_PERSIST_CONTEXT, true);
    graph = GraphFactory.open(configuration);
    assertEquals(totalVertices, graph.compute(SparkGraphComputer.class).program(PageRankVertexProgram.build().iterations(2).create(graph)).submit().get().graph().traversal().V().count().next().longValue());

    configuration = getBaseConfiguration();
    configuration.clearProperty(Constants.SPARK_SERIALIZER); // ensure proper default to GryoSerializer
    configuration.setProperty(Constants.GREMLIN_HADOOP_INPUT_LOCATION, "persisted-rdd");
    configuration.setProperty(Constants.GREMLIN_HADOOP_GRAPH_READER, PersistedInputRDD.class.getCanonicalName());
    configuration.setProperty(Constants.GREMLIN_HADOOP_GRAPH_WRITER, GryoOutputFormat.class.getCanonicalName());
    configuration.setProperty(Constants.GREMLIN_HADOOP_OUTPUT_LOCATION, outputLocation);
    configuration.setProperty(Constants.GREMLIN_SPARK_PERSIST_CONTEXT, true);
    graph = GraphFactory.open(configuration);
    assertEquals(totalVertices, graph.traversal().withComputer(SparkGraphComputer.class).V().count().next().longValue());

    configuration = getBaseConfiguration();
    configuration.setProperty(Constants.GREMLIN_HADOOP_INPUT_LOCATION, "persisted-rdd");
    configuration.setProperty(Constants.GREMLIN_HADOOP_GRAPH_READER, PersistedInputRDD.class.getCanonicalName());
    configuration.setProperty(Constants.GREMLIN_HADOOP_GRAPH_WRITER, PersistedOutputRDD.class.getCanonicalName());
    configuration.setProperty(Constants.GREMLIN_HADOOP_OUTPUT_LOCATION, outputLocation);
    configuration.setProperty(Constants.GREMLIN_SPARK_GRAPH_STORAGE_LEVEL, "MEMORY_ONLY"); // this should be ignored as you can't change the persistence level once created
    configuration.setProperty(Constants.GREMLIN_SPARK_PERSIST_STORAGE_LEVEL, "MEMORY_AND_DISK");
    configuration.setProperty(Constants.GREMLIN_SPARK_PERSIST_CONTEXT, true);
    graph = GraphFactory.open(configuration);
    assertEquals(totalVertices, graph.traversal().withComputer(SparkGraphComputer.class).V().count().next().longValue());
}

Source File: ParameterizedGroovyTranslatorTest.java From tinkerpop with Apache License 2.0

4 votes

@Test
public void shouldSupportStringSupplierLambdas() {
    final TinkerGraph graph = TinkerFactory.createModern();
    GraphTraversalSource g = graph.traversal();
    g = g.withStrategies(new TranslationStrategy(g, GroovyTranslator.of("g", true), false));
    final GraphTraversal.Admin<Vertex, Integer> t = g.withSideEffect("lengthSum", 0).withSack(1)
            .V()
            .filter(Lambda.predicate("it.get().label().equals('person')"))
            .flatMap(Lambda.function("it.get().vertices(Direction.OUT)"))
            .map(Lambda.<Traverser<Object>, Integer>function("it.get().value('name').length()"))
            .sideEffect(Lambda.consumer("{ x -> x.sideEffects(\"lengthSum\", x.<Integer>sideEffects('lengthSum') + x.get()) }"))
            .order().by(Lambda.comparator("a,b -> a <=> b"))
            .sack(Lambda.biFunction("{ a,b -> a + b }"))
            .asAdmin();

    final List<Integer> sacks = new ArrayList<>();
    final List<Integer> lengths = new ArrayList<>();
    while (t.hasNext()) {
        final Traverser.Admin<Integer> traverser = t.nextTraverser();
        sacks.add(traverser.sack());
        lengths.add(traverser.get());
    }
    assertFalse(t.hasNext());
    //
    assertEquals(6, lengths.size());
    assertEquals(3, lengths.get(0).intValue());
    assertEquals(3, lengths.get(1).intValue());
    assertEquals(3, lengths.get(2).intValue());
    assertEquals(4, lengths.get(3).intValue());
    assertEquals(5, lengths.get(4).intValue());
    assertEquals(6, lengths.get(5).intValue());
    ///
    assertEquals(6, sacks.size());
    assertEquals(4, sacks.get(0).intValue());
    assertEquals(4, sacks.get(1).intValue());
    assertEquals(4, sacks.get(2).intValue());
    assertEquals(5, sacks.get(3).intValue());
    assertEquals(6, sacks.get(4).intValue());
    assertEquals(7, sacks.get(5).intValue());
    //
    assertEquals(24, t.getSideEffects().<Number>get("lengthSum").intValue());

    final Script script = GroovyTranslator.of("g", true).translate(t.getBytecode());
    Bindings bindings = new SimpleBindings();
    script.getParameters().ifPresent(bindings::putAll);
    assertEquals(9, bindings.size());
    assertEquals("lengthSum", bindings.get("_args_0"));
    assertEquals(Integer.valueOf(0), bindings.get("_args_1"));
    assertEquals(Integer.valueOf(1), bindings.get("_args_2"));
    assertEquals(Lambda.predicate("it.get().label().equals('person')"), bindings.get("_args_3"));
    assertEquals(Lambda.function("it.get().vertices(Direction.OUT)"), bindings.get("_args_4"));
    assertEquals(Lambda.<Traverser<Object>, Integer>function("it.get().value('name').length()"), bindings.get("_args_5"));
    assertEquals(Lambda.consumer("{ x -> x.sideEffects(\"lengthSum\", x.<Integer>sideEffects('lengthSum') + x.get()) }"), bindings.get("_args_6"));
    assertEquals(Lambda.comparator("a,b -> a <=> b"), bindings.get("_args_7"));
    assertEquals(Lambda.biFunction("{ a,b -> a + b }"), bindings.get("_args_8"));
    assertEquals("g.withStrategies(org.apache.tinkerpop.gremlin.process.traversal.strategy.decoration.TranslationStrategy.instance())" +
                    ".withSideEffect(_args_0,_args_1).withSack(_args_2)" +
                    ".V()" +
                    ".filter(_args_3)" +
                    ".flatMap(_args_4)" +
                    ".map(_args_5)" +
                    ".sideEffect(_args_6)" +
                    ".order().by(_args_7)" +
                    ".sack(_args_8)",
            script.getScript());
}

Source File: VertexProgramStrategyTest.java From tinkerpop with Apache License 2.0

4 votes

private static GraphTraversal.Admin<?, ?> start() {
    return new DefaultGraphTraversal<>().asAdmin();
}

Source File: TitanGraphStepStrategyTest.java From titan1withtp3.1 with Apache License 2.0

4 votes

@Test
@IgnoreEngine(TraversalEngine.Type.COMPUTER)
public void shouldFoldInHasContainers() {
    GraphTraversal.Admin traversal = g.V().has("name", "marko").asAdmin();
    assertEquals(2, traversal.getSteps().size());
    assertEquals(HasStep.class, traversal.getEndStep().getClass());
    traversal.applyStrategies();
    assertEquals(1, traversal.getSteps().size());
    assertEquals(TitanGraphStep.class, traversal.getStartStep().getClass());
    assertEquals(TitanGraphStep.class, traversal.getEndStep().getClass());
    assertEquals(1, ((TitanGraphStep) traversal.getStartStep()).getHasContainers().size());
    assertEquals("name", ((TitanGraphStep<?, ?>) traversal.getStartStep()).getHasContainers().get(0).getKey());
    assertEquals("marko", ((TitanGraphStep<?, ?>) traversal.getStartStep()).getHasContainers().get(0).getValue());
    ////
    traversal = g.V().has("name", "marko").has("age", P.gt(20)).asAdmin();
    traversal.applyStrategies();
    assertEquals(1, traversal.getSteps().size());
    assertEquals(TitanGraphStep.class, traversal.getStartStep().getClass());
    assertEquals(2, ((TitanGraphStep) traversal.getStartStep()).getHasContainers().size());
    ////
    traversal = g.V().has("name", "marko").out().has("name", "daniel").asAdmin();
    traversal.applyStrategies();
    assertEquals(3, traversal.getSteps().size());
    assertEquals(TitanGraphStep.class, traversal.getStartStep().getClass());
    assertEquals(1, ((TitanGraphStep) traversal.getStartStep()).getHasContainers().size());
    assertEquals("name", ((TitanGraphStep<?, ?>) traversal.getStartStep()).getHasContainers().get(0).getKey());
    assertEquals("marko", ((TitanGraphStep<?, ?>) traversal.getStartStep()).getHasContainers().get(0).getValue());
    assertEquals(HasStep.class, traversal.getEndStep().getClass());
    ////
    traversal = g.V().has("name", "marko").out().V().has("name", "daniel").asAdmin();
    traversal.applyStrategies();
    assertEquals(3, traversal.getSteps().size());
    assertEquals(TitanGraphStep.class, traversal.getStartStep().getClass());
    assertEquals(1, ((TitanGraphStep) traversal.getStartStep()).getHasContainers().size());
    assertEquals("name", ((TitanGraphStep<?, ?>) traversal.getStartStep()).getHasContainers().get(0).getKey());
    assertEquals("marko", ((TitanGraphStep<?, ?>) traversal.getStartStep()).getHasContainers().get(0).getValue());
    assertEquals(TitanGraphStep.class, traversal.getSteps().get(2).getClass());
    assertEquals(1, ((TitanGraphStep) traversal.getSteps().get(2)).getHasContainers().size());
    assertEquals("name", ((TitanGraphStep<?, ?>) traversal.getSteps().get(2)).getHasContainers().get(0).getKey());
    assertEquals("daniel", ((TitanGraphStep<?,?>) traversal.getSteps().get(2)).getHasContainers().get(0).getValue());
    assertEquals(TitanGraphStep.class, traversal.getEndStep().getClass());
}

Source File: Neo4jGraph.java From tinkerpop with Apache License 2.0

3 votes

/**
 * Execute the Cypher query with provided parameters and get the result set as a {@link GraphTraversal}.
 *
 * @param query      the Cypher query to execute
 * @param parameters the parameters of the Cypher query
 * @return a fluent Gremlin traversal
 */
public <S, E> GraphTraversal<S, E> cypher(final String query, final Map<String, Object> parameters) {
    this.tx().readWrite();
    final GraphTraversal.Admin<S, E> traversal = new DefaultGraphTraversal<>(this);
    traversal.addStep(new CypherStartStep(traversal, query, new Neo4jCypherIterator<>((Iterator) this.baseGraph.execute(query, parameters), this)));
    return traversal;
}

Java Code Examples for org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal#Admin