uk.sponte.automation.seleniumpom.fieldInitialisers.FieldInitialiserForSeleniumLegacyWebElements Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of selenium-pom Show documentation

There is a newer version: 3.0.14

package uk.sponte.automation.seleniumpom.fieldInitialisers;

import com.google.inject.Inject;
import com.google.inject.Provider;
import org.openqa.selenium.SearchContext;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.internal.Locatable;
import org.openqa.selenium.internal.WrapsElement;
import org.openqa.selenium.support.pagefactory.Annotations;
import uk.sponte.automation.seleniumpom.PageFactory;
import uk.sponte.automation.seleniumpom.dependencies.DependencyInjector;
import uk.sponte.automation.seleniumpom.helpers.FrameWrapper;
import uk.sponte.automation.seleniumpom.orchestration.WebDriverFrameSwitchingOrchestrator;
import uk.sponte.automation.seleniumpom.proxies.handlers.WebElementHandler;

import java.lang.reflect.Field;
import java.lang.reflect.Proxy;

/**
 * Created by evops on 02/02/2016.
 */
public class FieldInitialiserForSeleniumLegacyWebElements
        implements FieldInitialiser {
    @Inject
    private DependencyInjector dependencyInjector;

    @Inject
    private WebDriverFrameSwitchingOrchestrator webDriverFrameSwitchingOrchestrator;

    @Inject private Provider pageFactory;

    @Override
    public Boolean initialiseField(Field field, Object page,
            SearchContext searchContext, FrameWrapper frame) {
        {
            if (!FieldAssessor.isValidWebElement(field))
                return false;

            Annotations annotations = new Annotations(field);

            try {
                WebElementHandler elementHandler = new WebElementHandler(
                        dependencyInjector, searchContext,
                        annotations.buildBy(), frame,
                        webDriverFrameSwitchingOrchestrator);
                pageFactory.get().addListener(elementHandler);

                WebElement proxyElement = (WebElement) Proxy.newProxyInstance(
                        WebElement.class.getClassLoader(),
                        new Class[] { WebElement.class, Locatable.class,
                                SearchContext.class, WrapsElement.class },
                        elementHandler
                );

                field.setAccessible(true);
                field.set(page, proxyElement);
            } catch (IllegalAccessException e) {
                e.printStackTrace();
            }
            return true;
        }
    }
}