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-15 13:59:36 -0800 (Thu, 15 Dec 2011) $'

 * '$Revision: 6788 $'

 *

 * 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.io.StringReader;

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.accesscontrol.AccessControlInterface;

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

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("testReplicateEML_AtoB"));

        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 baseDocid = DocumentUtil.generateDocumentId("replicationTest", 0);

    		String docid = baseDocid + "." + 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);

    		// update the object

    		docid = baseDocid + "." + 2;

    		object = "test2";

    		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 updated data

    		is = targetMetacat.read(docid);

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

    		assertEquals(object, replicatedObject);

    		// update the access control rules

    		m.setAccess(

    				docid, 

    				AccessControlInterface.PUBLIC, 

    				AccessControlInterface.READSTRING, 

    				AccessControlInterface.ALLOW, 

    				AccessControlInterface.ALLOWFIRST);

    		// wait for replication (forced)

    		Thread.sleep(forceReplicationSleep);

    		// check the target for the same data, logout to act as public

    		targetMetacat.logout();

    		is = targetMetacat.read(docid);

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

    		assertEquals(object, replicatedObject);

    		// delete the object

    		m.delete(docid);

    		// wait for replication (forced)

    		Thread.sleep(forceReplicationSleep);

    		// check that it is missing

			replicatedObject = null;

    		try {

        		is = targetMetacat.read(docid);

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

    		} catch (DocumentNotFoundException dnfe) {

    			// expect this

    			assertTrue(true);

    		}

    		assertNull(replicatedObject);

    	} catch (Exception e) {

    		e.printStackTrace();

			fail(e.getMessage());

		}

    }

    public void testReplicateEML_AtoB() {

    	try {

    		// the id

    		String baseDocid = DocumentUtil.generateDocumentId("replicationTest", 0);

    		String docid = baseDocid + "." + 1;

    		// the test data, no public access

    		String emlContent = null;

			emlContent = getTestEmlDoc(

    				"Test replication", //title, 

    				EML2_1_0, //emlVersion, 

    				null, //inlineData1, 

    				null, //inlineData2, 

    				null, //onlineUrl1, 

    				null, //onlineUrl2, 

    				null, //docAccessBlock , 

    				null, //inlineAccessBlock1, 

    				null, //inlineAccessBlock2, 

    				null, //onlineAccessBlock1, 

    				null //onlineAccessBlock2

    				);

    		StringReader xmlReader = new StringReader(emlContent);

			// insert data locally

    		m.login(username, password);

    		m.insert(docid, xmlReader, null);

    		// 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(emlContent, replicatedObject);

    		// update the object

    		docid = baseDocid + "." + 2;

    		//emlContent = getTestEmlDoc("Test replication2", EML2_1_0);

    		emlContent = getTestEmlDoc(

    				"Test replication", //title, 

    				EML2_1_0, //emlVersion, 

    				null, //inlineData1, 

    				null, //inlineData2, 

    				null, //onlineUrl1, 

    				null, //onlineUrl2, 

    				null, //docAccessBlock , 

    				null, //inlineAccessBlock1, 

    				null, //inlineAccessBlock2, 

    				null, //onlineAccessBlock1, 

    				null //onlineAccessBlock2

    				);

    		xmlReader = new StringReader(emlContent);

    		m.update(docid, xmlReader, null);

    		// wait for replication (forced)

    		Thread.sleep(forceReplicationSleep);

    		// check the target for the updated data

    		is = targetMetacat.read(docid);

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

    		assertEquals(emlContent, replicatedObject);

    		// update the access control rules

    		m.setAccess(

    				docid, 

    				AccessControlInterface.PUBLIC, 

    				AccessControlInterface.READSTRING, 

    				AccessControlInterface.ALLOW, 

    				AccessControlInterface.ALLOWFIRST);

    		// wait for replication (forced)

    		Thread.sleep(forceReplicationSleep);

    		// check the target for the same data, logout to act as public

    		targetMetacat.logout();

    		is = targetMetacat.read(docid);

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

    		assertEquals(emlContent, replicatedObject);

    		// delete the object

    		m.delete(docid);

    		// wait for replication (forced)

    		Thread.sleep(forceReplicationSleep);

    		// check that it is missing

			replicatedObject = null;

    		try {

        		is = targetMetacat.read(docid);

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

    		} catch (DocumentNotFoundException dnfe) {

    			// expect this

    			assertTrue(true);

    		}

    		assertNull(replicatedObject);

    	} catch (Exception e) {

    		e.printStackTrace();

			fail(e.getMessage());

		}

    }

}