org.arquillian.cube.impl.client.enricher.CubeControllerProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of arquillian-cube-docker Show documentation

There is a newer version: 2.0.0.Alpha1

Show newest version

package org.arquillian.cube.impl.client.enricher;

import java.lang.annotation.Annotation;

import org.arquillian.cube.CubeController;
import org.jboss.arquillian.core.api.Instance;
import org.jboss.arquillian.core.api.annotation.Inject;
import org.jboss.arquillian.test.api.ArquillianResource;
import org.jboss.arquillian.test.spi.enricher.resource.ResourceProvider;

/**
 *
 * @author Stefan Miklosovic
 *
 */
public class CubeControllerProvider implements ResourceProvider {

    @Inject
    private Instance cubeController;

    @Override
    public boolean canProvide(Class type) {
        return CubeController.class.isAssignableFrom(type);
    }

    @Override
    public Object lookup(ArquillianResource arg0, Annotation... arg1) {
        CubeController cubeController = this.cubeController.get();

        if (cubeController == null) {
            throw new IllegalStateException("CubeController was not found.");
        }

        return cubeController;
    }

}