com.doitnext.http.router.responsehandlers.DefaultErrorHandlerTest.java Source code

Introduction

Here is the source code for com.doitnext.http.router.responsehandlers.DefaultErrorHandlerTest.java

Source

/**
 * Copyright (C) 2013 Steve Owens (DoItNext.com) http://www.doitnext.com
 *
 * 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.doitnext.http.router.responsehandlers;

import java.io.IOException;
import java.lang.reflect.Method;

import javax.servlet.http.HttpServletResponse;

import org.apache.commons.io.IOUtils;
import org.junit.Assert;
import org.junit.Test;
import org.mockito.Mockito;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.mock.web.MockHttpServletResponse;

import com.doitnext.http.router.PathMatch;
import com.doitnext.http.router.Route;
import com.doitnext.http.router.annotations.ExceptionHandler;
import com.doitnext.http.router.annotations.OnException;
import com.doitnext.http.router.annotations.enums.HttpMethod;
import com.doitnext.pathutils.Path;
import com.doitnext.pathutils.PathTemplate;

/**
 * @author Steve Owens (steve@doitnext.com)
 *
 */
public class DefaultErrorHandlerTest {
    Logger logger = LoggerFactory.getLogger(DefaultErrorHandlerTest.class);

    @Test
    public void testGettersAndSetters() {
        DefaultErrorHandler h = new DefaultErrorHandler();
        Assert.assertTrue(h.getResponseTypes().contains("http://com.doitnext.http.router/exception"));
        Assert.assertTrue(h.getResponseFormats().contains("application/json"));
    }

    @Test
    public void testHandleResponseYes() throws IOException {
        DefaultErrorHandler h = new DefaultErrorHandler();
        Exception responseData = new IllegalArgumentException("Hidey Ho!!", new NumberFormatException("3r3"));
        MockHttpServletResponse response = new MockHttpServletResponse();
        boolean handled = h.handleResponse(null, null, response, responseData);
        Assert.assertTrue(handled);

        String content = response.getContentAsString();
        String expectedContent = IOUtils.toString(this.getClass().getResourceAsStream("error1.dat"), "UTF-8");
        Assert.assertEquals(expectedContent, content);
    }

    @Test
    public void testHandleValidStatusCode() throws IOException {
        DefaultErrorHandler h = new DefaultErrorHandler();
        Exception responseData = new IllegalArgumentException("Hidey Ho!!", new NumberFormatException("3r3"));
        MockHttpServletResponse response = new MockHttpServletResponse();
        response.setStatus(403);
        boolean handled = h.handleResponse(null, null, response, responseData);
        Assert.assertTrue(handled);
        Assert.assertEquals(403, response.getStatus());
    }

    @Test
    public void testHandleResponseFail() {
        DefaultErrorHandler h = new DefaultErrorHandler();
        HttpServletResponse response = Mockito.mock(HttpServletResponse.class);
        Exception responseData = new IllegalArgumentException("Hidey Ho!!", new NumberFormatException("3r3"));
        Exception e = new RuntimeException();
        Mockito.doThrow(e).when(response).setContentType(Mockito.anyString());
        logger.info(
                "Testing DefaultErrorHandler ability to handle exceptions while interacting with HttpServletResponse");
        boolean handled = h.handleResponse(null, null, response, responseData);
        Assert.assertFalse(handled);
        logger.info("Awesome!! DefaultErrorHandler handled the exception appropriately.");
    }

    @Test
    public void testHandleResponseWithExceptionHandlerAnnotation() throws Exception {
        DefaultErrorHandler h = new DefaultErrorHandler();
        MockHttpServletResponse response = new MockHttpServletResponse();

        PathTemplate pt = null;
        Path path = new Path("a/b/c", "nice=winds", pt);
        Method method = this.getClass().getMethod("exceptionGeneratingMethod");
        Route route = new Route(HttpMethod.GET, null, null, null, "application/json", pt, this.getClass(), method,
                null, this, null, h, false);
        PathMatch pm = new PathMatch(route, path);

        boolean handled = h.handleResponse(pm, null, response, new IllegalArgumentException());
        Assert.assertTrue(handled);
        Assert.assertEquals(409, response.getStatus());

        response.setStatus(0);
        handled = h.handleResponse(pm, null, response, new IOException());
        Assert.assertTrue(handled);
        Assert.assertEquals(402, response.getStatus());

        response.setStatus(0);
        handled = h.handleResponse(pm, null, response, new NullPointerException());
        Assert.assertTrue(handled);
        Assert.assertEquals(500, response.getStatus());

    }

    @ExceptionHandler({ @OnException(exceptionClass = IOException.class, statusCode = 402),
            @OnException(exceptionClass = IllegalArgumentException.class, statusCode = 409) })
    public void exceptionGeneratingMethod() {

    }

}