/*
* JBoss, Home of Professional Open Source
* Copyright 2016, 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.beanManager.annotatedType;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import java.net.URL;
import java.net.URLClassLoader;
import javax.enterprise.context.Dependent;
import javax.enterprise.inject.spi.AnnotatedMethod;
import javax.enterprise.inject.spi.AnnotatedType;
import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.junit.Arquillian;
import org.jboss.classfilewriter.AccessFlag;
import org.jboss.classfilewriter.ClassFile;
import org.jboss.classfilewriter.code.CodeAttribute;
import org.jboss.shrinkwrap.api.Archive;
import org.jboss.shrinkwrap.api.BeanArchive;
import org.jboss.shrinkwrap.api.ShrinkWrap;
import org.jboss.weld.annotated.slim.AnnotatedTypeIdentifier;
import org.jboss.weld.annotated.slim.backed.BackedAnnotatedType;
import org.jboss.weld.manager.BeanManagerImpl;
import org.jboss.weld.resources.ClassTransformer;
import org.jboss.weld.test.util.Utils;
import org.jboss.weld.tests.category.EmbeddedContainer;
import org.jboss.weld.util.bytecode.ClassFileUtils;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
/**
*
* @author Martin Kouba
* @see WELD-2062
*/
@Category(EmbeddedContainer.class)
@RunWith(Arquillian.class)
public class CreateAnnotatedTypeWithIdTest {
@Deployment
public static Archive<?> createTestArchive() {
return ShrinkWrap.create(BeanArchive.class, Utils.getDeploymentNameAsHash(CreateAnnotatedTypeWithIdTest.class)).addClass(Component.class);
}
@Test
public void testCreateAnnotatedTypeWithId(BeanManagerImpl beanManager) {
AnnotatedType<Component> annotatedType = beanManager.createAnnotatedType(Component.class);
assertTrue(annotatedType.isAnnotationPresent(Dependent.class));
assertFalse(hasPongMethod(annotatedType));
// Create a different class with the same name
ClassFile componentClassFile = new ClassFile(Component.class.getName(), Object.class.getName());
// Add void pong()
CodeAttribute b = componentClassFile.addMethod(AccessFlag.of(AccessFlag.PUBLIC, AccessFlag.SYNTHETIC), "pong", "V").getCodeAttribute();
b.returnInstruction();
Class<?> componentClass = ClassFileUtils.toClass(componentClassFile, new URLClassLoader(new URL[] {}), null);
@SuppressWarnings("unchecked")
BackedAnnotatedType<Component> newAnnotatedType = (BackedAnnotatedType<Component>) beanManager.createAnnotatedType(componentClass,
componentClass.getName() + componentClass.getClassLoader().hashCode());
assertFalse(newAnnotatedType.isAnnotationPresent(Dependent.class));
assertTrue(hasPongMethod(newAnnotatedType));
// Dispose annotated type
AnnotatedTypeIdentifier identifier = newAnnotatedType.getIdentifier();
beanManager.disposeAnnotatedType(componentClass, componentClass.getName() + componentClass.getClassLoader().hashCode());
assertNull(beanManager.getServices().get(ClassTransformer.class).getSlimAnnotatedTypeById(identifier));
}
private boolean hasPongMethod(AnnotatedType<?> annotatedType) {
for (AnnotatedMethod<?> annotatedMethod : annotatedType.getMethods()) {
if (annotatedMethod.getJavaMember().getName().equals("pong")) {
return true;
}
}
return false;
}
}