Examples with TransactionStateMachineSPI org.neo4j.bolt.runtime.statemachine.TransactionStateMachineSPI used on opensource projects

Example 11 with TransactionStateMachineSPI

use of org.neo4j.bolt.runtime.statemachine.TransactionStateMachineSPI 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);
}

Example 12 with TransactionStateMachineSPI

use of org.neo4j.bolt.runtime.statemachine.TransactionStateMachineSPI in project neo4j by neo4j.

the class TransactionStateMachineTest method shouldCloseResultAndTransactionHandlesWhenExecutionFails.

@Test
void shouldCloseResultAndTransactionHandlesWhenExecutionFails() 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, () -> stateMachine.run("SOME STATEMENT", null));
    assertEquals("some error", e.getMessage());
    assertThat(stateMachine.ctx.statementOutcomes.entrySet()).hasSize(0);
    assertNull(stateMachine.ctx.currentTransaction);
}

Example 13 with TransactionStateMachineSPI

use of org.neo4j.bolt.runtime.statemachine.TransactionStateMachineSPI in project neo4j by neo4j.

the class StatementProcessorProvider method getStatementProcessor.

public StatementProcessor getStatementProcessor(String databaseName) throws BoltProtocolBreachFatality, BoltIOException {
    memoryTracker.allocateHeap(TransactionStateMachine.SHALLOW_SIZE);
    TransactionStateMachineSPI transactionSPI = spiProvider.getTransactionStateMachineSPI(databaseName, resourceReleaseManger);
    return new TransactionStateMachine(databaseName, transactionSPI, authResult, clock, routingContext);
}

Example 14 with TransactionStateMachineSPI

use of org.neo4j.bolt.runtime.statemachine.TransactionStateMachineSPI in project neo4j by neo4j.

the class TransactionStateMachineSPIProviderV4Test method shouldReturnTransactionStateMachineSPIIfDatabaseExists.

@Test
void shouldReturnTransactionStateMachineSPIIfDatabaseExists() throws Throwable {
    String databaseName = "database";
    DatabaseManagementService managementService = managementService(databaseName);
    TransactionStateMachineSPIProvider spiProvider = newSpiProvider(managementService);
    TransactionStateMachineSPI spi = spiProvider.getTransactionStateMachineSPI(databaseName, mock(StatementProcessorReleaseManager.class));
    assertThat(spi).isInstanceOf(TransactionStateMachineV4SPI.class);
}

Example 15 with TransactionStateMachineSPI

use of org.neo4j.bolt.runtime.statemachine.TransactionStateMachineSPI in project neo4j by neo4j.

the class DefaultDatabaseTransactionStateMachineSPIProviderTest method shouldReturnDefaultTransactionStateMachineSPIWithEmptyDatabaseName.

@Test
void shouldReturnDefaultTransactionStateMachineSPIWithEmptyDatabaseName() throws Throwable {
    DatabaseManagementService managementService = managementServiceWithDatabase("neo4j");
    TransactionStateMachineSPIProvider spiProvider = newSpiProvider(managementService);
    TransactionStateMachineSPI spi = spiProvider.getTransactionStateMachineSPI(ABSENT_DB_NAME, mock(StatementProcessorReleaseManager.class));
    assertThat(spi).isInstanceOf(TransactionStateMachineSPI.class);
}