SpringTestUtils.java

#set( $symbol_pound = '#' )
#set( $symbol_dollar = '$' )
#set( $symbol_escape = '\' )
/* vim: set ts=2 et sw=2 cindent fo=qroca: */

package ${package}.web.testsupport;

import java.sql.Connection;
import java.sql.SQLException;

import javax.servlet.ServletContext;
import javax.sql.DataSource;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.ApplicationContext;
import org.springframework.context.support.FileSystemXmlApplicationContext;
import org.springframework.core.io.FileSystemResourceLoader;
import org.springframework.mock.web.MockServletContext;
import org.springframework.web.context.support.XmlWebApplicationContext;

/** Utility class to give support to test cases.
 */
public final class SpringTestUtils {

  /** A logger.
   */
  private static Logger log = LoggerFactory.getLogger(SpringTestUtils.class);

  /** a data source, as a singleton.
   */
  private static DataSource dataSource = null;

  /** Bean factory, as a singleton.
   */
  private static ApplicationContext beanFactory;

  /** User module Bean factory, as a singleton.
   */
  private static ApplicationContext userModuleBeanFactory;

  /** A private constructor so no instances are created.
   */
  private SpringTestUtils() {
  }

  /** Gets the configured data source.
   *
   * @return a DataSource, never null.
   */
  public static synchronized DataSource getDataSource() {
    if (dataSource == null) {
      dataSource = (DataSource) getBeanFactory().getBean("dataSource");
    }
    return dataSource;
  }

  /** Gets the connection to the database.
   *
   * @return a Connection, never null.
   *
   * @exception SQLException if a database access error occurs.
   */
  public static Connection getConnection() throws SQLException {
    Connection connection = getDataSource().getConnection();
    return connection;
  }

  /** This method returns a BeanFactory.
   *
   * @return a BeanFactory
   */
  public static synchronized ApplicationContext getBeanFactory() {
    if (beanFactory == null) {
      log.info("Creating a beanFactory");
      ServletContext sc = new MockServletContext("./src/main/webapp",
          new FileSystemResourceLoader());
      XmlWebApplicationContext appContext = new XmlWebApplicationContext();
      appContext.setServletContext(sc);
      appContext.setConfigLocations(new String[] {
        "/WEB-INF/applicationContext.xml",
        "/WEB-INF/applicationContextRuntime.xml"});
      appContext.refresh();
      beanFactory = appContext;
    }
    return beanFactory;
  }

  /** This method returns a BeanFactory.
   *
   * @return a BeanFactory
   */
  public static synchronized ApplicationContext getUserModuleBeanFactory() {
    if (userModuleBeanFactory == null) {
      log.info("Creating a beanFactory");
      userModuleBeanFactory = new FileSystemXmlApplicationContext(
          new String[]
          {"classpath:/com/globant/katari/user/view/spring-servlet.xml"},
          getBeanFactory());
    }
    return userModuleBeanFactory;
  }
}