Neo4jWithIndexTestCase.java

package org.neo4j.rdf.sparql;

import org.neo4j.index.IndexService;
import org.neo4j.index.lucene.LuceneIndexService;

public abstract class Neo4jWithIndexTestCase extends Neo4jTestCase
{
    private IndexService indexService;
    
    private void createIndexServiceIfNeeded()
    {
        if ( indexService() == null )
        {
            setIndexService( instantiateIndexService() );
        }
    }

    /**
     * In every setUp(), this class will check if there's an existing
     * IndexSerivce (using {@link #indexService()}). If not, one will be
     * created by invoking this method.
     * @return the newly instantiated index service
     */
    protected IndexService instantiateIndexService()
    {
        return new LuceneIndexService( graphDb() );
    }

    private void setIndexService( IndexService indexService )
    {
        this.indexService = indexService;
    }

    protected IndexService indexService()
    {
        return this.indexService;
    }
    
    @Override
    protected void setUp() throws Exception
    {
        super.setUp();
        createIndexServiceIfNeeded();
    }
    
    @Override protected void doShutdown()
    {
        if ( indexService() != null )
        {
            indexService().shutdown();
            setIndexService( null );
        }
        super.doShutdown();
    }
}