source: ether_2012/persistence/test/com/ether/HibernateTestHelper.java @ 319

package com.ether;

import java.util.ArrayList;

import org.hibernate.HibernateException;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.springframework.orm.hibernate3.HibernateCallback;
import org.springframework.orm.hibernate3.HibernateTemplate;
import org.springframework.transaction.annotation.Transactional;

public abstract class HibernateTestHelper<BEAN>
        extends TestHelper<BEAN>
{
    /**
     * The constructor.
     *
     * @param beanContextName        the bean context name.
     * @param beanName               the bean name.
     * @param otherContextNames      the other context names.
     * @param sessionFactoryBeanName the session factory bean name.
     */
    protected HibernateTestHelper( final String beanContextName, final String beanName, final String[] otherContextNames, final String sessionFactoryBeanName )
    {
        super( beanContextName, beanName, otherContextNames );
        _sessionFactoryBeanName = sessionFactoryBeanName;
    }

    /**
     * Executes operations before running test.
     * Implemented in ServiceTestHelper class
     */
    protected abstract void beforeHibernateTest()
            throws Exception;

    /**
     * Executes operations after running test.
     * Implemented in ServiceTestHelper class 
     */
    protected abstract void afterHibernateTest()
            throws Exception;

    @Override
    protected final void beforeBeanTest()
            throws Exception
    {
        final ArrayList tab = new ArrayList<String>();
        final SessionFactory sessionFactory;

        if( _sessionFactoryBeanName != null )
            sessionFactory = (SessionFactory) getApplicationContext().getBean( _sessionFactoryBeanName );
        else
            sessionFactory = (SessionFactory) getApplicationContext().getBean( DEFAULT_SESSION_FACTORY_BEAN_NAME );

        _hibernateTemplate = new HibernateTemplate( sessionFactory );
        beforeHibernateTest();
    }

    @Override
    protected final void afterBeanTest()
            throws Exception
    {
        afterHibernateTest();        
        _hibernateTemplate = null;
    }

    /**
     * Delete all datas.
     *
     * @param persistentObjectClasses the persistent object classes.
     * @throws Exception an exception.
     */
    @Transactional
    protected final void deleteAllDatas( final Class<?>[] persistentObjectClasses )
            throws Exception
    {
        _hibernateTemplate.execute( new HibernateCallback()
        {
            public Object doInHibernate( final Session session )
                    throws HibernateException
            {
                for( final Class<?> persistentObjectClass : persistentObjectClasses )
                    session.createQuery( "delete from " + persistentObjectClass.getName() ).executeUpdate();

                return null;
            }
        } );
    }

    private static final String DEFAULT_SESSION_FACTORY_BEAN_NAME = "sessionFactory";
    
    private String _sessionFactoryBeanName;
    private HibernateTemplate _hibernateTemplate;
}