All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.jboss.cdi.tck.tests.extensions.alternative.metadata.GroceryWrapper Maven / Gradle / Ivy

There is a newer version: 2.0.5.SP1
Show newest version
/*
 * JBoss, Home of Professional Open Source
 * Copyright 2010, Red Hat, Inc., and individual contributors
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * 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 org.jboss.cdi.tck.tests.extensions.alternative.metadata;

import javax.enterprise.context.RequestScoped;
import javax.enterprise.event.Observes;
import javax.enterprise.event.Reception;
import javax.enterprise.event.TransactionPhase;
import javax.enterprise.inject.Any;
import javax.enterprise.inject.Disposes;
import javax.enterprise.inject.Produces;
import javax.enterprise.inject.literal.InjectLiteral;
import javax.enterprise.inject.spi.AnnotatedConstructor;
import javax.enterprise.inject.spi.AnnotatedField;
import javax.enterprise.inject.spi.AnnotatedMethod;
import javax.enterprise.inject.spi.AnnotatedParameter;
import javax.enterprise.inject.spi.AnnotatedType;
import javax.enterprise.util.AnnotationLiteral;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import org.jboss.cdi.tck.util.annotated.AnnotatedConstructorWrapper;
import org.jboss.cdi.tck.util.annotated.AnnotatedFieldWrapper;
import org.jboss.cdi.tck.util.annotated.AnnotatedMethodWrapper;
import org.jboss.cdi.tck.util.annotated.AnnotatedParameterWrapper;
import org.jboss.cdi.tck.util.annotated.AnnotatedTypeWrapper;

public class GroceryWrapper extends AnnotatedTypeWrapper {
    private final Set typeClosure = new HashSet();
    private static boolean getBaseTypeOfFruitFieldUsed = false;
    private static boolean getBaseTypeOfInitializerTropicalFruitParameterUsed = false;
    private static boolean getBaseTypeOfBillDisposerParameterUsed = false;
    private static boolean getBaseTypeOfObserverInjectionPointUsed = false;
    private static boolean getBaseTypeOfObserverParameterUsed = false;
    private static boolean getTypeClosureUsed = false;
    private static boolean getTypeClosureOfProducerMethodUsed = false;

    public GroceryWrapper(AnnotatedType delegate) {
        super(delegate, false, RequestScoped.Literal.INSTANCE, new CheapLiteral(), new AnnotationLiteral() {
        }, new AnnotationLiteral() {
        });
        typeClosure.add(Grocery.class);
        typeClosure.add(Object.class);
    }

    @Override
    public Set getTypeClosure() {
        getTypeClosureUsed = true;
        return typeClosure;
    }

    @Override
    public Set> getConstructors() {
        Set> constructors = new HashSet>();
        for (AnnotatedConstructor constructor : super.getConstructors()) {
            if (constructor.getParameters().size() == 1) {
                AnnotatedConstructorWrapper constructorWrapper = new AnnotatedConstructorWrapper(constructor, this, true,
                        new InjectLiteral());
                constructors.add(constructorWrapper);
            } else {
                AnnotatedConstructorWrapper constructorWrapper = new AnnotatedConstructorWrapper(constructor, this, true);
                constructors.add(constructorWrapper);
            }
        }
        return constructors;
    }

    @Override
    public Set> getFields() {
        Set> fields = new HashSet>();
        for (AnnotatedField field : super.getFields()) {
            if (field.getJavaMember().getName().equals("vegetables")) {
                fields.add(wrapField(field, this, new InjectLiteral()));
            } else if (field.getJavaMember().getName().equals("fruit")) {
                fields.add(wrapFruitField(field, this, new CheapLiteral()));
            } else if (field.getBaseType().equals(Bread.class)) {
                fields.add(wrapField(field, this, new AnnotationLiteral() {
                }));
            } else {
                fields.add(new AnnotatedFieldWrapper(field, this, true));
            }
        }
        return fields;
    }

    @SuppressWarnings({ "unchecked", "rawtypes", "serial" })
    @Override
    public Set> getMethods() {
        Set> methods = new HashSet>();
        for (AnnotatedMethod method : super.getMethods()) {
            if (method.getJavaMember().getName().equals("getMilk")) {

                AnnotatedMethodWrapper wrappedMethod = new AnnotatedMethodWrapper(method, this, false,
                        new AnnotationLiteral() {
                        });
                methods.add(wrappedMethod);
            } else if (method.getJavaMember().getName().equals("getYogurt")) {
                // wrap the method and its parameters
                AnnotatedMethodWrapper wrappedMethod = new AnnotatedMethodWrapper(method, this, false, new ExpensiveLiteral(),
                        new AnnotationLiteral() {
                        });
                methods.add(wrapMethodParameters(method.getParameters(), wrappedMethod, false, new CheapLiteral()));
            } else if (method.getJavaMember().getName().equals("nonInjectAnnotatedInitializer")) {
                AnnotatedMethodWrapper wrappedMethod = new AnnotatedMethodWrapper(method, this, true,
                        new Annotation[] { new InjectLiteral() });
                methods.add(wrappedMethod);
            } else if (method.getJavaMember().getName().equals("initializer")) {

                AnnotatedMethodWrapper methodWrapper = new AnnotatedMethodWrapper(method, this, true);

                methodWrapper.replaceParameters(new AnnotatedParameterWrapper(method.getParameters().get(0), methodWrapper, false,
                        new CheapLiteral()) {
                    @Override
                    public Type getBaseType() {
                        getBaseTypeOfInitializerTropicalFruitParameterUsed = true;
                        return TropicalFruit.class;
                    }
                });
                methods.add(methodWrapper);

            } else if (method.getJavaMember().getName().equals("observer1")) {
                Annotation[] parameterAnnotations = new Annotation[] { new Observes() {

                    public TransactionPhase during() {
                        return TransactionPhase.IN_PROGRESS;
                    }

                    public Reception notifyObserver() {
                        return Reception.ALWAYS;
                    }

                    public Class annotationType() {
                        return Observes.class;
                    }

                }, new CheapLiteral() };
                AnnotatedMethodWrapper methodWrapper = new AnnotatedMethodWrapper(method, this, true);
                methods.add(wrapMethodParameters(method.getParameters(), methodWrapper, false, parameterAnnotations));
            } else if (method.getJavaMember().getName().equals("observer2")) {
                AnnotatedMethodWrapper methodWrapper = new AnnotatedMethodWrapper(method, this, true);
                methods.add(wrapMethodParameters(method.getParameters(), methodWrapper, true, new ExpensiveLiteral()));
            } else if (method.getJavaMember().getName().equals("observerMilk")) {
                AnnotatedMethodWrapper methodWrapper = new AnnotatedMethodWrapper(method, this, true);
                methodWrapper.replaceParameters(new AnnotatedParameterWrapper(methodWrapper.getParameter(0), methodWrapper, true),
                        new AnnotatedParameterWrapper(methodWrapper.getParameter(1), methodWrapper, true) {
                            @Override
                            public Type getBaseType() {
                                getBaseTypeOfObserverInjectionPointUsed = true;
                                return TropicalFruit.class;
                            }
                        });
                methods.add(methodWrapper);
            } else if (method.getJavaMember().getName().equals("destroyBill")) {
                AnnotatedMethodWrapper methodWrapper = new AnnotatedMethodWrapper(method, this, true);
                // replace the second parameter and keep the first one
                methodWrapper.replaceParameters(new AnnotatedParameterWrapper(methodWrapper.getParameter(0), methodWrapper, true),
                        new AnnotatedParameterWrapper(methodWrapper.getParameter(1), methodWrapper, true, Any.Literal.INSTANCE) {
                            @Override
                            public Type getBaseType() {
                                getBaseTypeOfBillDisposerParameterUsed = true;
                                return TropicalFruit.class;
                            }
                        });
                methods.add(methodWrapper);

            } else if (method.getJavaMember().getName().equals("observesVegetable")) {
                AnnotatedMethodWrapper methodWrapper = new AnnotatedMethodWrapper(method, this, true);
                methodWrapper.replaceParameters(new AnnotatedParameterWrapper(methodWrapper.getParameter(0), methodWrapper, true) {
                    @Override
                    public Type getBaseType() {
                        getBaseTypeOfObserverParameterUsed = true;
                        return Carrot.class;
                    }
                });
                methods.add(methodWrapper);

            } else if (method.getJavaMember().getName().equals("destroyVegetable")) {
                AnnotatedMethodWrapper methodWrapper = new AnnotatedMethodWrapper(method, this, true);
                methodWrapper.replaceParameters(new AnnotatedParameterWrapper(methodWrapper.getParameter(0), methodWrapper, true, new CheapLiteral()) {
                    @Override
                    public Type getBaseType() {
                        return Carrot.class;
                    }
                });
                methods.add(methodWrapper);

            } else if (method.getJavaMember().getName().equals("destroyYogurt")) {
                AnnotatedMethodWrapper methodWrapper = new AnnotatedMethodWrapper(method, this, true);
                methodWrapper.replaceParameters(
                        new AnnotatedParameterWrapper(methodWrapper.getParameter(0), methodWrapper, false,
                                new Annotation[] { new ExpensiveLiteral(), new Disposes() {
                                    public Class annotationType() {
                                        return Disposes.class;
                                    }
                                } }));
                methods.add(methodWrapper);

            } else if (method.getJavaMember().getName().equals("createVegetable")) {

                AnnotatedMethodWrapper methodWrapper = new AnnotatedMethodWrapper(method, this, true) {
                    @Override
                    public Set getTypeClosure() {
                        Set types = new HashSet();
                        types.add(Carrot.class);
                        getTypeClosureOfProducerMethodUsed = true;
                        return types;
                    }
                };
                methods.add(methodWrapper);

            } else {
                methods.add(method);
            }
        }
        return methods;
    }

    private  AnnotatedFieldWrapper wrapField(AnnotatedField delegate, AnnotatedTypeWrapper declaringType, Annotation... annotations) {
        return new AnnotatedFieldWrapper(delegate, declaringType, false, annotations);
    }

    private  AnnotatedFieldWrapper wrapFruitField(AnnotatedField delegate, AnnotatedTypeWrapper declaringType, Annotation... annotations) {
        return new AnnotatedFieldWrapper(delegate, declaringType, true, annotations) {
            @Override
            public Type getBaseType() {
                getBaseTypeOfFruitFieldUsed = true;
                return TropicalFruit.class;
            }

            @SuppressWarnings("serial")
            @Override
            public Set getTypeClosure() {
                return new HashSet() {
                    {
                        add(Object.class);
                        add(Fruit.class);
                        add(TropicalFruit.class);
                    }
                };
            }

        };
    }

    /**
     * This method allows you to add a set of Annotations to every method parameter. Note that the method will remove all
     * method-level annotations.
     */
    private  AnnotatedMethodWrapper wrapMethodParameters(List> params, AnnotatedMethodWrapper declaringMethod,
            final boolean keepOriginalAnnotations, final Annotation... annotations) {

        List> wrappedParams = new ArrayList>();
        for (AnnotatedParameter param : params) {
            wrappedParams.add(new AnnotatedParameterWrapper(param, declaringMethod, keepOriginalAnnotations, annotations[param.getPosition()]));
        }

        declaringMethod.replaceParameters(wrappedParams.toArray(new AnnotatedParameter[wrappedParams.size()]));
        return declaringMethod;
    }

    public static boolean isGetBaseTypeOfFruitFieldUsed() {
        return getBaseTypeOfFruitFieldUsed;
    }

    public static boolean isGetBaseTypeOfInitializerTropicalFruitParameterUsed() {
        return getBaseTypeOfInitializerTropicalFruitParameterUsed;
    }

    public static boolean isGetTypeClosureUsed() {
        return getTypeClosureUsed;
    }

    public static boolean isGetBaseTypeOfBillDisposerParameterUsed() {
        return getBaseTypeOfBillDisposerParameterUsed;
    }

    public static boolean isGetBaseTypeOfObserverInjectionPointUsed() {
        return getBaseTypeOfObserverInjectionPointUsed;
    }

    public static boolean isGetTypeClosureOfProducerMethodUsed() {
        return getTypeClosureOfProducerMethodUsed;
    }

    public static boolean isGetBaseTypeOfObserverParameterUsed() {
        return getBaseTypeOfObserverParameterUsed;
    }

}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy