/*
* 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.weld.tests.extensions.annotatedType;
import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.junit.Arquillian;
import org.jboss.shrinkwrap.api.Archive;
import org.jboss.shrinkwrap.api.BeanArchive;
import org.jboss.shrinkwrap.api.ShrinkWrap;
import org.jboss.weld.manager.BeanManagerImpl;
import org.jboss.weld.test.util.Utils;
import org.jboss.weld.tests.category.Integration;
import org.jboss.weld.tests.extensions.annotatedType.EcoFriendlyWashingMachine.EcoFriendlyWashingMachineLiteral;
import org.junit.Assert;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
import javax.enterprise.inject.Any;
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.Bean;
import javax.enterprise.inject.spi.Extension;
import javax.enterprise.inject.spi.InjectionPoint;
import javax.inject.Inject;
import java.lang.reflect.Constructor;
import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.Set;
@Category(Integration.class)
@RunWith(Arquillian.class)
public class AnnotatedTypeExtensionTest {
@Deployment
public static Archive<?> deploy() {
return ShrinkWrap.create(BeanArchive.class, Utils.getDeploymentNameAsHash(AnnotatedTypeExtensionTest.class))
.addPackage(AnnotatedTypeExtensionTest.class.getPackage())
.addClass(Utils.class)
.addAsServiceProvider(Extension.class, AnnotatedTypeExtension.class);
}
@Inject
private BeanManagerImpl beanManager;
@Test
public void testMultipleBeansOfSameType(Laundry laundry) {
Assert.assertNotNull(laundry.ecoFriendlyWashingMachine);
Assert.assertNotNull(laundry.fastWashingMachine);
}
/*
* description = "WELD-371"
*/
@Test
public void testAnnotationsAreOverridden() {
Bean<WashingMachine> bean = Utils.getBean(beanManager, WashingMachine.class, EcoFriendlyWashingMachineLiteral.INSTANCE);
Assert.assertTrue(Utils.annotationSetMatches(bean.getQualifiers(), Any.class, EcoFriendlyWashingMachine.class));
// Verify overriding the class structure works
Clothes.reset();
TumbleDryer tumbleDryer = Utils.getReference(beanManager, TumbleDryer.class);
Bean<TumbleDryer> tumbleDryerBean = Utils.getBean(beanManager, TumbleDryer.class);
Assert.assertNotNull(tumbleDryer);
Assert.assertFalse(containsConstructor(tumbleDryerBean.getInjectionPoints(), SerialNumber.class));
Assert.assertTrue(containsConstructor(tumbleDryerBean.getInjectionPoints(), Clothes.class));
Assert.assertNull(tumbleDryer.getSerialNumber());
Assert.assertNotNull(tumbleDryer.getClothes());
Assert.assertFalse(Clothes.getInjectionPoint().getAnnotated().isAnnotationPresent(Original.class));
AnnotatedConstructor<?> clothesConstructor = getConstructor(tumbleDryerBean.getInjectionPoints(), Clothes.class);
Assert.assertTrue(clothesConstructor.getParameters().get(0).isAnnotationPresent(Special.class));
Assert.assertFalse(clothesConstructor.getParameters().get(0).isAnnotationPresent(Original.class));
Assert.assertTrue(containsField(tumbleDryerBean.getInjectionPoints(), "plug"));
Assert.assertFalse(containsField(tumbleDryerBean.getInjectionPoints(), "coins"));
Assert.assertNotNull(tumbleDryer.getPlug());
Assert.assertNull(tumbleDryer.getCoins());
Assert.assertTrue(containsMethod(tumbleDryerBean.getInjectionPoints(), "setRunningTime", RunningTime.class));
Assert.assertFalse(containsMethod(tumbleDryerBean.getInjectionPoints(), "setHotAir", HotAir.class));
Assert.assertNotNull(tumbleDryer.getRunningTime());
Assert.assertNull(tumbleDryer.getHotAir());
AnnotatedMethod<?> runningTimeMethod = getMethod(tumbleDryerBean.getInjectionPoints(), "setRunningTime", RunningTime.class);
Assert.assertTrue(runningTimeMethod.getParameters().get(0).isAnnotationPresent(Special.class));
Assert.assertFalse(runningTimeMethod.getParameters().get(0).isAnnotationPresent(Original.class));
}
private static boolean containsField(Set<InjectionPoint> injectionPoints, String name) {
for (InjectionPoint ip : injectionPoints) {
if (ip.getAnnotated() instanceof AnnotatedField<?>) {
AnnotatedField<?> field = (AnnotatedField<?>) ip.getAnnotated();
if (field.getJavaMember().getName().equals(name)) {
return true;
}
}
}
return false;
}
private static boolean containsConstructor(Set<InjectionPoint> injectionPoints, Class<?>... parameters) {
return getConstructor(injectionPoints, parameters) != null;
}
private static AnnotatedConstructor<?> getConstructor(Set<InjectionPoint> injectionPoints, Class<?>... parameters) {
for (InjectionPoint ip : injectionPoints) {
if (ip.getAnnotated() instanceof AnnotatedParameter<?>) {
AnnotatedParameter<?> param = (AnnotatedParameter<?>) ip.getAnnotated();
if (param.getDeclaringCallable() instanceof AnnotatedConstructor<?>) {
Class<?>[] parameterTypes = ((Constructor<?>) param.getDeclaringCallable().getJavaMember()).getParameterTypes();
if (Arrays.equals(parameters, parameterTypes)) {
return (AnnotatedConstructor<?>) param.getDeclaringCallable();
}
}
}
}
return null;
}
private static boolean containsMethod(Set<InjectionPoint> injectionPoints, String name, Class<?>... parameters) {
return getMethod(injectionPoints, name, parameters) != null;
}
private static AnnotatedMethod<?> getMethod(Set<InjectionPoint> injectionPoints, String name, Class<?>... parameters) {
for (InjectionPoint ip : injectionPoints) {
if (ip.getAnnotated() instanceof AnnotatedParameter<?>) {
AnnotatedParameter<?> param = (AnnotatedParameter<?>) ip.getAnnotated();
if (param.getDeclaringCallable() instanceof AnnotatedMethod<?>) {
Class<?>[] parameterTypes = ((Method) param.getDeclaringCallable().getJavaMember()).getParameterTypes();
String methodName = param.getDeclaringCallable().getJavaMember().getName();
if (Arrays.equals(parameters, parameterTypes) && methodName.equals(name)) {
return (AnnotatedMethod<?>) param.getDeclaringCallable();
}
}
}
}
return null;
}
}