abstractdatabasetestertest.java

采用 Java 编写的数据库系统单元测试程序。

/*
 *
 * The DbUnit Database Testing Framework
 * Copyright (C)2002-2004, DbUnit.org
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */

package org.dbunit;

import junit.framework.TestCase;

import org.dbunit.database.IDatabaseConnection;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.ITable;
import org.dbunit.dataset.SortedTable;
import org.dbunit.operation.DatabaseOperation;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author Andres Almiray <[e-mail]>
 */
public abstract class AbstractDatabaseTesterTest extends TestCase
{
   protected IDatabaseConnection _connection;
   protected IDatabaseTester _databaseTester;

   protected final Logger logger = LoggerFactory.getLogger( getClass() );

   public AbstractDatabaseTesterTest( String s )
   {
      super( s );
   }

   protected DatabaseEnvironment getEnvironment() throws Exception
   {
      return DatabaseEnvironment.getInstance();
   }

   protected ITable createOrderedTable( String tableName, String orderByColumn ) throws Exception
   {
      return new SortedTable( _connection.createDataSet()
            .getTable( tableName ), new String[] { orderByColumn } );
   }

   // //////////////////////////////////////////////////////////////////////////
   // TestCase class

   protected void setUp() throws Exception
   {
      super.setUp();

      assertNotNull( "DatabaseTester is not set", getDatabaseTester() );
      getDatabaseTester().setSetUpOperation( getSetUpOperation() );
      getDatabaseTester().setDataSet( getDataSet() );
      getDatabaseTester().onSetup();

      _connection = getDatabaseTester().getConnection();
   }

   protected void tearDown() throws Exception
   {
      super.tearDown();

      assertNotNull( "DatabaseTester is not set", getDatabaseTester() );
      getDatabaseTester().setTearDownOperation( getTearDownOperation() );
      getDatabaseTester().setDataSet( getDataSet() );
      getDatabaseTester().onTearDown();

      DatabaseOperation.DELETE_ALL.execute( _connection, _connection.createDataSet() );

      _connection = null;
   }

   // //////////////////////////////////////////////////////////////////////////

   protected IDataSet getDataSet() throws Exception
   {
      return getEnvironment().getInitDataSet();
   }

   protected DatabaseOperation getSetUpOperation()
   {
      return DatabaseOperation.CLEAN_INSERT;
   }

   protected DatabaseOperation getTearDownOperation()
   {
      return DatabaseOperation.NONE;
   }

   protected abstract IDatabaseTester getDatabaseTester() throws Exception;

   /**
    * This method is used so sub-classes can disable the tests according to some
    * characteristics of the environment
    * 
    * @param testName name of the test to be checked
    * @return flag indicating if the test should be executed or not
    */
   protected boolean runTest( String testName )
   {
      return true;
   }

   protected void runTest() throws Throwable
   {
      if( runTest( getName() ) ){
         super.runTest();
      }else{
         if( logger.isDebugEnabled() ){
            logger.debug( "Skipping test " + getClass().getName() + "." + getName() );
         }
      }
   }

   public static boolean environmentHasFeature( TestFeature feature )
   {
      try{
         final DatabaseEnvironment environment = DatabaseEnvironment.getInstance();
         final boolean runIt = environment.support( feature );
         return runIt;
      }
      catch( Exception e ){
         throw new DatabaseUnitRuntimeException( e );
      }
   }
}