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

Example 26 with Driver

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

the class BoltStateHandlerTest method protocolVersionIsNotEmptyAfterConnect.

@Test
public void protocolVersionIsNotEmptyAfterConnect() throws CommandException {
    Driver driverMock = stubResultSummaryInAnOpenSession(mock(Result.class), mock(Session.class), "9.4.1-ALPHA");
    BoltStateHandler handler = new BoltStateHandler((s, authToken, config) -> driverMock, false);
    handler.connect(config);
    assertEquals("9.4.1-ALPHA", handler.getProtocolVersion());
}

Example 27 with Driver

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

the class BoltStateHandlerTest method whenInTransactionHandlerLetsTransactionDoTheWork.

@Test
public void whenInTransactionHandlerLetsTransactionDoTheWork() throws CommandException {
    Transaction transactionMock = mock(Transaction.class);
    Session sessionMock = mock(Session.class);
    when(sessionMock.beginTransaction()).thenReturn(transactionMock);
    Driver driverMock = stubResultSummaryInAnOpenSession(mock(Result.class), sessionMock, "neo4j-version");
    Result result = mock(Result.class);
    when(transactionMock.run(any(Query.class))).thenReturn(result);
    OfflineBoltStateHandler boltStateHandler = new OfflineBoltStateHandler(driverMock);
    boltStateHandler.connect();
    boltStateHandler.beginTransaction();
    BoltResult boltResult = boltStateHandler.runCypher("UNWIND [1,2] as num RETURN *", Collections.emptyMap()).get();
    assertEquals(result, boltResult.iterate());
    boltStateHandler.commitTransaction();
    assertFalse(boltStateHandler.isTransactionOpen());
}

Example 28 with Driver

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

the class BoltStateHandlerTest method actualDatabaseNameIsNotEmptyAfterConnect.

@Test
public void actualDatabaseNameIsNotEmptyAfterConnect() throws CommandException {
    Driver driverMock = stubResultSummaryInAnOpenSession(mock(Result.class), mock(Session.class), "9.4.1-ALPHA", "my_default_db");
    BoltStateHandler handler = new BoltStateHandler((s, authToken, config) -> driverMock, false);
    handler.connect(config);
    assertEquals("my_default_db", handler.getActualDatabaseAsReportedByServer());
}

Example 29 with Driver

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

the class BoltStateHandlerTest method shouldRunCypherQuery.

@Test
public void shouldRunCypherQuery() throws CommandException {
    Session sessionMock = mock(Session.class);
    Result resultMock = mock(Result.class);
    Record recordMock = mock(Record.class);
    Value valueMock = mock(Value.class);
    Driver driverMock = stubResultSummaryInAnOpenSession(resultMock, sessionMock, "neo4j-version");
    when(resultMock.list()).thenReturn(singletonList(recordMock));
    when(valueMock.toString()).thenReturn("999");
    when(recordMock.get(0)).thenReturn(valueMock);
    when(sessionMock.run(any(Query.class))).thenReturn(resultMock);
    OfflineBoltStateHandler boltStateHandler = new OfflineBoltStateHandler(driverMock);
    boltStateHandler.connect();
    BoltResult boltResult = boltStateHandler.runCypher("RETURN 999", new HashMap<>()).get();
    verify(sessionMock).run(any(Query.class));
    assertEquals("999", boltResult.getRecords().get(0).get(0).toString());
}

Example 30 with Driver

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

the class BoltStateHandlerTest method resetSessionOnReset.

@Test
public void resetSessionOnReset() throws Exception {
    // given
    Session sessionMock = mock(Session.class);
    Driver driverMock = stubResultSummaryInAnOpenSession(mock(Result.class), sessionMock, "neo4j-version");
    OfflineBoltStateHandler boltStateHandler = new OfflineBoltStateHandler(driverMock);
    boltStateHandler.connect();
    boltStateHandler.beginTransaction();
    // when
    boltStateHandler.reset();
    // then
    verify(sessionMock).reset();
}