org.fluentlenium.core.proxy.AtIndexElementLocator Maven / Gradle / Ivy

package org.fluentlenium.core.proxy;

import org.openqa.selenium.WebElement;
import org.openqa.selenium.support.pagefactory.ElementLocator;

import org.fluentlenium.core.domain.ElementUtils;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;

/**
 * {@link ElementLocator} retrieving a particular index element from another locator.
 */
public class AtIndexElementLocator implements ElementLocator {
    private final int index;
    protected ElementLocator listLocator;

    /**
     * Creates a new at-index element locator.
     *
     * @param listLocator element list locator
     * @param index       index to retrieve
     */
    public AtIndexElementLocator(ElementLocator listLocator, int index) {
        this.listLocator = listLocator;
        this.index = index;
    }

    @Override
    public WebElement findElement() {
        WebElement element = getWebElementAtIndex();
        if (element == null) {
            throw ElementUtils.noSuchElementException("Element " + this);
        }
        return element;
    }

    @Override
    public List findElements() {
        WebElement element = getWebElementAtIndex();
        if (element == null) {
            return Collections.emptyList();
        }
        return Arrays.asList(element);
    }

    @Override
    public String toString() {
        return listLocator.toString() + " (index=" + index + ")";
    }

    private WebElement getWebElementAtIndex() {
        List elements = listLocator.findElements();
        if (index >= elements.size()) {
            return null;
        }
        return elements.get(index);
    }
}