io.webdevice.support.SpringDeviceRegistry Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of webdevice-spring Show documentation

WebDevice.IO Spring Module

There is a newer version: 0.0.11

package io.webdevice.support;

import io.webdevice.device.Device;
import io.webdevice.device.DeviceNotProvidedException;
import io.webdevice.device.DeviceProvider;
import io.webdevice.device.DeviceRegistry;
import org.openqa.selenium.WebDriver;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.BeanFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Component;

import static io.cucumber.spring.CucumberTestContext.SCOPE_CUCUMBER_GLUE;

@Component
@Scope(SCOPE_CUCUMBER_GLUE)
public class SpringDeviceRegistry
        implements DeviceRegistry {
    private final BeanFactory factory;

    @Autowired
    public SpringDeviceRegistry(BeanFactory factory) {
        this.factory = factory;
    }

    @Override
    public  Device provide(String device) {
        DeviceProvider provider = providerOf(device);
        return provider.get();
    }

    @Override
    public  void release(Device device) {
        DeviceProvider provider = providerOf(device.getName());
        provider.accept(device);
    }

    @SuppressWarnings("unchecked")
    private  DeviceProvider providerOf(String device) {
        try {
            return factory.getBean(device, DeviceProvider.class);
        } catch (BeansException e) {
            throw new DeviceNotProvidedException(device, e);
        }
    }
}