Examples with GraphNode com.datastax.dse.driver.api.core.graph.GraphNode used on opensource projects

Example 36 with GraphNode

use of com.datastax.dse.driver.api.core.graph.GraphNode in project java-driver by datastax.

the class GraphTraversalITBase method should_resolve_path_with_some_labels.

/**
 * Validates that when traversing a path and labeling some of the elements during the traversal
 * that the output elements are properly labeled.
 *
 * @test_category dse:graph
 */
@Test
public void should_resolve_path_with_some_labels() {
    GraphResultSet rs = session().execute(newInstance(graphTraversalSource().V().hasLabel("person").has("name", "marko").as("a").outE("knows").inV().as("c", "d").outE("created").as("e", "f", "g").inV().path()));
    List<GraphNode> results = rs.all();
    assertThat(results.size()).isEqualTo(2);
    for (GraphNode result : results) {
        Path path = result.asPath();
        validatePathObjects(path);
        assertThat(path.labels()).hasSize(5);
        assertThat(path).hasLabel(0, "a").hasNoLabel(1).hasLabel(2, "c", "d").hasLabel(3, "e", "f", "g").hasNoLabel(4);
    }
}

Example 37 with GraphNode

use of com.datastax.dse.driver.api.core.graph.GraphNode in project java-driver by datastax.

the class GraphTraversalITBase method should_use_edge_id_as_parameter.

/**
 * Ensures that a previously returned {@link Edge}'s {@link Edge#id()} can be used as an input to
 * {@link
 * org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource#E(Object...)} to
 * retrieve the {@link Edge} and that the returned {@link Edge} is the same.
 *
 * @test_category dse:graph
 */
@Test
public void should_use_edge_id_as_parameter() {
    GraphTraversal<Edge, Edge> query = graphTraversalSource().E().has("weight", 0.2f);
    GraphResultSet resultSet = session().execute(newInstance(query));
    List<GraphNode> results = resultSet.all();
    assertThat(results.size()).isEqualTo(1);
    Edge created = results.get(0).asEdge();
    if (isGraphBinary()) {
        Map<Object, Object> properties = session().execute(newInstance(query.elementMap("weight", "software", "person"))).one().asMap();
        assertThatContainsProperties(properties, "weight", 0.2f);
        assertThatContainsLabel(properties, Direction.IN, "software");
        assertThatContainsLabel(properties, Direction.OUT, "person");
    } else {
        assertThat(created).hasProperty("weight", 0.2f).hasInVLabel("software").hasOutVLabel("person");
    }
    if (isGraphBinary()) {
        Map<Object, Object> inProperties = session().execute(newInstance(graphTraversalSource().E(created.id()).inV().elementMap("name", "lang"))).one().asMap();
        assertThatContainsProperties(inProperties, "name", "lop", "lang", "java");
    } else {
        resultSet = session().execute(newInstance(graphTraversalSource().E(created.id()).inV()));
        results = resultSet.all();
        assertThat(results.size()).isEqualTo(1);
        Vertex lop = results.get(0).asVertex();
        assertThat(lop).hasLabel("software").hasProperty("name", "lop").hasProperty("lang", "java");
    }
}

Example 38 with GraphNode

use of com.datastax.dse.driver.api.core.graph.GraphNode in project java-driver by datastax.

the class GraphTraversalITBase method should_handle_asynchronous_execution_graphson.

@Test
public void should_handle_asynchronous_execution_graphson() {
    Assumptions.assumeThat(isGraphBinary()).isFalse();
    StringBuilder names = new StringBuilder();
    CompletionStage<AsyncGraphResultSet> future = session().executeAsync(FluentGraphStatement.newInstance(graphTraversalSource().V().hasLabel("person")));
    try {
        // dumb processing to make sure the completable future works correctly and correct results are
        // returned
        Iterable<GraphNode> results = future.thenApply(AsyncGraphResultSet::currentPage).toCompletableFuture().get();
        for (GraphNode gn : results) {
            names.append(gn.asVertex().property("name").value());
        }
    } catch (InterruptedException | ExecutionException e) {
        fail("Shouldn't have thrown an exception waiting for the result to complete");
    }
    assertThat(names.toString()).contains("peter", "marko", "vadas", "josh");
}

Example 39 with GraphNode

use of com.datastax.dse.driver.api.core.graph.GraphNode in project java-driver by datastax.

the class GraphTraversalITBase method should_handle_subgraph_grap_binary.

/**
 * Ensures that a traversal that returns a sub graph can be retrieved.
 *
 * <p>The subgraph is all members in a knows relationship, thus is all people who marko knows and
 * the edges that connect them.
 */
@Test
public void should_handle_subgraph_grap_binary() {
    Assumptions.assumeThat(isGraphBinary()).isTrue();
    GraphResultSet rs = session().execute(newInstance(graphTraversalSource().E().hasLabel("knows").subgraph("subGraph").cap("subGraph")));
    List<GraphNode> results = rs.all();
    assertThat(results.size()).isEqualTo(1);
    String graph = results.get(0).as(String.class);
    assertThat(graph).contains("vertices:3").contains("edges:2");
}

Example 40 with GraphNode

use of com.datastax.dse.driver.api.core.graph.GraphNode in project java-driver by datastax.

the class GraphTraversalITBase method should_return_correct_results_when_bulked.

/**
 * Ensures that traversals with barriers (which return results bulked) contain the correct amount
 * of end results.
 *
 * <p>This will fail if ran against DSE < 5.0.9 or DSE < 5.1.2.
 */
@Test
public void should_return_correct_results_when_bulked() {
    Assumptions.assumeThat(ccmRule().getCcmBridge().getDseVersion().get().compareTo(Version.parse("5.1.2")) > 0).isTrue();
    GraphResultSet rs = session().execute(newInstance(graphTraversalSource().E().label().barrier()));
    List<String> results = rs.all().stream().map(GraphNode::asString).sorted().collect(Collectors.toList());
    assertThat(results).hasSize(6).containsSequence("created", "created", "created", "created").containsSequence("knows", "knows");
}