TestUtils.java

package org.neo4j.rdf.sail;

import java.io.File;

import org.neo4j.graphdb.GraphDatabaseService;
import org.neo4j.graphdb.Node;
import org.neo4j.graphdb.Relationship;
import org.neo4j.graphdb.Transaction;
import org.neo4j.kernel.EmbeddedGraphDatabase;
import org.neo4j.rdf.fulltext.FulltextIndex;
import org.neo4j.rdf.fulltext.SimpleFulltextIndex;
import org.neo4j.util.EntireGraphDeletor;

public abstract class TestUtils
{
	private static final File BASE_DIR = new File( "target/var" );
	
	private TestUtils()
	{
	}

	public static void deleteEntireNodeSpace( GraphDatabaseService graphDb )
	{
		Transaction tx = graphDb.beginTx();
		try
		{
			for ( Relationship rel : graphDb.getReferenceNode().getRelationships() )
			{
				Node node = rel.getOtherNode( graphDb.getReferenceNode() );
				rel.delete();
				new EntireGraphDeletor().delete( node );
			}
			tx.success();
		}
		finally
		{
			tx.finish();
		}
	}

	public static GraphDatabaseService createGraphDb()
	{
		String dir = new File( BASE_DIR, "db" ).getAbsolutePath();
		removeDir( new File( dir ) );
		final GraphDatabaseService graphDb = new EmbeddedGraphDatabase( dir );
		return graphDb;
	}
	
	public static FulltextIndex createFulltextIndex( GraphDatabaseService graphDb )
	{
		return new SimpleFulltextIndex( graphDb, new File( BASE_DIR, "fulltext" ) );
	}

	private static void removeDir( File dir )
	{
		if ( dir.exists() )
		{
			for ( File file : dir.listFiles() )
			{
				if ( file.isDirectory() )
				{
					removeDir( file );
				}
				else
				{
					file.delete();
				}
			}
		}
	}
}