com.salesforceiq.augmenteddriver.web.pageobjects.WebPageObjectActions Maven / Gradle / Ivy

package com.salesforceiq.augmenteddriver.web.pageobjects;

import com.google.common.base.Preconditions;
import com.google.common.base.Predicate;
import com.google.inject.Inject;
import com.google.inject.Injector;
import com.google.inject.Provider;
import com.salesforceiq.augmenteddriver.web.AugmentedWebDriver;
import com.salesforceiq.augmenteddriver.web.AugmentedWebElement;
import com.salesforceiq.augmenteddriver.web.AugmentedWebFunctions;
import com.salesforceiq.augmenteddriver.util.PageObjectWaiter;
import org.openqa.selenium.TimeoutException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Implementation based on Guice of WebPageObjectActionsInterface.
 */
public class WebPageObjectActions implements WebPageObjectActionsInterface {
    private static final Logger LOG = LoggerFactory.getLogger(WebPageObjectActions.class);

    @Inject
    private Injector injector;

    @Inject
    private PageObjectWaiter waiter;

    /**
     * Important we use a Provider, since we need the driver to be initialized when the first test starts to run
     * not at creation time, like Guice wants.
     */
    @Inject
    private Provider driverProvider;

    @Override
    public  T get(Class clazz) {
        Preconditions.checkNotNull(clazz);
        T instance = injector.getInstance(clazz);
        try {
            instance.assertPresent();
        } catch (TimeoutException | AssertionError e) {
            LOG.error(String.format("Page Object %s, not found, message: %s", clazz.getCanonicalName(), e.getMessage()));
            throw e;
        }
        return instance;
    }

    @Override
    public  T get(Class clazz, Predicate waitUntil) {
        Preconditions.checkNotNull(clazz);
        Preconditions.checkNotNull(waitUntil);

        T result = get(clazz);
        waiter.waitUntil(result, waitUntil);
        return result;
    }

    @Override
    public  T get(Class clazz, AugmentedWebElement container) {
        Preconditions.checkNotNull(clazz);
        Preconditions.checkNotNull(container);

        T instance = injector.getInstance(clazz);
        instance.setContainer(container);
        try {
            instance.assertPresent();
        } catch (TimeoutException | AssertionError e) {
            LOG.error(String.format("Page Object Container %s, not found, message: %s", clazz.getCanonicalName(), e.getMessage()));
            throw e;
        }
        return instance;
    }


    @Override
    public  T get(Class clazz, AugmentedWebElement container, Predicate waitUntil) {
        Preconditions.checkNotNull(clazz);
        Preconditions.checkNotNull(container);
        Preconditions.checkNotNull(waitUntil);

        T result = get(clazz, container);
        waiter.waitUntil(result, waitUntil);
        return result;
    }

    @Override
    public AugmentedWebDriver driver() {
        Preconditions.checkNotNull(driverProvider);

        return Preconditions.checkNotNull(driverProvider.get());
    }

    @Override
    public AugmentedWebFunctions augmented() {
        Preconditions.checkNotNull(driverProvider);
        Preconditions.checkNotNull(driverProvider.get());

        return Preconditions.checkNotNull(driverProvider.get().augmented());
    }

    @Override
    public PageObjectWaiter waiter() {
        return waiter;
    }
}