com.expedia.seiso.web.controller.ExceptionHandlerAdviceTests.java Source code

Introduction

Here is the source code for com.expedia.seiso.web.controller.ExceptionHandlerAdviceTests.java

Source

/* 
 * Copyright 2013-2015 the original author or authors.
 * 
 * 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.expedia.seiso.web.controller;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import junit.framework.Assert;
import lombok.val;

import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.springframework.validation.BindException;
import org.springframework.validation.BindingResult;
import org.springframework.validation.ObjectError;
import org.springframework.web.context.request.WebRequest;

import com.expedia.seiso.core.exception.ResourceNotFoundException;
import com.expedia.seiso.core.util.C;
import com.fasterxml.jackson.databind.JsonMappingException;

/**
 * @author Willie Wheeler
 */
public class ExceptionHandlerAdviceTests {

    // Class under test
    private ExceptionHandlerAdvice advice;

    // Test data
    private ResourceNotFoundException resourceNotFoundException;
    private JsonMappingException jsonMappingException;
    private RuntimeException runtimeException;

    @Mock
    private BindingResult bindingResult;
    @Mock
    private WebRequest webRequest;

    @Before
    public void setUp() throws Exception {
        this.advice = new ExceptionHandlerAdvice();
        MockitoAnnotations.initMocks(this);
        setUpTestData();
    }

    private void setUpTestData() {
        this.resourceNotFoundException = new ResourceNotFoundException("my-rnfe-message");
        this.jsonMappingException = new JsonMappingException("my-jme-message");
        this.runtimeException = new RuntimeException("my-re-message");
    }

    @Test
    public void handleNoSuchItemException() {
        val result = advice.handleResourceNotFoundException(resourceNotFoundException, webRequest);
        assertNotNull(result);
        assertEquals(C.EC_RESOURCE_NOT_FOUND, result.getCode());
        assertNotNull(result.getMessage());
    }

    @Test
    public void handleJsonMappingException() {
        val result = advice.handleJsonMappingException(jsonMappingException, webRequest);
        assertNotNull(result);
        assertEquals(C.EC_INVALID_REQUEST_JSON_PAYLOAD, result.getCode());
        assertNotNull(result.getMessage());
    }

    @Test
    public void handleRuntimeException() {
        val result = advice.handleRuntimeException(runtimeException, webRequest);
        assertNotNull(result);
        assertEquals(C.EC_INTERNAL_ERROR, result.getCode());
        assertNotNull(result.getMessage());
    }

    @Test
    public void handleBindExceptionException() throws Exception {

        ObjectError expectedError = new ObjectError("name", "message");
        ValidationErrorMap expected = new ValidationErrorMap();
        expected.addError(expectedError.getObjectName(), expectedError.getDefaultMessage());

        BindException bindException = new BindException(this, "foo");
        bindException.addError(expectedError);
        val actual = advice.handleBindException(bindException);

        Assert.assertEquals(expected.errorMap, actual.errorMap);
    }
}