jndidatabasetester.java

来自「采用 Java 编写的数据库系统单元测试程序。」· Java 代码 · 共 106 行

/*
 * 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 org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Properties;

import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import javax.sql.DataSource;

import org.dbunit.database.DatabaseConnection;
import org.dbunit.database.IDatabaseConnection;

/**
 * DatabaseTester that pulls a DataSource from a JNDI location.
 *
 * @author Andres Almiray <[e-mail]>
 */
public class JndiDatabaseTester extends AbstractDatabaseTester
{

    /**
     * Logger for this class
     */
    private static final Logger logger = LoggerFactory.getLogger(JndiDatabaseTester.class);

   private DataSource dataSource;
   private Properties environment;
   private boolean initialized = false;
   private String lookupName;

   /**
    * Creates a JndiDatabaseTester with specific JNDI properties.
    *
    * @param environment A Properties object with JNDI properties
    * @param lookupName the name of the resource in the JNDI context
    */
   public JndiDatabaseTester( Properties environment, String lookupName )
   {
      super();
      this.environment = environment;
      this.lookupName = lookupName;
   }

   /**
    * Creates a JndiDatabaseTester with default JNDI properties.
    *
    * @param lookupName the name of the resource in the JNDI context
    */
   public JndiDatabaseTester( String lookupName )
   {
      this( null, lookupName );
   }

   public IDatabaseConnection getConnection() throws Exception
   {
        logger.debug("getConnection() - start");

      if( !initialized ){
         initialize();
      }

      if( getSchema() != null ){
         return new DatabaseConnection( dataSource.getConnection(), getSchema() );
      }else{
         return new DatabaseConnection( dataSource.getConnection() );
      }
   }

   /**
    * Verifies the configured properties and locates the Datasource through
    * JNDI.<br>
    * This method is called by {@link getConnection} if the tester has not been
    * initialized yet.
    */
   private void initialize() throws NamingException
   {
        logger.debug("initialize() - start");

      Context context = new InitialContext( environment );
      assertNotNullNorEmpty( "lookupName", lookupName );
      Object obj = context.lookup( lookupName );
      assertTrue( "JNDI object with [" + lookupName + "] not found", obj!=null );
      assertTrue( "Object [" + obj + "] at JNDI location [" + lookupName
            + "] is not of type [" + DataSource.class.getName() + "]", obj instanceof DataSource );
      dataSource = (DataSource) obj;
      assertTrue( "DataSource is not set", dataSource!=null );
      initialized = true;
   }
}