Examples with BoltTransaction org.neo4j.bolt.dbapi.BoltTransaction used on opensource projects

Example 6 with BoltTransaction

use of org.neo4j.bolt.dbapi.BoltTransaction in project neo4j by neo4j.

the class TransactionStateMachineTest method shouldResetInExplicitTransactionWhileStatementIsRunningWhenValidated.

@Test
void shouldResetInExplicitTransactionWhileStatementIsRunningWhenValidated() throws Exception {
    BoltTransaction transaction = newTimedOutTransaction();
    TransactionStateMachineSPI stateMachineSPI = newTransactionStateMachineSPI(transaction);
    TransactionStateMachine stateMachine = newTransactionStateMachine(stateMachineSPI);
    // start an explicit transaction
    beginTx(stateMachine);
    assertThat(stateMachine.state).isEqualTo(TransactionStateMachine.State.EXPLICIT_TRANSACTION);
    assertNotNull(stateMachine.ctx.currentTransaction);
    stateMachine.run("RETURN 1", null);
    // verify transaction, which is timed out
    stateMachine.validateTransaction();
    assertThat(stateMachine.state).isEqualTo(TransactionStateMachine.State.AUTO_COMMIT);
    assertNull(stateMachine.ctx.currentTransaction);
    assertThat(stateMachine.ctx.statementOutcomes.entrySet()).hasSize(0);
    verify(transaction).getReasonIfTerminated();
    verify(transaction).rollback();
}

Example 7 with BoltTransaction

use of org.neo4j.bolt.dbapi.BoltTransaction in project neo4j by neo4j.

the class TransactionStateMachineTest method newTimedOutTransaction.

private static BoltTransaction newTimedOutTransaction() {
    BoltTransaction transaction = newTransaction();
    when(transaction.getReasonIfTerminated()).thenReturn(Optional.of(Status.Transaction.TransactionTimedOut));
    return transaction;
}

Example 8 with BoltTransaction

use of org.neo4j.bolt.dbapi.BoltTransaction in project neo4j by neo4j.

the class TransactionStateMachineTest method shouldDoNothingInAutoCommitTransactionUponInitialisationWhenValidated.

@Test
void shouldDoNothingInAutoCommitTransactionUponInitialisationWhenValidated() throws Exception {
    BoltTransaction transaction = newTimedOutTransaction();
    TransactionStateMachineSPI stateMachineSPI = newTransactionStateMachineSPI(transaction);
    TransactionStateMachine stateMachine = newTransactionStateMachine(stateMachineSPI);
    // We're in auto-commit state
    assertThat(stateMachine.state).isEqualTo(TransactionStateMachine.State.AUTO_COMMIT);
    assertNull(stateMachine.ctx.currentTransaction);
    // call validate transaction
    stateMachine.validateTransaction();
    assertThat(stateMachine.state).isEqualTo(TransactionStateMachine.State.AUTO_COMMIT);
    assertNull(stateMachine.ctx.currentTransaction);
    verify(transaction, never()).getReasonIfTerminated();
    verify(transaction, never()).rollback();
}

Example 9 with BoltTransaction

use of org.neo4j.bolt.dbapi.BoltTransaction in project neo4j by neo4j.

the class TransactionStateMachineTest method shouldTryToTerminateAllActiveStatements.

@Test
void shouldTryToTerminateAllActiveStatements() throws Exception {
    BoltTransaction transaction = newTimedOutTransaction();
    BoltResultHandle resultHandle = newResultHandle();
    doThrow(new RuntimeException("You shall not pass")).doThrow(new RuntimeException("Not pass twice")).when(resultHandle).terminate();
    TransactionStateMachineSPI stateMachineSPI = mock(TransactionStateMachineSPI.class);
    when(stateMachineSPI.beginTransaction(any(), any(), any(), any(), any(), any())).thenReturn(transaction);
    when(stateMachineSPI.executeQuery(any(), anyString(), any())).thenReturn(resultHandle);
    // V4
    when(stateMachineSPI.supportsNestedStatementsInTransaction()).thenReturn(true);
    TransactionStateMachine stateMachine = newTransactionStateMachine(stateMachineSPI);
    // We're in explicit-commit state
    beginTx(stateMachine, List.of());
    assertThat(stateMachine.state).isEqualTo(TransactionStateMachine.State.EXPLICIT_TRANSACTION);
    assertNotNull(stateMachine.ctx.currentTransaction);
    // We run two statements
    stateMachine.run("RETURN 1", null);
    stateMachine.run("RETURN 2", null);
    assertThat(stateMachine.state).isEqualTo(TransactionStateMachine.State.EXPLICIT_TRANSACTION);
    assertNotNull(stateMachine.ctx.currentTransaction);
    assertThat(stateMachine.ctx.statementCounter).isEqualTo(2);
    RuntimeException error = assertThrows(RuntimeException.class, () -> stateMachine.reset());
    assertThat(error.getCause().getMessage()).isEqualTo("You shall not pass");
    assertThat(error.getSuppressed().length).isEqualTo(1);
    assertThat(error.getSuppressed()[0].getMessage()).isEqualTo("Not pass twice");
}

Example 10 with BoltTransaction

use of org.neo4j.bolt.dbapi.BoltTransaction in project neo4j by neo4j.

the class TransactionStateMachineTest method shouldCloseResultHandlesWhenExecutionFailsInExplicitTransaction.

@Test
void shouldCloseResultHandlesWhenExecutionFailsInExplicitTransaction() throws Exception {
    BoltTransaction transaction = newTransaction();
    BoltResultHandle resultHandle = newResultHandle(new RuntimeException("some error"));
    TransactionStateMachineSPI stateMachineSPI = newTransactionStateMachineSPI(transaction, resultHandle);
    TransactionStateMachine stateMachine = newTransactionStateMachine(stateMachineSPI);
    RuntimeException e = assertThrows(RuntimeException.class, () -> {
        beginTx(stateMachine);
        stateMachine.run("SOME STATEMENT", null);
    });
    assertEquals("some error", e.getMessage());
    assertThat(stateMachine.ctx.statementOutcomes.entrySet()).hasSize(0);
    assertNotNull(stateMachine.ctx.currentTransaction);
}