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

Example 16 with TransactionStateMachineSPI

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

the class TransactionStateMachineTest method shouldCloseResultHandlesWhenConsumeFailsInExplicitTransaction.

@Test
void shouldCloseResultHandlesWhenConsumeFailsInExplicitTransaction() throws Throwable {
    BoltTransaction transaction = newTransaction();
    TransactionStateMachineSPI stateMachineSPI = newTransactionStateMachineSPI(transaction);
    TransactionStateMachine stateMachine = newTransactionStateMachine(stateMachineSPI);
    beginTx(stateMachine);
    stateMachine.run("SOME STATEMENT", null);
    StatementOutcome outcome = stateMachine.ctx.statementOutcomes.get(StatementMetadata.ABSENT_QUERY_ID);
    assertNotNull(outcome);
    assertNotNull(outcome.resultHandle);
    assertNotNull(outcome.result);
    RuntimeException e = assertThrows(RuntimeException.class, () -> {
        stateMachine.streamResult(StatementMetadata.ABSENT_QUERY_ID, ERROR);
    });
    assertEquals("some error", e.getMessage());
    assertThat(stateMachine.ctx.statementOutcomes.entrySet()).hasSize(0);
    assertNotNull(stateMachine.ctx.currentTransaction);
}

Example 17 with TransactionStateMachineSPI

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

the class TransactionStateMachineTest method newTransactionStateMachineSPI.

private static TransactionStateMachineSPI newTransactionStateMachineSPI(BoltTransaction transaction, BoltResultHandle resultHandle) {
    TransactionStateMachineSPI stateMachineSPI = mock(TransactionStateMachineSPI.class);
    when(stateMachineSPI.beginTransaction(any(), any(), any(), any(), any(), any())).thenReturn(transaction);
    when(stateMachineSPI.executeQuery(any(), anyString(), any())).thenReturn(resultHandle);
    return stateMachineSPI;
}

Example 18 with TransactionStateMachineSPI

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

the class TransactionStateMachineTest method shouldCloseResultAndTransactionHandlesWhenConsumeFails.

@Test
void shouldCloseResultAndTransactionHandlesWhenConsumeFails() throws Exception {
    BoltTransaction transaction = newTransaction();
    TransactionStateMachineSPI stateMachineSPI = newTransactionStateMachineSPI(transaction);
    TransactionStateMachine stateMachine = newTransactionStateMachine(stateMachineSPI);
    stateMachine.run("SOME STATEMENT", null);
    StatementOutcome outcome = stateMachine.ctx.statementOutcomes.get(StatementMetadata.ABSENT_QUERY_ID);
    assertNotNull(outcome);
    assertNotNull(outcome.resultHandle);
    assertNotNull(outcome.result);
    RuntimeException e = assertThrows(RuntimeException.class, () -> {
        stateMachine.streamResult(StatementMetadata.ABSENT_QUERY_ID, ERROR);
    });
    assertEquals("some error", e.getMessage());
    assertThat(stateMachine.ctx.statementOutcomes.entrySet()).hasSize(0);
    assertNull(stateMachine.ctx.currentTransaction);
}

Example 19 with TransactionStateMachineSPI

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

the class TransactionStateMachineTest method shouldOpenImplicitTransactionForPeriodicCommitQuery.

@Test
void shouldOpenImplicitTransactionForPeriodicCommitQuery() throws Exception {
    BoltTransaction transaction = newTransaction();
    TransactionStateMachineSPI stateMachineSPI = newTransactionStateMachineSPI(transaction);
    when(stateMachineSPI.isPeriodicCommit(PERIODIC_COMMIT_QUERY)).thenReturn(true);
    final BoltTransaction periodicTransaction = mock(BoltTransaction.class);
    when(stateMachineSPI.beginPeriodicCommitTransaction(any(), any(), any(), any(), any(), any())).thenReturn(periodicTransaction);
    TransactionStateMachine stateMachine = newTransactionStateMachine(stateMachineSPI);
    stateMachine.run(PERIODIC_COMMIT_QUERY, EMPTY_MAP);
    // transaction was created only to stream back result of the periodic commit query
    assertEquals(periodicTransaction, stateMachine.ctx.currentTransaction);
    InOrder inOrder = inOrder(stateMachineSPI);
    inOrder.verify(stateMachineSPI).isPeriodicCommit(PERIODIC_COMMIT_QUERY);
    // implicit transaction was started for periodic query execution
    inOrder.verify(stateMachineSPI).beginPeriodicCommitTransaction(any(LoginContext.class), any(), any(), any(), any(), any());
    // periodic commit query was executed after specific transaction started
    inOrder.verify(stateMachineSPI).executeQuery(any(BoltQueryExecutor.class), eq(PERIODIC_COMMIT_QUERY), eq(EMPTY_MAP));
}

Example 20 with TransactionStateMachineSPI

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

the class TransactionStateMachineTest method newTransactionStateMachineSPI.

private static TransactionStateMachineSPI newTransactionStateMachineSPI(BoltTransaction transaction) throws KernelException {
    BoltResultHandle resultHandle = newResultHandle();
    TransactionStateMachineSPI stateMachineSPI = mock(TransactionStateMachineSPI.class);
    when(stateMachineSPI.beginTransaction(any(), any(), any(), any(), any(), any())).thenReturn(transaction);
    when(stateMachineSPI.executeQuery(any(), anyString(), any())).thenReturn(resultHandle);
    return stateMachineSPI;
}