net.sf.javagimmicks.sql.testing.H2DbTestRule Maven / Gradle / Ivy

Go to download
package net.sf.javagimmicks.sql.testing;

import java.io.File;
import java.net.MalformedURLException;

import org.apache.commons.dbcp.BasicDataSource;
import org.junit.rules.TestRule;

/**
 * A JUnit {@link TestRule} that creates a temporary file-based H2 database
 * during test execution and wraps a Commons-DBCP {@link BasicDataSource} around
 * it.
 * 
 * The developer can influence the temporary folder, where the database should
 * be created and/or can provide initial configuration logic for the used
 * {@link BasicDataSource} via a {@link DataSourceConfigurator}.
 * 

 * Usage example:
 * 
 * 
 * public class H2DbTestRuleTest
 * {
 * 
 *    @Rule
 *    public H2DbTestRule _db = new H2DbTestRule();
 * 
 *    @Test
 *    public void test() throws SQLException
 *    {
 *       final Connection connection = _db.getConnection();
 *       assertNotNull(connection);
 * 
 *       // Do some DB operations
 * 
 *       conneciton.close();
 *    }
 * }
 * 
 */
public class H2DbTestRule extends AbstractDbTestRule
{
   /**
    * Creates a new instance using the given {@link DataSourceConfigurator} and
    * {@link File folder} for the internal database.
    * 
    * @param configurator
    *           a {@link DataSourceConfigurator} for performing custom
    *           configuration logic on the internal {@link BasicDataSource}
    * @param dbFolder
    *           the temporary folder where the database should be set up
    */
   public H2DbTestRule(final DataSourceConfigurator configurator, final File dbFolder)
   {
      super(configurator, dbFolder);
   }

   /**
    * Creates a new instance using the given {@link DataSourceConfigurator} for
    * the internal database.
    * 
    * @param configurator
    *           a {@link DataSourceConfigurator} for performing custom
    *           configuration logic on the internal {@link BasicDataSource}
    */
   public H2DbTestRule(final DataSourceConfigurator configurator)
   {
      super(configurator);
   }

   /**
    * Creates a new instance using the given {@link File folder} for the
    * internal database.
    * 
    * @param dbFolder
    *           the temporary folder where the database should be set up
    */
   public H2DbTestRule(final File dbFolder)
   {
      super(dbFolder);
   }

   /**
    * Creates a new instance with no special configuration.
    */
   public H2DbTestRule()
   {
      super();
   }

   @Override
   protected void configureDatasource(final BasicDataSource dataSource, final File tempFolder)
         throws MalformedURLException
   {
      dataSource.setDriverClassName(org.h2.Driver.class.getName());
      dataSource.setUrl(buildJdbcUrl(tempFolder));
      dataSource.setUsername("sa");
      dataSource.setPassword("sa");
   }

   private static String buildJdbcUrl(final File tempFolder) throws MalformedURLException
   {
      final String url = "jdbc:h2:" + tempFolder.toURI().toURL() + NAME_TEST_DB;
      return url;
   }
}