jp.vmi.selenium.selenese.locator.OptionIndexHandler Maven / Gradle / Ivy

package jp.vmi.selenium.selenese.locator;

import java.util.ArrayList;
import java.util.List;

import org.openqa.selenium.By;
import org.openqa.selenium.WebElement;

import com.thoughtworks.selenium.SeleniumException;

class OptionIndexHandler implements OptionLocatorHandler {

    @Override
    public String optionLocatorType() {
        return "index";
    }

    @Override
    public List handle(WebElement element, String arg) {
        try {
            int index = Integer.parseInt(arg);
            List options = element.findElements(By.tagName("option"));
            List result = new ArrayList();
            if (index < options.size())
                result.add(options.get(index));
            return result;
        } catch (NumberFormatException e) {
            throw new SeleniumException("Invalid index: " + arg);
        }
    }
}