private void setUpMocks() { queryService = mock( GraphDatabaseQueryService.class ); DependencyResolver resolver = mock( DependencyResolver.class ); txBridge = mock( ThreadToStatementContextBridge.class ); initialStatement = mock( KernelStatement.class ); statistics = new ConfiguredExecutionStatistics(); QueryRegistryOperations queryRegistryOperations = mock( QueryRegistryOperations.class ); InternalTransaction internalTransaction = mock( InternalTransaction.class ); when( internalTransaction.terminationReason() ).thenReturn( Optional.empty() ); when( initialStatement.queryRegistration() ).thenReturn( queryRegistryOperations ); when( queryService.getDependencyResolver() ).thenReturn( resolver ); when( resolver.resolveDependency( ThreadToStatementContextBridge.class ) ).thenReturn( txBridge ); when( queryService.beginTransaction( any(), any() ) ).thenReturn( internalTransaction ); KernelTransaction mockTransaction = mockTransaction( initialStatement ); when( txBridge.get() ).thenReturn( initialStatement ); when( txBridge.getKernelTransactionBoundToThisThread(true ) ).thenReturn( mockTransaction ); }
@Test public void checkKernelStatementOnCheck() { InternalTransaction initialTransaction = mock( InternalTransaction.class, new ReturnsDeepStubs() ); Kernel kernel = mock( Kernel.class ); ThreadToStatementContextBridge txBridge = mock( ThreadToStatementContextBridge.class ); KernelTransaction kernelTransaction = mockTransaction( initialStatement ); when( txBridge.getKernelTransactionBoundToThisThread( true ) ).thenReturn( kernelTransaction ); Neo4jTransactionalContext transactionalContext = new Neo4jTransactionalContext( null, txBridge, null, initialTransaction, initialStatement, null, kernel ); transactionalContext.check(); verify( kernelTransaction ).assertOpen(); }
KernelTransaction initialKTX = mockTransaction( initialStatement ); InternalTransaction initialTransaction = mock( InternalTransaction.class, new ReturnsDeepStubs() ); KernelTransaction.Type transactionType = KernelTransaction.Type.implicit; KernelTransaction secondKTX = mockTransaction( secondStatement ); InternalTransaction secondTransaction = mock( InternalTransaction.class ); when( secondTransaction.terminationReason() ).thenReturn( Optional.empty() );
KernelTransaction initialKTX = mockTransaction( initialStatement ); QueryRegistryOperations initialQueryRegistry = mock( QueryRegistryOperations.class ); ExecutingQuery executingQuery = mock( ExecutingQuery.class ); KernelTransaction secondKTX = mockTransaction( secondStatement ); InternalTransaction secondTransaction = mock( InternalTransaction.class ); when( secondTransaction.terminationReason() ).thenReturn( Optional.empty() );