com.oneandone.cdiunit.internal.mockito.MockitoExtension 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.cdiunit.internal.mockito;

import java.util.Set;

import javax.enterprise.context.spi.CreationalContext;
import javax.enterprise.event.Observes;
import javax.enterprise.inject.spi.Extension;
import javax.enterprise.inject.spi.InjectionPoint;
import javax.enterprise.inject.spi.InjectionTarget;
import javax.enterprise.inject.spi.ProcessInjectionTarget;

import org.mockito.MockitoAnnotations;

public class MockitoExtension implements Extension {
    public  void process(@Observes ProcessInjectionTarget event) {
        final InjectionTarget injectionTarget = event.getInjectionTarget();
        event.setInjectionTarget(new InjectionTarget() {

            public T produce(CreationalContext ctx) {
                T o = injectionTarget.produce(ctx);
                MockitoAnnotations.initMocks(o);
                return o;
            }

            public void dispose(T instance) {
                injectionTarget.dispose(instance);
            }

            public Set getInjectionPoints() {
                return injectionTarget.getInjectionPoints();
            }

            public void inject(T instance, CreationalContext ctx) {
                injectionTarget.inject(instance, ctx);
            }

            public void postConstruct(T instance) {
                injectionTarget.postConstruct(instance);
            }

            public void preDestroy(T instance) {
                injectionTarget.preDestroy(instance);
            }
        });
    }
}