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);

        init();

    }

    /**

	 * 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() 

	{

	}

	public void init()

	{

	    try

	    {

        }

        catch(Exception e)

        {

            e.printStackTrace();

            fail("Could not initialize CrudServiceTest: " + e.getMessage());

        }

	}

	/**

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

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

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

		return suite;

	}

	/**

	 * public SystemMetadata getSystemMetadata(AuthToken token, Identifier guid)

     *       throws InvalidToken, ServiceFailure, NotAuthorized, NotFound, 

     *       InvalidRequest, NotImplemented

	 */

	public void testGetSystemMetadata()

	{

	}

	/**

	 * 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()

	{

	    try

	    {

	        String testDoc = getTestDoc();

	        CrudService cs = CrudService.getInstance();

	        AuthToken token = getToken();

	        //run create

	        Identifier id = createDoc(token);

            String gotDoc = getDoc(token, id);

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

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

        }

        catch(Exception e)

        {

            e.printStackTrace();

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

        }

	}

	/**

	 * getInstance()

	 */

	public void 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() 

	{

		assertTrue(1 == 1);

	}

	/**

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

	 */

	private String getDoc(AuthToken token, Identifier id)

	  throws Exception

	{

	    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);

        }

        return sb.toString();

	}

	/**

	 * return a test document

	 */

	private String getTestDoc()

	{

	    return "<?xml version=\"1.0\"?><test></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) throws Exception

	{

	    Identifier id;

        CrudService cs = CrudService.getInstance();

        String testDoc = getTestDoc();

        id = new Identifier();

        String docid = generateDocumentId();

        id.setValue(docid);

        //create the system metadata then run the create method

        StringBufferInputStream sbis = new StringBufferInputStream(testDoc);

        SystemMetadata sm = new SystemMetadata();

        //set the id

        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);

        //create the doc

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

        return id;

	}

	/**

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

    }

}