use of org.neo4j.kernel.GraphDatabaseQueryService in project neo4j by neo4j.
the class TransactionHandleTest method deadlockExceptionHasCorrectStatus.
@Test
@SuppressWarnings("unchecked")
public void deadlockExceptionHasCorrectStatus() throws Exception {
// given
QueryExecutionEngine executionEngine = mock(QueryExecutionEngine.class);
when(executionEngine.executeQuery(anyString(), anyMap(), any(TransactionalContext.class))).thenThrow(new DeadlockDetectedException("deadlock"));
GraphDatabaseQueryService queryService = mock(GraphDatabaseQueryService.class);
TransactionHandle handle = new TransactionHandle(mockKernel(), executionEngine, queryService, mock(TransactionRegistry.class), uriScheme, true, AUTH_DISABLED, anyLong(), NullLogProvider.getInstance());
ExecutionResultSerializer output = mock(ExecutionResultSerializer.class);
// when
handle.execute(statements(new Statement("query", map(), false, (ResultDataContent[]) null)), output, mock(HttpServletRequest.class));
// then
verify(output).errors(argThat(hasErrors(Status.Transaction.DeadlockDetected)));
}
use of org.neo4j.kernel.GraphDatabaseQueryService in project neo4j by neo4j.
the class TransactionStateMachineSPITest method createTxSpi.
private static TransactionStateMachineSPI createTxSpi(TransactionIdStore txIdStore, Duration txAwaitDuration, AvailabilityGuard availabilityGuard, Clock clock) {
GraphDatabaseQueryService queryService = mock(GraphDatabaseQueryService.class);
when(queryService.getDependencyResolver()).thenReturn(mock(DependencyResolver.class));
return new TransactionStateMachineSPI(mock(GraphDatabaseAPI.class), new ThreadToStatementContextBridge(), mock(QueryExecutionEngine.class), txIdStore, availabilityGuard, queryService, txAwaitDuration, clock);
}
use of org.neo4j.kernel.GraphDatabaseQueryService in project neo4j by neo4j.
the class TransactionHandleTest method shouldCreateTransactionContextOnlyWhenFirstNeeded.
@Test
public void shouldCreateTransactionContextOnlyWhenFirstNeeded() throws Exception {
// given
TransitionalPeriodTransactionMessContainer kernel = mockKernel();
ExecutionResultSerializer output = mock(ExecutionResultSerializer.class);
TransactionRegistry registry = mock(TransactionRegistry.class);
// when
QueryExecutionEngine engine = mock(QueryExecutionEngine.class);
Result executionResult = mock(Result.class);
TransactionalContext transactionalContext = prepareKernelWithQuerySession(kernel);
when(engine.executeQuery("query", map(), transactionalContext)).thenReturn(executionResult);
when(registry.begin(any(TransactionHandle.class))).thenReturn(1337L);
GraphDatabaseQueryService queryService = mock(GraphDatabaseQueryService.class);
TransactionHandle handle = new TransactionHandle(kernel, engine, queryService, registry, uriScheme, true, AUTH_DISABLED, anyLong(), NullLogProvider.getInstance());
// then
verifyZeroInteractions(kernel);
// when
handle.execute(statements(new Statement("query", map(), false, (ResultDataContent[]) null)), output, mock(HttpServletRequest.class));
// then
verify(kernel).newTransaction(any(Type.class), any(SecurityContext.class), anyLong());
InOrder outputOrder = inOrder(output);
outputOrder.verify(output).transactionCommitUri(uriScheme.txCommitUri(1337));
outputOrder.verify(output).statementResult(executionResult, false, (ResultDataContent[]) null);
outputOrder.verify(output).notifications(anyCollectionOf(Notification.class));
outputOrder.verify(output).transactionStatus(anyLong());
outputOrder.verify(output).errors(argThat(hasNoErrors()));
outputOrder.verify(output).finish();
verifyNoMoreInteractions(output);
}
use of org.neo4j.kernel.GraphDatabaseQueryService in project neo4j by neo4j.
the class TransactionHandleTest method shouldRollbackTransactionAndTellRegistryToForgetItsHandle.
@Test
public void shouldRollbackTransactionAndTellRegistryToForgetItsHandle() throws Exception {
// given
TransitionalPeriodTransactionMessContainer kernel = mockKernel();
TransitionalTxManagementKernelTransaction transactionContext = kernel.newTransaction(explicit, AUTH_DISABLED, -1);
TransactionRegistry registry = mock(TransactionRegistry.class);
when(registry.begin(any(TransactionHandle.class))).thenReturn(1337L);
GraphDatabaseQueryService queryService = mock(GraphDatabaseQueryService.class);
TransactionHandle handle = new TransactionHandle(kernel, mock(QueryExecutionEngine.class), queryService, registry, uriScheme, true, AUTH_DISABLED, anyLong(), NullLogProvider.getInstance());
ExecutionResultSerializer output = mock(ExecutionResultSerializer.class);
// when
handle.rollback(output);
// then
InOrder transactionOrder = inOrder(transactionContext, registry);
transactionOrder.verify(transactionContext).rollback();
transactionOrder.verify(registry).forget(1337L);
InOrder outputOrder = inOrder(output);
outputOrder.verify(output).errors(argThat(hasNoErrors()));
outputOrder.verify(output).finish();
verifyNoMoreInteractions(output);
}
use of org.neo4j.kernel.GraphDatabaseQueryService in project neo4j by neo4j.
the class TransactionHandleTest method shouldCommitTransactionAndTellRegistryToForgetItsHandle.
@Test
public void shouldCommitTransactionAndTellRegistryToForgetItsHandle() throws Exception {
// given
TransitionalPeriodTransactionMessContainer kernel = mockKernel();
TransitionalTxManagementKernelTransaction transactionContext = kernel.newTransaction(explicit, AUTH_DISABLED, -1);
TransactionRegistry registry = mock(TransactionRegistry.class);
QueryExecutionEngine engine = mock(QueryExecutionEngine.class);
TransactionalContext transactionalContext = prepareKernelWithQuerySession(kernel);
Result result = mock(Result.class);
when(engine.executeQuery("query", map(), transactionalContext)).thenReturn(result);
when(registry.begin(any(TransactionHandle.class))).thenReturn(1337L);
GraphDatabaseQueryService queryService = mock(GraphDatabaseQueryService.class);
TransactionHandle handle = new TransactionHandle(kernel, engine, queryService, registry, uriScheme, false, AUTH_DISABLED, anyLong(), NullLogProvider.getInstance());
ExecutionResultSerializer output = mock(ExecutionResultSerializer.class);
// when
Statement statement = new Statement("query", map(), false, (ResultDataContent[]) null);
handle.commit(statements(statement), output, mock(HttpServletRequest.class));
// then
InOrder transactionOrder = inOrder(transactionContext, registry);
transactionOrder.verify(transactionContext).commit();
transactionOrder.verify(registry).forget(1337L);
InOrder outputOrder = inOrder(output);
outputOrder.verify(output).statementResult(result, false, (ResultDataContent[]) null);
outputOrder.verify(output).notifications(anyCollectionOf(Notification.class));
outputOrder.verify(output).errors(argThat(hasNoErrors()));
outputOrder.verify(output).finish();
verifyNoMoreInteractions(output);
}
Aggregations