Metacat

/**

 *  '$RCSfile$'

 *  Copyright: 2008 Regents of the University of California and the

 *              National Center for Ecological Analysis and Synthesis

 *  Purpose: To test the Access Controls in metacat by JUnit

 *

 *   '$Author: daigle $'

 *     '$Date: 2008-10-10 17:15:19 -0700 (Fri, 10 Oct 2008) $'

 * '$Revision: 4444 $'

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 2 of the License, or

 * (at your option) any later version.

 *

 * This program 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 General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 */

package edu.ucsb.nceas;

import junit.framework.TestCase;

import java.io.IOException;

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.sql.ResultSetMetaData;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.HashMap;

import java.util.Hashtable;

import java.util.Vector;

import org.apache.commons.httpclient.HttpClient;

import edu.ucsb.nceas.metacat.DBConnection;

import edu.ucsb.nceas.metacat.DBConnectionPool;

import edu.ucsb.nceas.metacat.service.PropertyService;

import edu.ucsb.nceas.metacat.service.ServiceException;

import edu.ucsb.nceas.metacat.util.RequestUtil;

import edu.ucsb.nceas.utilities.PropertyNotFoundException;

import edu.ucsb.nceas.utilities.SortedProperties;

/**

 * A base JUnit class for Metacat tests

 */

public class MCTestCase

    extends TestCase {

    private static boolean printDebug = false;

	protected static String EML2_0_0 = "EML2_0_0";

	protected static String EML2_0_1 = "EML2_0_1";

	protected static String EML2_1_0 = "EML2_1_0";

	protected boolean SUCCESS = true;

	protected boolean FAILURE = false;

	protected static HttpClient httpClient = null;

	static {

		try {

			SortedProperties testProperties = 

				new SortedProperties("build/tests/test.properties");

			testProperties.load();

			String metacatContextDir = testProperties.getProperty("metacat.contextDir");

			PropertyService.getInstance(metacatContextDir + "/WEB-INF");

		    String printDebugString = PropertyService.getProperty("test.printdebug");

		    printDebug = Boolean.parseBoolean(printDebugString);

		} catch (IOException ioe) {

			System.err.println("Could not read property file in static block: " 

					+ ioe.getMessage());

		} catch (PropertyNotFoundException pnfe) {

			System.err.println("Could not get property in static block: " 

					+ pnfe.getMessage());

		} catch (ServiceException se) {

			System.err.println("Could not get PropertyService instance in static block: " 

					+ se.getMessage());

		}

	}

    /**

     * Constructor to build the test

     */

    public MCTestCase() {

        super();

    }

    /**

     * Constructor to build the test

     *

     * @param name the name of the test method

     */

    public MCTestCase(String name) {

        super(name);

    }

    protected static void debug(String debugMessage) {

    	if (printDebug) {

    		System.err.println(debugMessage);

    	}

    }

	protected static Vector<Hashtable<String, Object>> dbSelect(String sqlStatement,

			String methodName) throws SQLException {

		Vector<Hashtable<String, Object>> resultVector = new Vector<Hashtable<String, Object>>();

		DBConnectionPool connPool = DBConnectionPool.getInstance();

		DBConnection dbconn = DBConnectionPool.getDBConnection(methodName);

		int serialNumber = dbconn.getCheckOutSerialNumber();

		PreparedStatement pstmt = null;

		debug("Selecting from db: " + sqlStatement);

		pstmt = dbconn.prepareStatement(sqlStatement);

		pstmt.execute();

		ResultSet resultSet = pstmt.getResultSet();

		ResultSetMetaData rsMetaData = resultSet.getMetaData();

		int numColumns = rsMetaData.getColumnCount();

		debug("Number of data columns: " + numColumns);

		while (resultSet.next()) {

			Hashtable<String, Object> hashTable = new Hashtable<String, Object>();

			for (int i = 1; i <= numColumns; i++) {

				if (resultSet.getObject(i) != null) {

					hashTable.put(rsMetaData.getColumnName(i), resultSet.getObject(i));

				}

			}

			debug("adding data row to result vector");

			resultVector.add(hashTable);

		}

		resultSet.close();

		pstmt.close();

		DBConnectionPool.returnDBConnection(dbconn, serialNumber);

		return resultVector;

	}

	protected static void dbQuery(String sqlStatement, String methodName)

			throws SQLException {

		DBConnectionPool connPool = DBConnectionPool.getInstance();

		DBConnection dbconn = DBConnectionPool.getDBConnection(methodName);

		int serialNumber = dbconn.getCheckOutSerialNumber();

		Statement statement = dbconn.createStatement();

		debug("Executing against db: " + sqlStatement);

		statement.executeQuery(sqlStatement);

		statement.close();

		DBConnectionPool.returnDBConnection(dbconn, serialNumber);

	}

	protected static void dbUpdate(String sqlStatement, String methodName)

			throws SQLException {

		DBConnectionPool connPool = DBConnectionPool.getInstance();

		DBConnection dbconn = DBConnectionPool.getDBConnection(methodName);

		int serialNumber = dbconn.getCheckOutSerialNumber();

		Statement statement = dbconn.createStatement();

		debug("Executing against db: " + sqlStatement);

		statement.executeUpdate(sqlStatement);

		statement.close();

		DBConnectionPool.returnDBConnection(dbconn, serialNumber);

	}

	protected static void httpPost(String url, HashMap<String,String> paramMap)  throws IOException {

		debug("Posting to: " + url);

		if (httpClient == null) {

			httpClient = new HttpClient();

		}

		String postResponse = RequestUtil.post(httpClient, url, paramMap);

		debug("Post response: " + postResponse);

		if (postResponse.contains("<error>")) {

			fail("Error posting to metacat: " + postResponse);

		}

	}

	protected static void resetHttpClient() {

		httpClient = null;

	}

}