/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package org.apache.ode.daohib.bpel;

import java.util.Properties;

import javax.resource.spi.ConnectionManager;
import javax.sql.DataSource;
import javax.transaction.TransactionManager;

import junit.framework.TestCase;

import org.apache.commons.dbcp.BasicDataSource;
import org.apache.derby.jdbc.EmbeddedXADataSource;
import org.apache.ode.bpel.dao.BpelDAOConnection;
import org.apache.ode.il.EmbeddedGeronimoFactory;
import org.hibernate.cfg.Environment;

/**
 * Testing BpelDAOConnectionImpl.listInstance. We're just producing a lot of
 * different filter combinations and test if they execute ok. To really test
 * that the result is the one expected would take a huge test database (with at
 * least a process and an instance for every possible combination).
 */
public abstract class BaseTestDAO extends TestCase {

    protected BpelDAOConnection daoConn;
    protected TransactionManager txm;
    protected ConnectionManager connectionManager;
    private DataSource ds;
    /*
     * Make this true and change the getDataSource
     * method to point to correct driver , database and userid \ password
     */
    private static boolean  externalDB = false;

    protected void initTM() throws Exception {
        EmbeddedGeronimoFactory factory = new EmbeddedGeronimoFactory();
        connectionManager = new org.apache.geronimo.connector.outbound.GenericConnectionManager();
        txm = factory.getTransactionManager();
        ds = getDataSource();
        txm.begin();

        BpelDAOConnectionFactoryImpl factoryImpl = new BpelDAOConnectionFactoryImpl();
        factoryImpl.setTransactionManager(txm);
        factoryImpl.setDataSource(ds);
        Properties props = new Properties();
        props.put(Environment.HBM2DDL_AUTO, "create-drop");
        factoryImpl.init(props);

        daoConn = factoryImpl.getConnection();
    }

    protected void stopTM() throws Exception {
        txm.commit();
    }

    protected DataSource getDataSource() {
        if (externalDB) {
            BasicDataSource ds = new BasicDataSource();
            try {
                ds.setDriverClassName("com.mysql.jdbc.Driver");
                ds.setUsername("sa");
                ds.setPassword("sa");
                ds.setUrl("jdbc:mysql://localhost/bpmsdbJunit");
                this.ds = ds;
            } catch (Exception ex) {
                ex.printStackTrace();
                System.out.println("######### Couldn't get External connection! #############");
            }
        }
        if (ds == null) {
            EmbeddedXADataSource ds = new EmbeddedXADataSource();
            ds.setCreateDatabase("create");
            ds.setDatabaseName("target/testdb");
            ds.setUser("sa");
            ds.setPassword("");
            this.ds = ds;
        }
        return ds;
    }

    protected TransactionManager getTransactionManager() {
        return txm;
    }

}
