use of org.neo4j.server.http.cypher.format.api.Statement in project neo4j by neo4j.
the class InvocationTest method shouldSuspendTransactionAndReleaseForOtherRequestsAfterExecutingStatements.
@Test
void shouldSuspendTransactionAndReleaseForOtherRequestsAfterExecutingStatements() {
// given
when(internalTransaction.execute("query", emptyMap())).thenReturn(executionResult);
when(registry.begin(any(TransactionHandle.class))).thenReturn(1337L);
TransactionHandle handle = getTransactionHandle(executionEngine, registry);
InputEventStream inputEventStream = mock(InputEventStream.class);
Statement statement = new Statement("query", map());
when(inputEventStream.read()).thenReturn(statement, NULL_STATEMENT);
mockDefaultResult();
Invocation invocation = new Invocation(log, handle, uriScheme.txCommitUri(1337L), mock(MemoryPool.class, RETURNS_MOCKS), inputEventStream, false);
// when
invocation.execute(outputEventStream);
// then
InOrder transactionOrder = inOrder(transactionContext, registry);
transactionOrder.verify(registry).release(1337L, handle);
verify(internalTransaction).execute("query", emptyMap());
InOrder outputOrder = inOrder(outputEventStream);
outputOrder.verify(outputEventStream).writeStatementStart(statement, List.of("c1", "c2", "c3"));
verifyDefaultResultRows(outputOrder);
outputOrder.verify(outputEventStream).writeStatementEnd(queryExecutionType, queryStatistics, executionPlanDescription, notifications);
outputOrder.verify(outputEventStream).writeTransactionInfo(TransactionNotificationState.OPEN, uriScheme.txCommitUri(1337L), 0);
verifyNoMoreInteractions(outputEventStream);
}
Aggregations