Search in sources :

Example 1 with RunMessage

use of org.neo4j.bolt.v4.messaging.RunMessage in project neo4j by neo4j.

the class BoltV43TransportIT method shouldReturnUpdatedBookmarkAfterAutoCommitTransaction.

@ParameterizedTest(name = "{0}")
@MethodSource("argumentsProvider")
public void shouldReturnUpdatedBookmarkAfterAutoCommitTransaction(Class<? extends TransportConnection> connectionClass) throws Exception {
    init(connectionClass);
    assumeFalse(FabricDatabaseManager.fabricByDefault(Config.defaults()));
    negotiateBoltV43();
    // bookmark is expected to advance once the auto-commit transaction is committed
    var lastClosedTransactionId = getLastClosedTransactionId();
    var expectedBookmark = new BookmarkWithDatabaseId(lastClosedTransactionId + 1, getDatabaseId()).toString();
    connection.send(util.chunk(new RunMessage("CREATE ()"), new PullMessage(asMapValue(map("n", -1L)))));
    assertThat(connection).satisfies(util.eventuallyReceives(msgSuccess(), msgSuccess(responseMessage -> assertThat(responseMessage).containsEntry("bookmark", expectedBookmark))));
}
Also used : BookmarkWithDatabaseId(org.neo4j.bolt.v4.runtime.bookmarking.BookmarkWithDatabaseId) PullMessage(org.neo4j.bolt.v4.messaging.PullMessage) RunMessage(org.neo4j.bolt.v4.messaging.RunMessage) ParameterizedTest(org.junit.jupiter.params.ParameterizedTest) MethodSource(org.junit.jupiter.params.provider.MethodSource)

Example 2 with RunMessage

use of org.neo4j.bolt.v4.messaging.RunMessage in project neo4j by neo4j.

the class FailedStateIT method getBoltStateMachineInFailedState.

private BoltStateMachineV4 getBoltStateMachineInFailedState() throws BoltConnectionFatality, InterruptedException {
    BoltStateMachineV4 machine = newStateMachine();
    machine.process(newHelloMessage(), nullResponseHandler());
    RunMessage runMessage = mock(RunMessage.class);
    when(runMessage.databaseName()).thenReturn(ABSENT_DB_NAME);
    when(runMessage.statement()).thenThrow(new RuntimeException("error here"));
    BoltResponseRecorder recorder = new BoltResponseRecorder();
    machine.process(runMessage, recorder);
    assertThat(recorder.nextResponse()).satisfies(failedWithStatus(Status.General.UnknownError));
    assertThat(machine.state()).isInstanceOf(FailedState.class);
    return machine;
}
Also used : BoltStateMachineV4(org.neo4j.bolt.v4.BoltStateMachineV4) BoltResponseRecorder(org.neo4j.bolt.testing.BoltResponseRecorder) RunMessage(org.neo4j.bolt.v4.messaging.RunMessage)

Example 3 with RunMessage

use of org.neo4j.bolt.v4.messaging.RunMessage in project neo4j by neo4j.

the class ReadyStateIT method shouldMoveToFailedStateOnRun_fail.

@Test
void shouldMoveToFailedStateOnRun_fail() throws Throwable {
    // Given
    BoltStateMachineV4 machine = newStateMachine();
    machine.process(newHelloMessage(), nullResponseHandler());
    // When
    BoltResponseRecorder recorder = new BoltResponseRecorder();
    RunMessage runMessage = mock(RunMessage.class);
    when(runMessage.databaseName()).thenReturn(ABSENT_DB_NAME);
    when(runMessage.statement()).thenThrow(new RuntimeException("Fail"));
    machine.process(runMessage, recorder);
    // Then
    assertThat(recorder.nextResponse()).satisfies(failedWithStatus(Status.General.UnknownError));
    assertThat(machine.state()).isInstanceOf(FailedState.class);
}
Also used : BoltStateMachineV4(org.neo4j.bolt.v4.BoltStateMachineV4) BoltResponseRecorder(org.neo4j.bolt.testing.BoltResponseRecorder) RunMessage(org.neo4j.bolt.v4.messaging.RunMessage) Test(org.junit.jupiter.api.Test) ParameterizedTest(org.junit.jupiter.params.ParameterizedTest)

Example 4 with RunMessage

use of org.neo4j.bolt.v4.messaging.RunMessage in project neo4j by neo4j.

the class BoltV41TransportIT method shouldReturnUpdatedBookmarkAfterExplicitTransaction.

@ParameterizedTest(name = "{0}")
@MethodSource("argumentsProvider")
public void shouldReturnUpdatedBookmarkAfterExplicitTransaction(Class<? extends TransportConnection> connectionClass) throws Exception {
    init(connectionClass);
    negotiateBoltV41();
    // bookmark is expected to advance once the auto-commit transaction is committed
    var lastClosedTransactionId = getLastClosedTransactionId();
    var expectedBookmark = new BookmarkWithDatabaseId(lastClosedTransactionId + 1, getDatabaseId()).toString();
    connection.send(util.chunk(new BeginMessage()));
    assertThat(connection).satisfies(util.eventuallyReceives(msgSuccess()));
    connection.send(util.chunk(new RunMessage("CREATE ()"), new PullMessage(asMapValue(map("n", -1L)))));
    assertThat(connection).satisfies(util.eventuallyReceives(msgSuccess(), msgSuccess(message -> assertThat(message).doesNotContainEntry("bookmark", expectedBookmark))));
    connection.send(util.chunk(CommitMessage.COMMIT_MESSAGE));
    assertThat(connection).satisfies(util.eventuallyReceives(msgSuccess(message -> assertThat(message).containsEntry("bookmark", expectedBookmark))));
}
Also used : BeginMessage(org.neo4j.bolt.v4.messaging.BeginMessage) BookmarkWithDatabaseId(org.neo4j.bolt.v4.runtime.bookmarking.BookmarkWithDatabaseId) PullMessage(org.neo4j.bolt.v4.messaging.PullMessage) RunMessage(org.neo4j.bolt.v4.messaging.RunMessage) ParameterizedTest(org.junit.jupiter.params.ParameterizedTest) MethodSource(org.junit.jupiter.params.provider.MethodSource)

Example 5 with RunMessage

use of org.neo4j.bolt.v4.messaging.RunMessage in project neo4j by neo4j.

the class BoltV41TransportIT method shouldReturnUpdatedBookmarkAfterAutoCommitTransaction.

@ParameterizedTest(name = "{0}")
@MethodSource("argumentsProvider")
public void shouldReturnUpdatedBookmarkAfterAutoCommitTransaction(Class<? extends TransportConnection> connectionClass) throws Exception {
    init(connectionClass);
    negotiateBoltV41();
    // bookmark is expected to advance once the auto-commit transaction is committed
    var lastClosedTransactionId = getLastClosedTransactionId();
    var expectedBookmark = new BookmarkWithDatabaseId(lastClosedTransactionId + 1, getDatabaseId()).toString();
    connection.send(util.chunk(new RunMessage("CREATE ()"), new PullMessage(asMapValue(map("n", -1L)))));
    assertThat(connection).satisfies(util.eventuallyReceives(msgSuccess(), msgSuccess(responseMessage -> assertThat(responseMessage).containsEntry("bookmark", expectedBookmark))));
}
Also used : BookmarkWithDatabaseId(org.neo4j.bolt.v4.runtime.bookmarking.BookmarkWithDatabaseId) PullMessage(org.neo4j.bolt.v4.messaging.PullMessage) RunMessage(org.neo4j.bolt.v4.messaging.RunMessage) ParameterizedTest(org.junit.jupiter.params.ParameterizedTest) MethodSource(org.junit.jupiter.params.provider.MethodSource)

Aggregations

RunMessage (org.neo4j.bolt.v4.messaging.RunMessage)22 ParameterizedTest (org.junit.jupiter.params.ParameterizedTest)15 MethodSource (org.junit.jupiter.params.provider.MethodSource)9 BoltStateMachineV4 (org.neo4j.bolt.v4.BoltStateMachineV4)9 PullMessage (org.neo4j.bolt.v4.messaging.PullMessage)9 BeginMessage (org.neo4j.bolt.v4.messaging.BeginMessage)8 BookmarkWithDatabaseId (org.neo4j.bolt.v4.runtime.bookmarking.BookmarkWithDatabaseId)8 BoltResponseRecorder (org.neo4j.bolt.testing.BoltResponseRecorder)7 Test (org.junit.jupiter.api.Test)6 RecordedBoltResponse (org.neo4j.bolt.testing.RecordedBoltResponse)2 ByteBuf (io.netty.buffer.ByteBuf)1 EmbeddedChannel (io.netty.channel.embedded.EmbeddedChannel)1 IOException (java.io.IOException)1 BoltChannel (org.neo4j.bolt.BoltChannel)1 BoltProtocol (org.neo4j.bolt.BoltProtocol)1 RequestMessage (org.neo4j.bolt.messaging.RequestMessage)1 BoltResponseHandler (org.neo4j.bolt.runtime.BoltResponseHandler)1 SynchronousBoltConnection (org.neo4j.bolt.runtime.SynchronousBoltConnection)1 BoltStateMachine (org.neo4j.bolt.runtime.statemachine.BoltStateMachine)1 BoltTestUtil.newTestBoltChannel (org.neo4j.bolt.testing.BoltTestUtil.newTestBoltChannel)1