AbstractFormServiceTests.java example

Explorer
citrus-master
/*
 * 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.service.form;

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

import java.io.File;
import java.io.IOException;
import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.alibaba.citrus.service.form.impl.FormServiceImpl;
import com.alibaba.citrus.service.requestcontext.RequestContext;
import com.alibaba.citrus.service.requestcontext.RequestContextChainingService;
import com.alibaba.citrus.service.resource.support.context.ResourceLoadingXmlApplicationContext;
import com.alibaba.citrus.util.internal.Servlet3Util;
import com.meterware.httpunit.GetMethodWebRequest;
import com.meterware.httpunit.HttpUnitOptions;
import com.meterware.httpunit.PostMethodWebRequest;
import com.meterware.httpunit.WebRequest;
import com.meterware.httpunit.WebResponse;
import com.meterware.httpunit.protocol.UploadFileSpec;
import com.meterware.servletunit.InvocationContext;
import com.meterware.servletunit.ServletRunner;
import com.meterware.servletunit.ServletUnitClient;
import org.junit.After;
import org.junit.Before;
import org.springframework.context.ApplicationContext;
import org.springframework.core.io.FileSystemResource;

public abstract class AbstractFormServiceTests {
    protected static ApplicationContext factory;
    protected        FormServiceImpl    formService;

    // web client
    protected ServletUnitClient client;
    protected WebResponse       clientResponse;

    // servlet request/response
    protected InvocationContext   invocationContext;
    protected HttpServletRequest  request;
    protected HttpServletResponse response;
    protected ServletConfig       config;

    // request contexts
    protected RequestContextChainingService requestContexts;
    protected RequestContext                rc;
    protected HttpServletRequest            newRequest;
    protected HttpServletResponse           newResponse;

    static {
        Servlet3Util.setDisableServlet3Features(true); // 禁用servlet3，因为httpunit还不支持
    }

    protected final void getFormService(String name) {
        getFormService(name, factory);
    }

    protected final void getFormService(String name, ApplicationContext factory) {
        formService = (FormServiceImpl) factory.getBean(name);
        assertNotNull(formService);
    }

    protected static ApplicationContext createContext(String name) {
        return createContext(name, null);
    }

    protected static ApplicationContext createContext(String name, boolean withUpload) {
        ApplicationContext parent = withUpload ? createContext("services-base-with-upload.xml", null) : createContext(
                "services-base-without-upload.xml", null);

        return createContext(name, parent);
    }

    protected static ApplicationContext createContext(String name, ApplicationContext parent) {
        return new ResourceLoadingXmlApplicationContext(new FileSystemResource(new File(srcdir, name)), parent);
    }

    @Before
    public final void createWebClient() throws Exception {
        // Servlet container
        ServletRunner servletRunner = new ServletRunner();
        servletRunner.registerServlet("/*", NoopServlet.class.getName());

        // Servlet client
        client = servletRunner.newClient();

        // Charset
        HttpUnitOptions.setDefaultCharacterSet("GBK");
    }

    @After
    public final void commit() throws Exception {
        if (rc != null) {
            requestContexts.commitRequestContext(rc);
        }
    }

    protected final void invokeGet(Object[][] args) throws Exception {
        invokeGet(factory, args);
    }

    protected final void invokeGet(ApplicationContext factory, Object[][] args) throws Exception {
        invokeServlet(factory, new GetMethodWebRequest("http://localhost/"), args);
    }

    protected final void invokePost(Object[][] args) throws Exception {
        invokePost(factory, args);
    }

    protected final void invokePost(ApplicationContext factory, Object[][] args) throws Exception {
        invokeServlet(factory, new PostMethodWebRequest("http://localhost/"), args);
    }

    protected final void invokePostMime(Object[][] args) throws Exception {
        invokePostMime(factory, args);
    }

    protected final void invokePostMime(ApplicationContext factory, Object[][] args) throws Exception {
        invokeServlet(factory, new PostMethodWebRequest("http://localhost/", true), args);
    }

    private void invokeServlet(ApplicationContext factory, WebRequest wr, Object[][] args) throws Exception {
        assertNotNull(wr);

        if (args != null) {
            for (Object[] pair : args) {
                assertTrue(pair != null && pair.length == 2);

                String name = (String) pair[0];
                Object value = pair[1];

                if (value instanceof String[]) {
                    wr.setParameter(name, (String[]) value);
                } else if (value instanceof File[] || value instanceof File) {
                    File[] files = value instanceof File ? new File[] { (File) value } : (File[]) value;
                    UploadFileSpec[] specs = new UploadFileSpec[files.length];

                    for (int i = 0; i < files.length; i++) {
                        if (files[i].getName().contains(".")) {
                            specs[i] = new UploadFileSpec(files[i]);
                        } else {
                            specs[i] = new UploadFileSpec(files[i], ""); // 对于无后缀的文件，不设contentType
                        }
                    }

                    wr.setParameter(name, specs);
                } else {
                    wr.setParameter(name, (String) value);
                }
            }
        }

        invocationContext = client.newInvocation(wr);
        request = invocationContext.getRequest();
        response = invocationContext.getResponse();
        config = invocationContext.getServlet().getServletConfig();

        // request context
        requestContexts = (RequestContextChainingService) factory.getBean("requestContexts");
        rc = requestContexts.getRequestContext(config.getServletContext(), request, response);
        assertNotNull(rc);

        newRequest = rc.getRequest();
        newResponse = rc.getResponse();
    }

    protected final void commitToClient() throws Exception {
        clientResponse = client.getResponse(invocationContext);
    }

    public static class NoopServlet extends HttpServlet {
        private static final long serialVersionUID = 3034658026956449398L;

        @Override
        protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException,
                                                                                              IOException {
        }

        @Override
        protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException,
                                                                                               IOException {
            doGet(request, response);
        }
    }
}