src/test/java/org/apache/continuum/store/ApplicationContextAwareStoreTestCase.java

package org.apache.continuum.store;

import org.apache.continuum.dao.api.GenericDao;
import org.apache.continuum.model.CommonUpdatableEntity;
import org.apache.openjpa.persistence.test.SingleEMTestCase;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;

/**
 * Test Case support class that allows extensions to load test data from specified SQL files.
 * <p/>
 * This also implements Spring's {@link ApplicationContextAware} interface that allows the {@link ApplicationContext} to
 * be made available to this test case's extensions.
 *
 * @author <a href='mailto:rahul.thakur.xdev@gmail.com'>Rahul Thakur</a>
 * @version $Id$
 * @since 1.2
 */
public abstract class ApplicationContextAwareStoreTestCase
    extends SingleEMTestCase
    implements ApplicationContextAware
{
    /**
     * Continuum Store persistent unit defined in <code>persistence.xml</code> used by tests.
     */
    private static final String PERSISTENT_UNIT_CONTINUUM_STORE = "continuum-store";

    /**
     * Spring application context.
     */
    private ApplicationContext applicationContext;

    /**
     * {@inheritDoc}
     * <p/>
     * Spring IoC container injects the {@link ApplicationContext} through here.
     *
     * @see org.springframework.context.ApplicationContextAware#setApplicationContext(org.springframework.context.ApplicationContext)
     */
    public void setApplicationContext( ApplicationContext applicationContext )
        throws BeansException
    {
        this.applicationContext = applicationContext;
    }

    /**
     * Imports sql from the specified file.
     *
     * @param sqlResource Resource containing sql
     */
    public void setSqlSource( File sqlResource )
    {
        try
        {
            // TODO: Use Logger!
            // System.out.println( "Loading sql: " + sqlResource.getAbsolutePath() );
            List<String> statements = new ArrayList<String>( 20 );
            BufferedReader br = new BufferedReader( new InputStreamReader( new FileInputStream( sqlResource ) ) );
            String line;
            StringBuffer currentStatement = new StringBuffer();
            while ( ( line = br.readLine() ) != null )
            {
                if ( line.trim().length() == 0 )
                {
                    continue;
                }
                if ( line.trim().startsWith( "#" ) )
                {
                    continue;
                }

                currentStatement.append( line );
                if ( line.endsWith( ";" ) )
                {
                    statements.add( currentStatement.toString() );
                    currentStatement = new StringBuffer();
                }
            }
            // append a line if missing a ';'
            if ( currentStatement.length() > 0 )
            {
                statements.add( currentStatement.toString() );
            }
            runSQLStatements( statements );
        }
        catch ( Throwable e )
        {
            // TODO: User logger!
            System.err.println( "Problem executing SQL!" );
            e.printStackTrace();
        }
    }

    /**
     * Run a bunch of SQL statements.
     *
     * @param statements Statements to run.
     * @throws SQLException
     */
    public void runSQLStatements( final List<String> statements )
        throws SQLException
    {
        for ( String qry : statements )
        {
            Connection con = (Connection) this.em.getConnection();
            try
            {
                Statement stmt = con.createStatement();
                System.out.println( qry );
                stmt.execute( qry );
                con.commit();
            }
            catch ( SQLException e )
            {
                try
                {
                    con.rollback();
                }
                catch ( SQLException e1 )
                {
                    // TODO: Use logger!
                    System.err.println( "Unable to rollback transaction." );
                    throw e1;
                }
                throw e;
            }
        }
    }

    @SuppressWarnings("unchecked")
    protected <T extends CommonUpdatableEntity> GenericDao<T> getDao( String daoBeanReference )
    {
        return (GenericDao<T>) getObject( daoBeanReference );
    }

    protected Object getObject( String beanReference )
    {
        return applicationContext.getBean( beanReference );
    }

    /**
     * Returns the name of the persistent-unit setup in <code>persistence.xml</code>.
     */
    @Override
    protected String getPersistenceUnitName()
    {
        return PERSISTENT_UNIT_CONTINUUM_STORE;
    }

}