/*
 * Copyright (c) 2002-2012 Alibaba Group Holding Limited.
 * All rights reserved.
 *
 * 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.alibaba.citrus.turbine.pipeline.valve;

import static com.alibaba.citrus.test.TestUtil.*;
import static org.junit.Assert.*;

import java.io.FileNotFoundException;
import java.io.IOException;

import com.alibaba.citrus.service.pipeline.impl.PipelineImpl;
import com.alibaba.citrus.service.requestcontext.lazycommit.LazyCommitRequestContext;
import com.alibaba.citrus.service.requestcontext.util.RequestContextUtil;
import com.alibaba.citrus.webx.util.ErrorHandlerHelper;
import org.junit.Before;
import org.junit.Test;
import org.springframework.beans.factory.BeanCreationException;

public class HandleExceptionValveTests extends AbstractValveTests {
    @Before
    public void initPipeline() throws Exception {
        pipeline = (PipelineImpl) factory.getBean("handleException");
    }

    @Test
    public void create() {
        try {
            new HandleExceptionValve(null);
            fail();
        } catch (IllegalArgumentException e) {
            assertThat(e, exception("no request"));
        }
    }

    @Test
    public void createExceptionHandler() {
        HandleExceptionValve.ExceptionHandler handler;

        try {
            new HandleExceptionValve.ExceptionHandler(null, -1, null, null, null);
            fail();
        } catch (IllegalArgumentException e) {
            assertThat(e, exception("no exception type"));
        }

        handler = new HandleExceptionValve.ExceptionHandler(Exception.class, -10, "  ", null, null);

        assertEquals(Exception.class, handler.getExceptionType());
        assertEquals(-1, handler.getStatusCode());
        assertNull(handler.getPage());
    }

    @Test
    public void create_noDefaultPage() {
        try {
            factory.getBean("handleException_noDefaultPage");
            fail();
        } catch (BeanCreationException e) {
            assertThat(e, exception(IllegalArgumentException.class, "no defaultPage"));
        }
    }

    @Test
    public void matchExceptions() throws Exception {
        // 精确匹配
        assertException(IOException.class, "error_io_exception.vm");
        assertException(Exception.class, "error_exception.vm");
        assertException(IllegalArgumentException.class, "error_illegal_argument_exception.vm");
        assertException(RuntimeException.class, "error_runtime_exception.vm");

        // FileNotFoundException未指定，但是从IOException派生
        assertException(FileNotFoundException.class, "error_io_exception.vm");

        // ArithmeticException未指定，但是从RuntimeException派生
        assertException(ArithmeticException.class, "error_runtime_exception.vm");

        // NumberFormatException未指定，但是从IllegalArgumentException派生
        assertException(NumberFormatException.class, "error_illegal_argument_exception.vm");

        // ClassNotFoundException未指定，但是从Exception派生
        assertException(ClassNotFoundException.class, "error_exception.vm");

        // Error未匹配，取默认值
        assertException(Error.class, "error.vm", false);
    }

    private void assertException(Class<? extends Throwable> type, String target) throws Exception {
        assertException(type, target, true);
    }

    private void assertException(Class<? extends Throwable> type, String target, boolean indirect) throws Exception {
        // 直接异常
        ErrorHandlerHelper.getInstance(newRequest).init("app1", type.newInstance(), null);
        pipeline.newInvocation().invoke();

        assertEquals(target, rundata.getTarget());

        // 间接异常
        if (indirect) {
            Throwable t = new Exception(type.newInstance());

            ErrorHandlerHelper.getInstance(newRequest).init("app1", t, null);
            pipeline.newInvocation().invoke();

            assertEquals(target, rundata.getTarget());
        }
    }

    @Test
    public void helper() {
        Exception e = new Exception();

        ErrorHandlerHelper.getInstance(newRequest).init("app1", e, null);
        pipeline.newInvocation().invoke();

        ErrorHandlerHelper helper = (ErrorHandlerHelper) rundata.getContext().get("error");

        assertEquals(e, helper.getException());
    }

    @Test
    public void helper_specifiedName() {
        pipeline = (PipelineImpl) factory.getBean("handleException_helperName");

        Exception e = new Exception();

        ErrorHandlerHelper.getInstance(newRequest).init("app1", e, null);
        pipeline.newInvocation().invoke();

        ErrorHandlerHelper helper = (ErrorHandlerHelper) rundata.getContext().get("error1");

        assertEquals(e, helper.getException());
    }

    @Test
    public void statusCode() {
        ErrorHandlerHelper.getInstance(newRequest).init("app1", new Exception(), null);
        pipeline.newInvocation().invoke();

        assertEquals(501, RequestContextUtil.findRequestContext(newRequest, LazyCommitRequestContext.class).getStatus());

        ErrorHandlerHelper helper = (ErrorHandlerHelper) rundata.getContext().get("error");

        assertEquals(501, helper.getStatusCode());
        assertEquals("NOT_IMPLEMENTED", helper.getMessage());
    }

    @Test
    public void statusCode_default() {
        ErrorHandlerHelper.getInstance(newRequest).init("app1", new IllegalArgumentException(), null);
        pipeline.newInvocation().invoke();

        ErrorHandlerHelper helper = (ErrorHandlerHelper) rundata.getContext().get("error");

        assertEquals(500, helper.getStatusCode());
        assertEquals("INTERNAL_SERVER_ERROR", helper.getMessage());
    }

    @Test
    public void logging() {
        pipeline = (PipelineImpl) factory.getBean("handleException_logging");

        ErrorHandlerHelper.getInstance(newRequest).init("app1", new Exception(), null);
        pipeline.newInvocation().invoke();

        ErrorHandlerHelper.getInstance(newRequest).init("app1", new IOException(), null);
        pipeline.newInvocation().invoke();

        ErrorHandlerHelper.getInstance(newRequest).init("app1", new IllegalArgumentException(), null);
        pipeline.newInvocation().invoke();
    }
}