com.automationrockstars.gir.ui.context.SearchContextService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gir-webdriver Show documentation

The newest version!

/*
 * 
 */

package com.automationrockstars.gir.ui.context;

import com.automationrockstars.design.gir.webdriver.DriverFactory;
import com.google.common.base.Predicate;
import com.google.common.collect.Iterators;
import org.openqa.selenium.SearchContext;

import java.util.Iterator;
import java.util.NoSuchElementException;
import java.util.ServiceLoader;

public class SearchContextService {

    public static SearchContext provideForWeb() {
        return DriverFactory.getDriver();
    }

    public static SearchContext provideForImage() {
        Iterator providers = ServiceLoader.load(SearchContextProvider.class).iterator();
        try {
            SearchContextProvider result = Iterators.find(providers, new Predicate() {

                @Override
                public boolean apply(SearchContextProvider input) {
                    return input.canProvide(Image.class);
                }
            });
            return result.provide();
        } catch (NoSuchElementException e) {
            throw new RuntimeException("Cannot find provider for @Image context", e);
        }

    }
}