Metacat

/**

 *  '$RCSfile$'

 *  Copyright: 2004 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: leinfelder $'

 *     '$Date: 2011-12-13 11:51:20 -0800 (Tue, 13 Dec 2011) $'

 * '$Revision: 6776 $'

 *

 * 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.metacat.replication;

import java.io.InputStream;

import java.net.URL;

import org.apache.commons.io.IOUtils;

import junit.framework.Test;

import junit.framework.TestSuite;

import edu.ucsb.nceas.MCTestCase;

import edu.ucsb.nceas.metacat.MetaCatServlet;

import edu.ucsb.nceas.metacat.client.Metacat;

import edu.ucsb.nceas.metacat.client.MetacatFactory;

import edu.ucsb.nceas.metacat.client.MetacatInaccessibleException;

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

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

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

/**

 * A JUnit test for testing Metacat replication

 */

public class ReplicationTest

    extends MCTestCase {

    private static final long forceReplicationSleep = 1 * 60 * 1000;

	private String targetReplicationServer = null;

	private Metacat targetMetacat = null;

	/**

     * Constructor to build the test

     *

     * @param name the name of the test method

     */

    public ReplicationTest(String name) {

        super(name);

    }

    /**

     * Establish a testing framework by initializing appropriate objects

     */

    public void setUp() {

        try {

        	// get the target ("B server")

            targetReplicationServer = PropertyService.getProperty("test.replication.targetServer");

            m = MetacatFactory.createMetacatConnection(metacatUrl);

            targetMetacat = MetacatFactory.createMetacatConnection("https://" + targetReplicationServer + "/metacat");

        }

        catch (MetacatInaccessibleException mie) {

            System.err.println("Metacat is: " + metacatUrl);

            fail("Metacat connection failed." + mie.getMessage());

        } catch (Exception e) {

        	e.printStackTrace();

            fail(e.getMessage());

        }

    }

    /**

     * Release any objects after tests are complete

     */

    public void tearDown() {

    }

    /**

     * Create a suite of tests to be run together

     */

    public static Test suite() {

        TestSuite suite = new TestSuite();

        suite.addTest(new ReplicationTest("initialize"));

        // Test basic functions

        suite.addTest(new ReplicationTest("testCertificate"));

        suite.addTest(new ReplicationTest("testReplicateData_AtoB"));

//        suite.addTest(new ReplicationTest("testReplicateData_BtoA"));

        return suite;

    }

    /**

     * Run an initial test that always passes to check that the test

     * harness is working.

     */

    public void initialize() {

        assertTrue(1 == 1);

    }

    public void testCertificate() {

    	try {

    		URL u = new URL("https://" + targetReplicationServer  + "/servlet/replication?server=" + MetacatUtil.getLocalReplicationServerName() + "&action=test");

			String test = ReplicationService.getURLContent(u);

			assertTrue(test.contains("Test successfully"));

    	} catch (Exception e) {

    		e.printStackTrace();

			fail(e.getMessage());

		}

    }

    public void testReplicateData_AtoB() {

    	try {

    		// the id

    		String docid = DocumentUtil.generateDocumentId("replicationTest", 1);

    		// the test data

    		String object = "test";

			// insert data locally

    		m.login(username, password);

    		m.upload(docid, "testObject", IOUtils.toInputStream(object, MetaCatServlet.DEFAULT_ENCODING), object.getBytes(MetaCatServlet.DEFAULT_ENCODING).length);

    		// wait for replication (forced)

    		Thread.sleep(forceReplicationSleep);

    		// check the target for the same data

    		targetMetacat.login(username, password);

    		InputStream is = targetMetacat.read(docid);

    		String replicatedObject = IOUtils.toString(is, MetaCatServlet.DEFAULT_ENCODING);

    		assertEquals(object, replicatedObject);

    	} catch (Exception e) {

    		e.printStackTrace();

			fail(e.getMessage());

		}

    }

    public void testReplicateData_BtoA() {

    	try {

    		// the id

    		String docid = DocumentUtil.generateDocumentId("replicationTest", 1);

    		// the test data

    		String object = "test";

			// insert data locally

    		targetMetacat.login(username, password);

    		targetMetacat.upload(docid, "testObject", IOUtils.toInputStream(object, MetaCatServlet.DEFAULT_ENCODING), object.getBytes(MetaCatServlet.DEFAULT_ENCODING).length);

    		// wait for replication (forced)

    		Thread.sleep(forceReplicationSleep);

    		// check the target for the same data

    		m.login(username, password);

    		InputStream is = m.read(docid);

    		String replicatedObject = IOUtils.toString(is, MetaCatServlet.DEFAULT_ENCODING);

    		assertEquals(object, replicatedObject);

    	} catch (Exception e) {

    		e.printStackTrace();

			fail(e.getMessage());

		}

    }

}