Examples with Query org.neo4j.driver.Query used on opensource projects

Example 1 with Query

use of org.neo4j.driver.Query in project neo4j by neo4j.

the class TableOutputFormatterTest method mockResult.

private Result mockResult(List<String> cols, Object... data) {
    Result result = mock(Result.class);
    Query query = mock(Query.class);
    ResultSummary summary = mock(ResultSummary.class);
    when(summary.query()).thenReturn(query);
    when(result.keys()).thenReturn(cols);
    List<Record> records = new ArrayList<>();
    List<Object> input = asList(data);
    int width = cols.size();
    for (int row = 0; row < input.size() / width; row++) {
        records.add(record(cols, input.subList(row * width, (row + 1) * width)));
    }
    when(result.list()).thenReturn(records);
    when(result.consume()).thenReturn(summary);
    when(result.consume()).thenReturn(summary);
    return result;
}

Example 2 with Query

use of org.neo4j.driver.Query in project neo4j by neo4j.

the class OutputFormatterTest method shouldReportTotalDBHits.

@Test
public void shouldReportTotalDBHits() {
    Value labelScan = buildOperator("NodeByLabelScan", 1002L, 1001L, null);
    Value filter = buildOperator("Filter", 1402, 280, labelScan);
    Value planMap = buildOperator("ProduceResults", 0, 280, filter);
    ProfiledPlan plan = PROFILED_PLAN_FROM_VALUE.apply(planMap);
    ResultSummary summary = new InternalResultSummary(new Query("PROFILE MATCH (n:LABEL) WHERE 20 < n.age < 35 return n"), new InternalServerInfo("agent", new BoltServerAddress("localhost:7687"), ServerVersion.vInDev, BoltProtocolV43.VERSION), new InternalDatabaseInfo("neo4j"), QueryType.READ_ONLY, null, plan, plan, Collections.emptyList(), 39, 55);
    // When
    Map<String, Value> info = OutputFormatter.info(summary);
    // Then
    assertThat(info.get("DbHits").asLong(), equalTo(2404L));
}