appserver/tests/appserv-tests/devtests/web/servlet-3.0/asyncContextHasOriginalRequestAndResponse/servlet/test/MyFilter.java

/*
 * Copyright (c) 1997, 2018 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package test;

import java.io.*;
import jakarta.servlet.*;
import jakarta.servlet.http.*;
import jakarta.servlet.annotation.*;

@WebFilter(urlPatterns = {"/test"}, asyncSupported = true)
public class MyFilter implements Filter {

    public void init(FilterConfig filterConfig) throws ServletException {
        // Do nothing
    }

    public void doFilter(ServletRequest request, ServletResponse response,
                         FilterChain chain)
                throws IOException, ServletException {

        String mode = request.getParameter("mode");
        if (!"noarg".equals(mode) && !"original".equals(mode) &&
                !"wrap".equals(mode)) {
            throw new ServletException("Invalid mode");
        }

        if ("wrap".equals(mode)) {
            chain.doFilter(new MyWrapper((HttpServletRequest) request),
                response);
        } else {
            chain.doFilter(request, response);
        }

        AsyncContext ac = request.getAsyncContext();
        if ("noarg".equals(mode) && !ac.hasOriginalRequestAndResponse()) {
            throw new ServletException(
                "AsycContext#hasOriginalRequestAndResponse returned false, " +
                "should have returned true");
        } else if ("original".equals(mode) &&
                !ac.hasOriginalRequestAndResponse()) {
            throw new ServletException(
                "AsycContext#hasOriginalRequestAndResponse returned false, " +
                "should have returned true");
        } else if ("wrap".equals(mode) &&
                ac.hasOriginalRequestAndResponse()) {
            throw new ServletException(
                "AsycContext#hasOriginalRequestAndResponse returned true, " +
                "should have returned false");
        }

        ac.complete();
    }

    public void destroy() {
        // Do nothing
    }

    private static class MyWrapper extends HttpServletRequestWrapper {

        public MyWrapper(HttpServletRequest req) {
            super(req);
        }
    }
}