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.restservice;

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.FileInputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import java.util.Date;

import junit.framework.Test;

import junit.framework.TestSuite;

import org.apache.commons.io.IOUtils;

import org.dataone.service.exceptions.BaseException;

import org.dataone.service.exceptions.InvalidSystemMetadata;

import org.dataone.service.exceptions.ServiceFailure;

import org.dataone.service.types.v1.SystemMetadata;

import org.jibx.runtime.BindingDirectory;

import org.jibx.runtime.IBindingFactory;

import org.jibx.runtime.IMarshallingContext;

import edu.ucsb.nceas.MCTestCase;

/**

 * @author berkley

 * class to test ResourceHandler

 */

public class ResourceHandlerTest extends MCTestCase

{

    public ResourceHandlerTest(String name)

    {

        super(name);

    }

    /**

     * Create a suite of tests to be run together

     */

    public static Test suite() 

    {

        TestSuite suite = new TestSuite();

        //suite.addTest(new ResourceHandlerTest("testFindBoundary"));        

        //suite.addTest(new ResourceHandlerTest("testProcessMMP"));

        suite.addTest(new ResourceHandlerTest("testReplicate"));

        return suite;

    }

    public void testReplicate()

    {

    }

    public void testFindBoundary()

    {

        try

        {

            File f = new File("test/testMimeInput.txt");

            FileInputStream fis = new FileInputStream(f);

            String[] s = ResourceHandler.findBoundaryString(fis);

            assertEquals(s[0], "--1288304583346");

            //System.out.println("r: " + s[1].trim());

            assertTrue(s[1].trim().startsWith(s[0]));

        }

        catch(Exception e)

        {

            fail("unexpected error in testFindBoundary: " + e.getMessage());

        }

    }

    public void testProcessMMP()

    {

        try

        {

            runTestProccessMMP("test/testMimeInput.txt", "test/testMimeInput-object.txt");

            runTestProccessMMP("test/testMimeInput2.txt", "test/testMimeInput2-object.txt");

            runTestProccessMMP("test/testMimeInput3.txt", "test/testMimeInput3-object.txt");

        }

        catch(Exception e)

        {

            e.printStackTrace();

            fail("Unexpected error in testProcessMMP: " + e.getMessage());

        }

    }

    /**

     * create a mime multipart message from object and sysmeta and write it to out

     */

    private void createMimeMultipart(OutputStream out, InputStream object,

            SystemMetadata sysmeta) throws IOException, BaseException

    {

        if (sysmeta == null) {

            throw new InvalidSystemMetadata("1000",

                    "System metadata was null.  Can't create multipart form.");

        }

        Date d = new Date();

        String boundary = d.getTime() + "";

        String mime = "MIME-Version:1.0\n";

        mime += "Content-type:multipart/mixed; boundary=\"" + boundary + "\"\n";

        boundary = "--" + boundary + "\n";

        mime += boundary;

        mime += "Content-Disposition: attachment; filename=systemmetadata\n\n";

        out.write(mime.getBytes());

        out.flush();

        //write the sys meta

        try

        {

            ByteArrayInputStream bais = serializeSystemMetadata(sysmeta);

            IOUtils.copy(bais, out);

        }

        catch(Exception e)

        {

            throw new ServiceFailure("1000",

                    "Could not serialize the system metadata to multipart: "

                            + e.getMessage());

        }

        out.write("\n".getBytes()); 

        if (object != null) 

        {    

            out.write(boundary.getBytes());

            out.write("Content-Disposition: attachment; filename=object\n\n".getBytes());

            try {

                mime += IOUtils.copy(object, out);

            } 

            catch (IOException ioe) 

            {

                throw new ServiceFailure("1000",

                        "Error serializing object to multipart: "

                                + ioe.getMessage());

            }

            out.write("\n".getBytes());

        }

        out.write((boundary + "--").getBytes());        

    }

    protected ByteArrayInputStream serializeSystemMetadata(SystemMetadata sysmeta)

        throws Exception 

    {

        ByteArrayOutputStream sysmetaOut = new ByteArrayOutputStream();

        serializeServiceType(SystemMetadata.class, sysmeta, sysmetaOut);

        ByteArrayInputStream sysmetaStream = new ByteArrayInputStream(

                sysmetaOut.toByteArray());

        return sysmetaStream;

    }

    protected void serializeServiceType(Class type, Object object,

            OutputStream out) throws Exception {

        IBindingFactory bfact = BindingDirectory.getFactory(type);

        IMarshallingContext mctx = bfact.createMarshallingContext();

        mctx.marshalDocument(object, "UTF-8", null, out);

    }

    private void runTestProccessMMP(String filename, String answerFilename)

        throws Exception

    {

        /*PropertyService propServ = PropertyService.getInstance("/Users/berkley/tools/tomcat/webapps/knb/WEB-INF");

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

        System.out.println("running process test on file " + filename);

        File f = new File(filename);

        File fObject = new File(answerFilename);

        FileInputStream fobjectis = new FileInputStream(fObject);

        FileInputStream fis = new FileInputStream(f);

        ResourceHandler rh = new ResourceHandler(null, null, null);

        Hashtable<String, File> h = rh.processMMP(fis);

        FileInputStream smFis = new FileInputStream(h.get("sysmeta"));

        FileInputStream objFis = new FileInputStream(h.get("object"));

        String sm = IOUtils.toString(smFis);

        String obj = IOUtils.toString(objFis);

        //System.out.println("sm: " + sm);

        //System.out.println("obj: " + obj);

        String fObjectStr = IOUtils.toString(fobjectis);

        assertTrue(fObjectStr.trim().equals(obj.trim()));

        assertTrue(sm.indexOf("<d1:systemMetadata") != -1);

        assertTrue(sm.indexOf("</d1:systemMetadata>") != -1);*/

    }

}