org.neo4j.test.TestGraphDatabaseFactory Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of it-test-support Show documentation
Show all versions of it-test-support Show documentation
ONgDB Community Edition integration tests support.
/*
* Copyright (c) 2018-2020 "Graph Foundation"
* Graph Foundation, Inc. [https://graphfoundation.org]
*
* Copyright (c) 2002-2020 "Neo4j,"
* Neo4j Sweden AB [http://neo4j.com]
*
* This file is part of ONgDB.
*
* ONgDB is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see > INDEX_PROVIDERS_FILTER = extension -> extension instanceof AbstractIndexProviderFactory;
public TestGraphDatabaseFactory()
{
this( NullLogProvider.getInstance() );
}
public TestGraphDatabaseFactory( LogProvider logProvider )
{
super( new TestGraphDatabaseFactoryState() );
setUserLogProvider( logProvider );
}
public GraphDatabaseService newImpermanentDatabase()
{
return newImpermanentDatabaseBuilder().newGraphDatabase();
}
public GraphDatabaseService newImpermanentDatabase( File storeDir )
{
File absoluteDirectory = storeDir.getAbsoluteFile();
GraphDatabaseBuilder databaseBuilder = newImpermanentDatabaseBuilder( absoluteDirectory );
databaseBuilder.setConfig( GraphDatabaseSettings.active_database, absoluteDirectory.getName() );
databaseBuilder.setConfig( GraphDatabaseSettings.databases_root_path, absoluteDirectory.getParentFile().getAbsolutePath() );
return databaseBuilder.newGraphDatabase();
}
public GraphDatabaseService newImpermanentDatabase( Map,String> config )
{
GraphDatabaseBuilder builder = newImpermanentDatabaseBuilder();
setConfig( config, builder );
return builder.newGraphDatabase();
}
public GraphDatabaseService newImpermanentDatabase( File storeDir , Map,String> config )
{
GraphDatabaseBuilder builder = newImpermanentDatabaseBuilder(storeDir);
setConfig( config, builder );
return builder.newGraphDatabase();
}
private void setConfig( Map,String> config, GraphDatabaseBuilder builder )
{
for ( Map.Entry,String> entry : config.entrySet() )
{
Setting key = entry.getKey();
String value = entry.getValue();
builder.setConfig( key, value );
}
}
public GraphDatabaseBuilder newImpermanentDatabaseBuilder()
{
return newImpermanentDatabaseBuilder( ImpermanentGraphDatabase.PATH );
}
@Override
protected void configure( GraphDatabaseBuilder builder )
{
// Reduce the default page cache memory size to 8 mega-bytes for test databases.
builder.setConfig( GraphDatabaseSettings.pagecache_memory, "8m" );
builder.setConfig( new BoltConnector( "bolt" ).type, BOLT.name() );
builder.setConfig( new BoltConnector( "bolt" ).enabled, "false" );
}
private void configure( GraphDatabaseBuilder builder, File storeDir )
{
configure( builder );
builder.setConfig( GraphDatabaseSettings.logs_directory, new File( storeDir, "logs" ).getAbsolutePath() );
}
@Override
protected TestGraphDatabaseFactoryState getCurrentState()
{
return (TestGraphDatabaseFactoryState) super.getCurrentState();
}
@Override
protected TestGraphDatabaseFactoryState getStateCopy()
{
return new TestGraphDatabaseFactoryState( getCurrentState() );
}
public FileSystemAbstraction getFileSystem()
{
return getCurrentState().getFileSystem();
}
public TestGraphDatabaseFactory setFileSystem( FileSystemAbstraction fileSystem )
{
getCurrentState().setFileSystem( fileSystem );
return this;
}
@Override
public TestGraphDatabaseFactory setMonitors( Monitors monitors )
{
getCurrentState().setMonitors( monitors );
return this;
}
@Override
public TestGraphDatabaseFactory setUserLogProvider( LogProvider logProvider )
{
return (TestGraphDatabaseFactory) super.setUserLogProvider( logProvider );
}
public TestGraphDatabaseFactory setInternalLogProvider( LogProvider logProvider )
{
getCurrentState().setInternalLogProvider( logProvider );
return this;
}
public TestGraphDatabaseFactory setClock( SystemNanoClock clock )
{
getCurrentState().setClock( clock );
return this;
}
public TestGraphDatabaseFactory addKernelExtensions( Iterable> newKernelExtensions )
{
getCurrentState().addKernelExtensions( newKernelExtensions );
return this;
}
public TestGraphDatabaseFactory addKernelExtension( KernelExtensionFactory newKernelExtension )
{
return addKernelExtensions( Collections.singletonList( newKernelExtension ) );
}
public TestGraphDatabaseFactory setKernelExtensions( Iterable> newKernelExtensions )
{
getCurrentState().setKernelExtensions( newKernelExtensions );
return this;
}
public TestGraphDatabaseFactory removeKernelExtensions( Predicate> filter )
{
getCurrentState().removeKernelExtensions( filter );
return this;
}
@Override
public TestGraphDatabaseFactory addURLAccessRule( String protocol, URLAccessRule rule )
{
return (TestGraphDatabaseFactory) super.addURLAccessRule( protocol, rule );
}
public GraphDatabaseBuilder newImpermanentDatabaseBuilder( final File storeDir )
{
final TestGraphDatabaseFactoryState state = getStateCopy();
GraphDatabaseBuilder.DatabaseCreator creator =
createImpermanentDatabaseCreator( storeDir, state );
TestGraphDatabaseBuilder builder = createImpermanentGraphDatabaseBuilder( creator );
configure( builder, storeDir );
return builder;
}
private TestGraphDatabaseBuilder createImpermanentGraphDatabaseBuilder(
GraphDatabaseBuilder.DatabaseCreator creator )
{
return new TestGraphDatabaseBuilder( creator );
}
@Override
protected GraphDatabaseService newEmbeddedDatabase( File storeDir, Config config,
GraphDatabaseFacadeFactory.Dependencies dependencies )
{
return new TestGraphDatabaseFacadeFactory( getCurrentState() ).newFacade( storeDir, config,
GraphDatabaseDependencies.newDependencies( dependencies ) );
}
protected GraphDatabaseBuilder.DatabaseCreator createImpermanentDatabaseCreator( final File storeDir,
final TestGraphDatabaseFactoryState state )
{
return new GraphDatabaseBuilder.DatabaseCreator()
{
@Override
public GraphDatabaseService newDatabase( @Nonnull Config config )
{
return new TestGraphDatabaseFacadeFactory( state, true ).newFacade( storeDir, config,
GraphDatabaseDependencies.newDependencies( state.databaseDependencies() ) );
}
};
}
public static class TestGraphDatabaseFacadeFactory extends GraphDatabaseFacadeFactory
{
private final TestGraphDatabaseFactoryState state;
private final boolean impermanent;
protected TestGraphDatabaseFacadeFactory( TestGraphDatabaseFactoryState state, boolean impermanent )
{
this( state, impermanent, DatabaseInfo.COMMUNITY, CommunityEditionModule::new );
}
protected TestGraphDatabaseFacadeFactory( TestGraphDatabaseFactoryState state, boolean impermanent,
DatabaseInfo databaseInfo, Function editionFactory )
{
super( databaseInfo, editionFactory );
this.state = state;
this.impermanent = impermanent;
}
TestGraphDatabaseFacadeFactory( TestGraphDatabaseFactoryState state )
{
this( state, false );
}
@Override
protected PlatformModule createPlatform( File storeDir, Config config, Dependencies dependencies )
{
File absoluteStoreDir = storeDir.getAbsoluteFile();
File databasesRoot = absoluteStoreDir.getParentFile();
if ( !config.isConfigured( GraphDatabaseSettings.shutdown_transaction_end_timeout ) )
{
config.augment( GraphDatabaseSettings.shutdown_transaction_end_timeout, "0s" );
}
config.augment( GraphDatabaseSettings.ephemeral, Settings.FALSE );
config.augment( GraphDatabaseSettings.active_database, absoluteStoreDir.getName() );
config.augment( GraphDatabaseSettings.databases_root_path, databasesRoot.getAbsolutePath() );
if ( impermanent )
{
config.augment( ephemeral, TRUE );
return new ImpermanentTestDatabasePlatformModule( databasesRoot, config, dependencies, this.databaseInfo );
}
else
{
return new TestDatabasePlatformModule( databasesRoot, config, dependencies, this.databaseInfo );
}
}
class TestDatabasePlatformModule extends PlatformModule
{
TestDatabasePlatformModule( File storeDir, Config config, Dependencies dependencies, DatabaseInfo databaseInfo )
{
super( storeDir, config, databaseInfo, dependencies );
}
@Override
protected FileSystemAbstraction createFileSystemAbstraction()
{
FileSystemAbstraction fs = state.getFileSystem();
if ( fs != null )
{
return fs;
}
else
{
return createNewFileSystem();
}
}
protected FileSystemAbstraction createNewFileSystem()
{
return super.createFileSystemAbstraction();
}
@Override
protected LogService createLogService( LogProvider userLogProvider )
{
LogProvider internalLogProvider = state.getInternalLogProvider();
if ( internalLogProvider == null )
{
if ( !impermanent )
{
return super.createLogService( userLogProvider );
}
internalLogProvider = NullLogProvider.getInstance();
}
return new SimpleLogService( userLogProvider, internalLogProvider );
}
@Override
protected SystemNanoClock createClock()
{
SystemNanoClock clock = state.clock();
return clock != null ? clock : super.createClock();
}
}
private class ImpermanentTestDatabasePlatformModule extends TestDatabasePlatformModule
{
ImpermanentTestDatabasePlatformModule( File storeDir, Config config, Dependencies dependencies, DatabaseInfo databaseInfo )
{
super( storeDir, config, dependencies, databaseInfo );
}
@Override
protected FileSystemAbstraction createNewFileSystem()
{
return new EphemeralFileSystemAbstraction();
}
@Override
protected StoreLocker createStoreLocker()
{
return new StoreLocker( fileSystem, storeLayout );
}
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy