com.talis.jersey.apitest.DefaultExceptionMapperAcceptanceTest.java Source code

Introduction

Here is the source code for com.talis.jersey.apitest.DefaultExceptionMapperAcceptanceTest.java

Source

/*
 *    Copyright 2011 Talis Systems Ltd
 * 
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 * 
 *        http://www.apache.org/licenses/LICENSE-2.0
 * 
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */
package com.talis.jersey.apitest;

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

import java.io.IOException;
import java.io.InputStream;
import java.net.ServerSocket;

import org.apache.commons.io.IOUtils;
import org.apache.http.Header;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.util.EntityUtils;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import com.google.inject.Guice;
import com.google.inject.Injector;
import com.google.inject.Module;
import com.talis.jersey.HttpServer;
import com.talis.jersey.guice.GenericServerInfoModule;
import com.talis.jersey.guice.JerseyServletModule;
import com.talis.jersey.guice.NoopAuthenticationModule;

public class DefaultExceptionMapperAcceptanceTest {

    int httpPort;
    HttpServer embeddedServer;
    HttpClient httpClient = new DefaultHttpClient();
    Injector injector;

    @Before
    public void setUp() throws Exception {
        httpPort = findFreePort();
        Module[] modules = { new JerseyServletModule("com.talis.jersey.apitest"), new NoopAuthenticationModule(),
                new GenericServerInfoModule() };
        injector = Guice.createInjector(modules);
        embeddedServer = new HttpServer();
        embeddedServer.start(httpPort, injector);
    }

    @After
    public void tearDown() throws Exception {
    }

    @Test
    public void testUnknownResourceIs404() throws Exception {
        HttpGet httpGet = new HttpGet(getUrl(httpPort, "foo"));
        HttpResponse response = httpClient.execute(httpGet);
        assertEquals(404, response.getStatusLine().getStatusCode());
    }

    @Test
    public void testStubResourceWorksFor200() throws Exception {
        HttpGet httpGet = new HttpGet(getUrl(httpPort, "stub"));
        HttpResponse response = httpClient.execute(httpGet);
        assertEquals(200, response.getStatusLine().getStatusCode());
    }

    @Test
    public void testStubResourceThrows400() throws Exception {
        HttpGet httpGet = new HttpGet(getUrl(httpPort, "stub/badReq"));
        HttpResponse response = httpClient.execute(httpGet);
        assertMsgAndStatus(response, 400, StubResource.SORRY_BAD_REQUEST);
    }

    @Test
    public void testStubResourceThrows404() throws Exception {
        HttpGet httpGet = new HttpGet(getUrl(httpPort, "stub/notFound"));
        HttpResponse response = httpClient.execute(httpGet);
        assertMsgAndStatus(response, 404, StubResource.SORRY_NOT_FOUND);
    }

    @Test
    public void testStubResourceThrows500() throws Exception {
        HttpGet httpGet = new HttpGet(getUrl(httpPort, "stub/internalErr"));
        HttpResponse response = httpClient.execute(httpGet);
        assertMsgAndStatus(response, 500, StubResource.SORRY_INTERNAL_ERROR);
    }

    @Test
    public void testStubResourceThrows503WithDefaultRetry() throws Exception {
        HttpGet httpGet = new HttpGet(getUrl(httpPort, "stub/unavailableErr"));
        HttpResponse response = httpClient.execute(httpGet);
        assertMsgAndStatus(response, 503, StubResource.SORRY_UNAVAILABLE_ERROR);
        Header retryAfter = response.getFirstHeader("Retry-After");
        assertNotNull(retryAfter);
    }

    @Test
    public void testStubResourceThrows503WithSpecificRetry() throws Exception {
        HttpGet httpGet = new HttpGet(getUrl(httpPort, "stub/unavailableErr/3min"));
        HttpResponse response = httpClient.execute(httpGet);
        assertMsgAndStatus(response, 503, StubResource.SORRY_UNAVAILABLE_ERROR);
        Header retryAfter = response.getFirstHeader("Retry-After");
        assertEquals("180", retryAfter.getValue());
    }

    private void assertMsgAndStatus(HttpResponse response, int statusCode, String msg) throws IOException {
        assertEquals(statusCode, response.getStatusLine().getStatusCode());
        HttpEntity entity = response.getEntity();
        try {
            InputStream content = entity.getContent();
            assertEquals(msg, IOUtils.toString(content));
        } finally {
            EntityUtils.consume(entity);
        }
    }

    private int findFreePort() throws IOException {
        ServerSocket serverSocket = new ServerSocket(0);
        int localPort = serverSocket.getLocalPort();
        serverSocket.close();
        return localPort;
    }

    private String getUrl(int port, String path) {
        return String.format("http://localhost:%d/%s", port, path);
    }

}