org.mule.modules.freshbooks.automation.util.RequestAndResponseUtil.java Source code

Introduction

Here is the source code for org.mule.modules.freshbooks.automation.util.RequestAndResponseUtil.java

Source

/**
 * Copyright (c) MuleSoft, Inc. All rights reserved. http://www.mulesoft.com
 *
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.md file.
 */

package org.mule.modules.freshbooks.automation.util;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;

import javax.xml.bind.JAXBElement;
import javax.xml.bind.JAXBException;

import org.custommonkey.xmlunit.XMLTestCase;
import org.custommonkey.xmlunit.XMLUnit;
import org.mule.modules.freshbooks.exception.FreshBooksException;
import org.mule.modules.freshbooks.model.Response;
import org.mule.modules.freshbooks.util.MessageUtils;
import org.springframework.core.io.ClassPathResource;
import org.xml.sax.SAXException;

public abstract class RequestAndResponseUtil extends XMLTestCase {

    protected void assertRequest(String file, Object req) throws IOException, SAXException, JAXBException {

        JAXBElement jaxbElement = MessageUtils.getInstance().createJaxbElement(req);
        String documentToPost = MessageUtils.getInstance().getXmlDocument(jaxbElement);
        XMLUnit.setIgnoreComments(true);
        XMLUnit.setIgnoreWhitespace(true);
        assertXMLEqual(getResourceAsString(new ClassPathResource(file).getInputStream()), documentToPost);
    }

    private static String getResourceAsString(InputStream in) throws IOException {
        ByteArrayOutputStream baos = new ByteArrayOutputStream(1024);
        byte[] buf = new byte[1024];
        int sz = 0;
        try {
            while (true) {
                sz = in.read(buf);

                baos.write(buf, 0, sz);
                if (sz < buf.length)
                    break;
            }
        } finally {
            try {
                in.close();
            } catch (Exception e) {

            }
        }
        return new String(baos.toByteArray());
    }

    protected Response parseResponse(String file) {
        // Response response;
        try {
            // response = (Response) FreshbooksMessageUtils.getInstance().parseResponse(getResourceAsString(new ClassPathResource(file).getInputStream()));
            // return type.getResponseClass().getDeclaredConstructor(Response.class).newInstance(response);
            // } catch (Exception e) {
            // throw new FreshbooksException(e);
            // }
            return (Response) MessageUtils.getInstance()
                    .parseResponse(getResourceAsString(new ClassPathResource(file).getInputStream()));
        } catch (Exception e) {
            throw new FreshBooksException(e);
        }
    }
}