com.oneandone.ejbcdiunit.cdiunit.internal.TestScopeExtension Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ejb-cdi-unit Show documentation

A module that can be used together with cdiunit to build en ejb-test-environment.

The newest version!

/*
 * Copyright 2011 Bryn Cooke 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 com.oneandone.ejbcdiunit.cdiunit.internal;

import javax.enterprise.event.Observes;
import javax.enterprise.inject.spi.AnnotatedType;
import javax.enterprise.inject.spi.Extension;
import javax.enterprise.inject.spi.ProcessAnnotatedType;

import org.apache.deltaspike.core.api.literal.ApplicationScopedLiteral;
import org.apache.deltaspike.core.util.metadata.builder.AnnotatedTypeBuilder;

public class TestScopeExtension implements Extension {

    private static final ApplicationScopedLiteral APPLICATIONSCOPED = new ApplicationScopedLiteral();

    private Class testClass;

    public TestScopeExtension() {}

    public TestScopeExtension(Class testClass) {
        this.testClass = testClass;
    }


     void processAnnotatedType(@Observes ProcessAnnotatedType pat) {
        final AnnotatedType annotatedType = pat.getAnnotatedType();
        if (annotatedType.getJavaClass().equals(testClass)) {
            AnnotatedTypeBuilder builder = new AnnotatedTypeBuilder().readFromType(annotatedType).addToClass(APPLICATIONSCOPED);
            pat.setAnnotatedType(builder.create());
        }
    }

}