PortableExtensionBeanRegistrationTestServlet.java

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright (c) 2010-2017 Oracle and/or its affiliates. All rights reserved.
 *
 * The contents of this file are subject to the terms of either the GNU
 * General Public License Version 2 only ("GPL") or the Common Development
 * and Distribution License("CDDL") (collectively, the "License").  You
 * may not use this file except in compliance with the License.  You can
 * obtain a copy of the License at
 * https://oss.oracle.com/licenses/CDDL+GPL-1.1
 * or LICENSE.txt.  See the License for the specific
 * language governing permissions and limitations under the License.
 *
 * When distributing the software, include this License Header Notice in each
 * file and include the License file at LICENSE.txt.
 *
 * GPL Classpath Exception:
 * Oracle designates this particular file as subject to the "Classpath"
 * exception as provided by Oracle in the GPL Version 2 section of the License
 * file that accompanied this code.
 *
 * Modifications:
 * If applicable, add the following below the License Header, with the fields
 * enclosed by brackets [] replaced by your own identifying information:
 * "Portions Copyright [year] [name of copyright owner]"
 *
 * Contributor(s):
 * If you wish your version of this file to be governed by only the CDDL or
 * only the GPL Version 2, indicate your decision by adding "[Contributor]
 * elects to include this software in this distribution under the [CDDL or GPL
 * Version 2] license."  If you don't indicate a single choice of license, a
 * recipient has the option to distribute your version of this file under
 * either the CDDL, the GPL Version 2 or to extend the choice of license to
 * its licensees as provided above.  However, if you add GPL Version 2 code
 * and therefore, elected the GPL Version 2 license, then the option applies
 * only if the new code is made subject to such option by the copyright
 * holder.
 */

package test.servlet;

import java.io.IOException;
import java.io.PrintWriter;
import java.lang.reflect.Proxy;
import java.util.Iterator;
import java.util.Set;

import javax.enterprise.context.Dependent;
import javax.enterprise.inject.Any;
import javax.enterprise.inject.Default;
import javax.enterprise.inject.spi.Bean;
import javax.enterprise.inject.spi.BeanManager;
import javax.enterprise.util.AnnotationLiteral;
import javax.inject.Inject;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import test.beans.Preferred;
import test.beans.TestBean;
import test.beans.TransactionInterceptor;
import test.extension.ServiceFrameworkExtension;
import test.fwk.FrameworkService;
import test.fwk.SomeFwkServiceImpl;
import test.fwk.SomeFwkService;

@WebServlet(name = "mytest", urlPatterns = { "/myurl" })
public class PortableExtensionBeanRegistrationTestServlet extends HttpServlet {
    @Inject
    @Preferred
    TestBean tb;
    
    @Inject
    BeanManager bm;
    
    @Inject
    @FrameworkService(serviceCriteria="TEST", dynamic=true)
    SomeFwkService sfc_proxy;

//    @Inject
//    @FrameworkService(serviceCriteria="FOO", dynamic=false)
//    SomeFwkService sfc_unproxied;

    String msg = "";

    public void service(HttpServletRequest req, HttpServletResponse res)
            throws IOException, ServletException {

        PrintWriter writer = res.getWriter();
        writer.write("Hello from Servlet 3.0.");

        if (tb == null)
            msg += "Injection of request scoped bean failed";

        tb.m1();
        if (!TransactionInterceptor.aroundInvokeCalled)
            msg += "Business method interceptor aroundInvoke not called";
        tb.m2();
        if (TransactionInterceptor.aroundInvokeInvocationCount != 2)
            msg += "Business method interceptor invocation on method-level "
                    + "interceptor annotation count not expected. "
                    + "expected =2, actual="
                    + TransactionInterceptor.aroundInvokeInvocationCount;
        if (!TransactionInterceptor.errorMessage.trim().equals(""))
            msg += TransactionInterceptor.errorMessage;
        
        //check if our portable extension was called
        if (!ServiceFrameworkExtension.beforeBeanDiscoveryCalled)
            msg += "Portable Extension lifecycle observer method: " +
            		"beforeBeanDiscovery not called";

        if (!ServiceFrameworkExtension.afterBeanDiscoveryCalled)
            msg += "Portable Extension lifecycle observer method: " +
            		"afterBeanDiscovery not called or injection of BeanManager " +
            		"in an observer method failed";
        
        if (!ServiceFrameworkExtension.processAnnotatedTypeCalled)
            msg += "Portable Extension lifecycle observer method: process " +
            		"annotated type not called";

        //BeanManager lookup
        if (bm == null)
            msg += "Injection of BeanManager into servlet failed";
        
        //Get all beans and count the number of beans in the test package
        Set<Bean<?>> allBeans = bm.getBeans(Object.class, new AnnotationLiteral<Any>(){});

        if (sfc_proxy == null)
            msg += "Injection of (proxied) service bean added via extension not available";

//        if (sfc_unproxied == null)
//            msg += "Injection of (unproxied) service bean added via extension not available";

        //check if proxied service reference is indeed proxied
        if (!Proxy.isProxyClass(sfc_proxy.getClass()))
            msg += "Expected Proxied service reference, but got back an " +
            		"unproxied service reference ";
        //invoke a method o
        if(!sfc_proxy.fooMethod())
            msg += "invocation of method on proxied service failed";
        
//        //check if unproxied service reference is not proxied
//        if (Proxy.isProxyClass(sfc_unproxied.getClass()))
//            msg += "Expected unProxied service reference, but got back an " +
//                    "proxied service reference ";
//        //invoke a method o
//        if(!sfc_unproxied.fooMethod())
//            msg += "invocation of method on unproxied service failed";
        
        for (Iterator<Bean<?>> iterator = allBeans.iterator(); iterator.hasNext();) {
            Bean<?> bean = iterator.next();
            if (bean.getBeanClass().getName().contains("SomeFramework")){
                msg += testFrameworkBean(bean);
            }
        }
        
        if (!ServiceFrameworkExtension.afterProcessBeanCalled)
            msg += "Portable Extension lifecycle observer method: after " +
                    "ProcessBean not called for a bean registered by the portable" +
                    "extension";
        
        writer.write(msg + "\n");
    }


    private String testFrameworkBean(Bean<?> bean) {
        String msg = "";
        if (!bean.getQualifiers().contains(new AnnotationLiteral<Any>() {}) 
                || !bean.getQualifiers().contains(new AnnotationLiteral<Default>() {}))  
            msg += "FrameworkBean does not have the default expected qualifiers";
        
        if (!bean.getScope().equals(Dependent.class))
            msg += "FrameworkBean scope is not dependent";
        
        if(!bean.getBeanClass().getName().equals(SomeFwkServiceImpl.class.getName()))
            msg += "Framework Bean: Incorrect bean class";
        return msg;
    }

}