Search in sources :

Example 1 with BoltKernelGraphDatabaseServiceProvider

use of org.neo4j.bolt.dbapi.impl.BoltKernelGraphDatabaseServiceProvider in project neo4j by neo4j.

the class DatabaseServiceBookmarkTest method createDbSpi.

private BoltGraphDatabaseServiceSPI createDbSpi(TransactionIdStore txIdStore, Duration txAwaitDuration, DatabaseAvailabilityGuard availabilityGuard, SystemNanoClock clock) {
    var queryExecutionEngine = mock(QueryExecutionEngine.class);
    var db = mock(Database.class);
    when(db.getNamedDatabaseId()).thenReturn(DATABASE_ID);
    when(db.getDatabaseAvailabilityGuard()).thenReturn(availabilityGuard);
    var dependencyResolver = mock(Dependencies.class);
    when(dependencyResolver.resolveDependency(QueryExecutionEngine.class)).thenReturn(queryExecutionEngine);
    when(dependencyResolver.resolveDependency(DatabaseAvailabilityGuard.class)).thenReturn(availabilityGuard);
    when(dependencyResolver.resolveDependency(TransactionIdStore.class)).thenReturn(txIdStore);
    when(dependencyResolver.resolveDependency(Database.class)).thenReturn(db);
    when(db.getDependencyResolver()).thenReturn(dependencyResolver);
    var facade = mock(GraphDatabaseAPI.class);
    when(facade.getDependencyResolver()).thenReturn(dependencyResolver);
    var tx = mock(InternalTransaction.class);
    when(facade.beginTransaction(any(), any(), any())).thenReturn(tx);
    var queryService = mock(GraphDatabaseQueryService.class);
    when(queryService.getDependencyResolver()).thenReturn(dependencyResolver);
    when(dependencyResolver.resolveDependency(GraphDatabaseQueryService.class)).thenReturn(queryService);
    var managementService = mock(DatabaseManagementService.class);
    when(managementService.database(DATABASE_ID.name())).thenReturn(facade);
    var transactionIdTracker = new TransactionIdTracker(managementService, new Monitors(), clock);
    return new BoltKernelGraphDatabaseServiceProvider(facade, transactionIdTracker, txAwaitDuration, mock(MemoryTracker.class, RETURNS_MOCKS));
}
Also used : BoltKernelGraphDatabaseServiceProvider(org.neo4j.bolt.dbapi.impl.BoltKernelGraphDatabaseServiceProvider) Monitors(org.neo4j.monitoring.Monitors) TransactionIdTracker(org.neo4j.bolt.txtracking.TransactionIdTracker) MemoryTracker(org.neo4j.memory.MemoryTracker)

Aggregations

BoltKernelGraphDatabaseServiceProvider (org.neo4j.bolt.dbapi.impl.BoltKernelGraphDatabaseServiceProvider)1 TransactionIdTracker (org.neo4j.bolt.txtracking.TransactionIdTracker)1 MemoryTracker (org.neo4j.memory.MemoryTracker)1 Monitors (org.neo4j.monitoring.Monitors)1