com.hp.autonomy.frontend.find.core.web.AbstractErrorControllerTest.java Source code

Introduction

Here is the source code for com.hp.autonomy.frontend.find.core.web.AbstractErrorControllerTest.java

Source

/*
 * Copyright 2015 Hewlett-Packard Development Company, L.P.
 * Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
 */

package com.hp.autonomy.frontend.find.core.web;

import org.apache.http.HttpStatus;
import org.hamcrest.beans.HasPropertyWithValue;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockHttpServletResponse;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.RequestDispatcher;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import java.net.MalformedURLException;

import static org.hamcrest.core.Is.is;
import static org.junit.Assert.assertNotNull;
import static org.mockito.Matchers.any;
import static org.mockito.Matchers.argThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

@RunWith(MockitoJUnitRunner.class)
public abstract class AbstractErrorControllerTest {
    @Mock
    protected ControllerUtils controllerUtils;

    protected CustomErrorController errorController;

    private HttpServletRequest request;
    private HttpServletResponse response;

    @Before
    public void setUp() throws MalformedURLException {
        request = new MockHttpServletRequest();
        response = new MockHttpServletResponse();

        when(controllerUtils.buildErrorModelAndView(any(ErrorModelAndViewInfo.class)))
                .thenReturn(mock(ModelAndView.class));
    }

    @Test
    public void authenticationErrorPage() {
        assertNotNull(errorController.authenticationErrorPage(request, response));
        verify(controllerUtils)
                .buildErrorModelAndView(argThat(new HasPropertyWithValue<ErrorModelAndViewInfo>("mainMessageCode",
                        is(CustomErrorController.MESSAGE_CODE_AUTHENTICATION_ERROR_MAIN))));
    }

    @Test
    public void clientAuthenticationErrorPage() {
        assertNotNull(errorController.clientAuthenticationErrorPage(HttpStatus.SC_GONE, request));
        verify(controllerUtils)
                .buildErrorModelAndView(argThat(new HasPropertyWithValue<ErrorModelAndViewInfo>("mainMessageCode",
                        is(CustomErrorController.MESSAGE_CODE_CLIENT_AUTHENTICATION_ERROR_MAIN))));
    }

    @Test
    public void serverErrorPageWithUUID() {
        request.setAttribute(RequestDispatcher.ERROR_EXCEPTION, mock(Exception.class));

        assertNotNull(errorController.serverErrorPage(request, response));
        verify(controllerUtils)
                .buildErrorModelAndView(argThat(new HasPropertyWithValue<ErrorModelAndViewInfo>("mainMessageCode",
                        is(CustomErrorController.MESSAGE_CODE_INTERNAL_SERVER_ERROR_MAIN))));
        verify(controllerUtils)
                .buildErrorModelAndView(argThat(new HasPropertyWithValue<ErrorModelAndViewInfo>("subMessageCode",
                        is(CustomErrorController.MESSAGE_CODE_INTERNAL_SERVER_ERROR_SUB))));
    }

    @Test
    public void serverErrorPageWithoutUUID() {
        assertNotNull(errorController.serverErrorPage(request, response));
        verify(controllerUtils)
                .buildErrorModelAndView(argThat(new HasPropertyWithValue<ErrorModelAndViewInfo>("mainMessageCode",
                        is(CustomErrorController.MESSAGE_CODE_INTERNAL_SERVER_ERROR_MAIN))));
        verify(controllerUtils)
                .buildErrorModelAndView(argThat(new HasPropertyWithValue<ErrorModelAndViewInfo>("subMessageCode",
                        is(CustomErrorController.MESSAGE_CODE_INTERNAL_SERVER_ERROR_SUB))));
    }

    @Test
    public void notFoundError() {
        assertNotNull(errorController.notFoundError(request, response));
        verify(controllerUtils)
                .buildErrorModelAndView(argThat(new HasPropertyWithValue<ErrorModelAndViewInfo>("mainMessageCode",
                        is(CustomErrorController.MESSAGE_CODE_NOT_FOUND_MAIN))));
    }
}