RestProcessITCase.java

/*
 *   R Service Bus
 *   
 *   Copyright (c) Copyright of Open Analytics NV, 2010-2015
 *
 *   ===========================================================================
 *
 *   This program is free software: you can redistribute it and/or modify
 *   it under the terms of the GNU Affero General Public License as published by
 *   the Free Software Foundation, either version 3 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 Affero General Public License for more details.
 *
 *   You should have received a copy of the GNU Affero General Public License
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package eu.openanalytics.rsb;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;

import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.Map;

import org.apache.activemq.util.ByteArrayInputStream;
import org.custommonkey.xmlunit.XMLUnit;
import org.junit.Before;
import org.junit.Test;
import org.w3c.dom.Document;
import org.xml.sax.SAXException;

import com.meterware.httpunit.HttpException;
import com.meterware.httpunit.PostMethodWebRequest;
import com.meterware.httpunit.WebConversation;
import com.meterware.httpunit.WebRequest;
import com.meterware.httpunit.WebResponse;

import eu.openanalytics.httpunit.DeleteMethodWebRequest;

/**
 * @author "OpenAnalytics <rsb.development@openanalytics.eu>"
 */
public class RestProcessITCase extends AbstractITCase {
    private String restProcessUri;

    @Before
    public void prepareTests() throws IOException {
        restProcessUri = RSB_BASE_URI + "/api/rest/process";
    }

    @Test
    public void processBadMethod() throws Exception {
        doTestUnsupportedDeleteMethod(restProcessUri);
    }

    @Test
    public void processBadApplicationName() throws Exception {
        final WebConversation wc = new WebConversation();
        final WebRequest request = new PostMethodWebRequest(restProcessUri, new ByteArrayInputStream("ignored".getBytes()),
                "application/xml");
        request.setHeaderField("X-RSB-Application-Name", ":bad_app$name!");

        try {
            wc.sendRequest(request);
            fail("an exception should have been raised");
        } catch (final HttpException he) {
            assertEquals(400, he.getResponseCode());
        }
    }

    @Test
    public void processBadContentType() throws Exception {
        final WebConversation wc = new WebConversation();
        final WebRequest request = new PostMethodWebRequest(restProcessUri, new ByteArrayInputStream("ignored".getBytes()),
                "application/unsupported");
        request.setHeaderField("X-RSB-Application-Name", "myApp");

        try {
            wc.sendRequest(request);
            fail("an exception should have been raised");
        } catch (final HttpException he) {
            assertEquals(415, he.getResponseCode());
        }
    }

    @Test
    public void submitValidXmlJob() throws Exception {
        final String applicationName = newTestApplicationName();
        final WebResponse response = sendPostRequest(applicationName, "application/xml", getTestData("r-job-sample.xml"), 200);
        final Document resultDoc = XMLUnit.buildTestDocument(response.getText());
        assertEquals("statDataPackage", resultDoc.getDocumentElement().getNodeName());
    }

    @Test
    public void submitInvalidXmlJob() throws Exception {
        final String applicationName = newTestApplicationName();
        final WebResponse response = sendPostRequest(applicationName, "application/xml", new ByteArrayInputStream("<bad/>".getBytes()), 400);
        final Document resultDoc = XMLUnit.buildTestDocument(response.getText());
        assertEquals("errorResult", resultDoc.getDocumentElement().getNodeName());
    }

    @Test
    public void submitValidJsonJob() throws Exception {
        final String applicationName = newTestApplicationName();
        final WebResponse response = sendPostRequest(applicationName, "application/json", getTestData("r-job-sample.json"), 200);
        final List<?> responseObject = Util.fromJson(response.getText(), List.class);
        assertEquals(10, responseObject.size());
    }

    @Test
    public void submitInvalidJsonJob() throws Exception {
        final String applicationName = newTestApplicationName();
        final WebResponse response = sendPostRequest(applicationName, "application/json", new ByteArrayInputStream("not*json".getBytes()),
                400);
        final Map<?, ?> responseObject = Util.fromJson(response.getText(), Map.class);
        assertEquals(applicationName, responseObject.get("applicationName"));
    }

    private WebResponse sendPostRequest(final String applicationName, final String contentType, final InputStream job,
            final int expectedResponseStatusCode) throws IOException, SAXException {
        final WebConversation wc = new WebConversation();
        wc.setExceptionsThrownOnErrorStatus(false);
        final PostMethodWebRequest request = new PostMethodWebRequest(restProcessUri, job, contentType);
        request.setHeaderField("X-RSB-Application-Name", applicationName);

        final WebResponse response = wc.sendRequest(request);
        assertEquals(expectedResponseStatusCode, response.getResponseCode());
        assertEquals(contentType, response.getHeaderField("Content-Type"));
        return response;
    }

    private static void doTestUnsupportedDeleteMethod(final String requestUri) throws IOException, SAXException {
        final WebConversation wc = new WebConversation();
        final WebRequest request = new DeleteMethodWebRequest(requestUri);
        try {
            wc.sendRequest(request);
            fail("an exception should have been raised");
        } catch (final HttpException he) {
            assertEquals(405, he.getResponseCode());
        }
    }
}