Metacat

/**

 *  '$RCSfile$'

 *  Copyright: 2010 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: berkley $'

 *     '$Date: 2010-05-03 14:26:08 -0700 (Fri, 14 Aug 2009) $'

 * '$Revision: 5027 $'

 *

 * 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.dataone;

import java.util.*;

import java.io.*;

import java.security.MessageDigest;

import edu.ucsb.nceas.MCTestCase;

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

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

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

import edu.ucsb.nceas.metacat.dataone.CrudService;

import junit.framework.Test;

import junit.framework.TestSuite;

import org.dataone.service.types.*;

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

import edu.ucsb.nceas.metacat.client.rest.MetacatRestClient;

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

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

/**

 * A JUnit test for testing the dataone CrudService class

 */

public class CrudServiceTest extends MCTestCase 

{   

    /**

    * consstructor for the test

    */

    public CrudServiceTest(String name)

    {

        super(name);

    }

    /**

	 * Establish a testing framework by initializing appropriate objects

	 */

	public void setUp() throws Exception 

	{

		super.setUp();

	}

	/**

	 * 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 CrudServiceTest("initialize"));

		suite.addTest(new CrudServiceTest("testSingletonAccessor"));

		suite.addTest(new CrudServiceTest("testCreateAndGet"));

		suite.addTest(new CrudServiceTest("testGetSystemMetadata"));*/

		suite.addTest(new CrudServiceTest("testUpdate"));

		//suite.addTest(new CrudServiceTest(""));

		//suite.addTest(new CrudServiceTest(""));

		return suite;

	}

	/**

	 * public Identifier update(AuthToken token, Identifier guid, 

     *       InputStream object, Identifier obsoletedGuid, SystemMetadata sysmeta) 

     *         throws InvalidToken, ServiceFailure, NotAuthorized, IdentifierNotUnique, 

     *           UnsupportedType, InsufficientResources, NotFound, InvalidSystemMetadata, 

     *           NotImplemented

	 */

	public void testUpdate()

	{

	    printTestHeader("testUpdate");

	    try

	    {

	        CrudService cs = CrudService.getInstance();

	        AuthToken token = getToken();

            //create a doc

            System.out.println("CST: creating document");

            Identifier id = createDoc(token, getTestDoc());

            System.out.println("CST: document created.  Id: " + id.getValue());

            //get the doc and sysmetadata

            System.out.println("CST: getting document with id " + id.getValue());

            String gotDoc = getDoc(token, id);

            System.out.println("CST: getting sysmeta for doc with id " + id.getValue());

            SystemMetadata sm = getSystemMetadata(token, id);

            //update the doc

            System.out.println("$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$ doing the update now $$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$");

            gotDoc = gotDoc.replaceAll("XXX", "YYY");

            System.out.println("CST: Changed test doc to " + gotDoc);

            Identifier newid = new Identifier();

            newid.setValue(generateDocumentId());

            System.out.println("CST: Generated newid " + newid);

            StringBufferInputStream sbis = new StringBufferInputStream(gotDoc);

            SystemMetadata newsm = createSystemMetadata(newid, gotDoc);

            System.out.println("CST: Created system metadata for new doc...updating.");

            Identifier updatedid = cs.update(token, newid, sbis, id, newsm);

            System.out.println("CST: Called update. Returned id " + id);

            //get doc - check that it matches update

            String newdoc = getDoc(token, newid);

            assertTrue(gotDoc.equals(newdoc));

            //get sysmeta - check that ids and other fields are updated

            SystemMetadata newnewsm = getSystemMetadata(token, id);

            List obsoletedByList = newnewsm.getObsoletedByList();

            assertTrue(obsoletedByList.size() == 1);

            String obsBy = ((Identifier)obsoletedByList.get(0)).getValue();

            assertTrue(obsBy.equals(id.getValue()));

        }

        catch(Exception e)

        {

            e.printStackTrace();

            fail("Error in testUpdate: " + e.getMessage());

        }

	}

	/**

	 * public SystemMetadata getSystemMetadata(AuthToken token, Identifier guid)

     *       throws InvalidToken, ServiceFailure, NotAuthorized, NotFound, 

     *       InvalidRequest, NotImplemented

	 */

	public void testGetSystemMetadata()

	{

	    printTestHeader("testGetSystemMetadata");

	    try

	    {

            CrudService cs = CrudService.getInstance();

            AuthToken token = getToken();

            //run create

            Identifier id = createDoc(token, getTestDoc());

            //get the systemMetadata and make sure it is the correct object for this testDoc

            SystemMetadata sm = getSystemMetadata(token, id);

            assertTrue(sm.getIdentifier().getValue().equals(id.getValue()));

            assertTrue(sm.getChecksum().getValue().equals(checksum(getTestDoc())));

            try

            {

                Identifier fakeid = new Identifier();

                fakeid.setValue("somethingfake.234234");

                getSystemMetadata(token, fakeid);

                fail("getSystemMetadata should have thrown an exception.");

            }

            catch(Exception e)

            {

                assertTrue(true);

            }

        }

        catch(Exception e)

        {

            e.printStackTrace();

            fail("Error testing system metadata: " + e.getMessage());

        }

	}

	/**

	 * create(AuthToken token, Identifier guid, InputStream object, SystemMetadata sysmeta) 

     *   throws InvalidToken, ServiceFailure, NotAuthorized, IdentifierNotUnique, UnsupportedType, 

     *       InsufficientResources, InvalidSystemMetadata, NotImplemented

     *

     * public InputStream get(AuthToken token, Identifier guid)

     *       throws InvalidToken, ServiceFailure, NotAuthorized, NotFound, 

     *       NotImplemented

	 */

	public void testCreateAndGet()

	{

	    printTestHeader("testCreateAndGet");

	    try

	    {

	        CrudService cs = CrudService.getInstance();

	        AuthToken token = getToken();

	        //run create

	        Identifier id = createDoc(token, getTestDoc());

	        //run get

            String gotDoc = getDoc(token, id);

            System.out.println("got doc: " + gotDoc);

            //compare the docs

            assertTrue(gotDoc.trim().equals(getTestDoc().trim()));

            try

            {

                Identifier fakeid = new Identifier();

                fakeid.setValue("somethingfake.234234");

                getDoc(token, fakeid);

                fail("testCreateAndGet should have thrown an exception.");

            }

            catch(Exception e)

            {

                assertTrue(true);

            }

        }

        catch(Exception e)

        {

            e.printStackTrace();

            fail("Error in testCreate: " + e.getMessage());

        }

	}

	/**

	 * getInstance()

	 */

	public void testSingletonAccessor()

	{

	    printTestHeader("testSingletonAccessor");

	    CrudService cs = CrudService.getInstance();

	    assertNotNull(cs);

	}

	/**

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

	 * working.

	 */

	public void initialize() 

	{

	    printTestHeader("initialize");

		assertTrue(1 == 1);

	}

	/**

	 * return the systemMetadata object for id

	 */

	private SystemMetadata getSystemMetadata(AuthToken token, Identifier id)

	  throws Exception

	{

	    System.out.println("getting system metadata for id: " + id.getValue());

	    CrudService cs = CrudService.getInstance();

	    return cs.getSystemMetadata(token, id);

	}

	/**

	 * get a doc from metacat using CrudService.get()

	 */

	private String getDoc(AuthToken token, Identifier id)

	  throws Exception

	{

	    System.out.println("getting doc for id " + id.getValue());

	    CrudService cs = CrudService.getInstance();

	    //try to get the same doc then compare them

        InputStream gotDocStream = cs.get(token, id);

        StringBuffer sb = new StringBuffer();

        byte[] b = new byte[1024];

        int numread = gotDocStream.read(b, 0, 1024);

        while(numread != -1)

        {

            sb.append(new String(b, 0, numread));

            numread = gotDocStream.read(b, 0, 1024);

        }

        System.out.println("returning doc for id " + id.getValue());

        return sb.toString();

	}

	/**

	 * return a test document

	 */

	private String getTestDoc()

	{

	    return "<?xml version=\"1.0\"?><test><somecontent>This is some content XXX</somecontent></test>\n";

	}

	/**

	 * authenticate and return a token

	 */

	private AuthToken getToken()

	  throws Exception

	{

	    CrudService cs = CrudService.getInstance();

        //login and get a sessionid

        System.out.println("creating MetacatRestClient with url " + cs.getContextUrl());

        MetacatRestClient restClient = new MetacatRestClient(cs.getContextUrl());

        String username = PropertyService.getProperty("test.mcUser");

        String password = PropertyService.getProperty("test.mcPassword");

        System.out.println("logging in with username: " + username + " and password " + password + " to context " + cs.getContextUrl());

        String response = restClient.login(username, password);

        //System.out.println("response to login: " + response);

        String sessionid = restClient.getSessionId();

        SessionService sessionService = SessionService.getInstance();

        sessionService.registerSession(new SessionData(sessionid, username, new String[0], password, "CrudServiceLogin"));

        System.out.println("sessionid: " + sessionid);

        AuthToken token = new AuthToken(sessionid);

        return token;

	}

	/**

	 * create a doc using CrudService.create() and return its id

	 */

	private Identifier createDoc(AuthToken token, String testDoc) throws Exception

	{

	    Identifier id;

        CrudService cs = CrudService.getInstance();

        id = new Identifier();

        String docid = generateDocumentId();

        id.setValue(docid);

        System.out.println("creating doc with id " + id.getValue());            

        //create the system metadata then run the create method

        StringBufferInputStream sbis = new StringBufferInputStream(testDoc);

        SystemMetadata sm = createSystemMetadata(id, testDoc);

        //create the doc

        cs.create(token, id, sbis, sm);

        System.out.println("document with id " + id.getValue() + " created.");

        return id;

	}

	/**

	 * create a generic SystemMetadata object for testing

	 */

	private SystemMetadata createSystemMetadata(Identifier id, String testDoc)

	  throws Exception

	{

	    SystemMetadata sm = new SystemMetadata();

        //set the id

        System.out.println("creating system metadata object for document with id " + id.getValue());

        sm.setIdentifier(id);

        System.out.println("sm id is " + id.getValue());

        sm.setObjectFormat(ObjectFormat.convert("eml://ecoinformatics.org/eml-2.1.0"));

        System.out.println("sm objectformat: " + sm.getObjectFormat());

        //create the checksum

        String checksumS = checksum(testDoc);

        ChecksumAlgorithm ca = ChecksumAlgorithm.convert("MD5");

        Checksum checksum = new Checksum();

        checksum.setValue(checksumS);

        checksum.setAlgorithm(ca);

        sm.setChecksum(checksum);

        System.out.println("sm checksum is " + checksumS);

        //set the size

        sm.setSize(testDoc.getBytes().length);

        System.out.println("sm size: " + testDoc.getBytes().length);

        //submitter

        Principal p = new Principal();

        p.setValue("joe");

        sm.setSubmitter(p);

        sm.setRightsHolder(p);

        sm.setDateUploaded(new Date());

        sm.setDateSysMetadataModified(new Date());

        NodeReference nr = new NodeReference();

        nr.setValue("metacat");

        sm.setOriginMemberNode(nr);

        sm.setAuthoritativeMemberNode(nr);

        return sm;

	}

	/**

	 * print a header to start each test

	 */

	private void printTestHeader(String testName)

	{

	    System.out.println();

	    System.out.println("********************************** " + testName + " **********************************");

	}

	/**

	 * produce an md5 checksum for item

	 */

	private String checksum(String item)

	  throws Exception

	{

        StringBufferInputStream fis =  new StringBufferInputStream(item);

        byte[] buffer = new byte[1024];

        MessageDigest complete = MessageDigest.getInstance("MD5");

        int numRead;

        do 

        {

          numRead = fis.read(buffer);

          if (numRead > 0) 

          {

            complete.update(buffer, 0, numRead);

          }

        } while (numRead != -1);

        return getHex(complete.digest());

	}

	/**

	 * convert a byte array to a hex string

	 */

	private static String getHex( byte [] raw ) 

	{

	    final String HEXES = "0123456789ABCDEF";

        if ( raw == null ) {

          return null;

        }

        final StringBuilder hex = new StringBuilder( 2 * raw.length );

        for ( final byte b : raw ) {

          hex.append(HEXES.charAt((b & 0xF0) >> 4))

             .append(HEXES.charAt((b & 0x0F)));

        }

        return hex.toString();

    }

}