com.redhat.darcy.webdriver.locators.ByAttribute.java Source code

Introduction

Here is the source code for com.redhat.darcy.webdriver.locators.ByAttribute.java

Source

/*
 Copyright 2014 Red Hat, Inc. and/or its affiliates.
    
 This file is part of darcy-webdriver.
    
 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.
    
 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.
    
 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.redhat.darcy.webdriver.locators;

import org.openqa.selenium.By;
import org.openqa.selenium.SearchContext;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.internal.FindsByXPath;

import java.util.List;

/**
 * {@link By} implementation that finds elements based on whether or not a 
 * specified attributes exactly matches some text.
 * 
 * @author ahenning
 */
public class ByAttribute extends By {
    protected final String attribute;
    protected final String word;

    /**
     * Returns instance of ByAttribute that will search under the "value" 
     * attribute.
     * @param value
     * @return
     */
    public static By value(String value) {
        return new ByAttribute("value", value);
    }

    public static By index(int index) {
        return new ByAttribute("index", String.valueOf(index));
    }

    public static By labelFor(String what) {
        return new ByAttribute("for", what);
    }

    public ByAttribute(String attribute, String word) {
        this.attribute = attribute;
        this.word = word;
    }

    @Override
    public List<WebElement> findElements(SearchContext context) {
        return ((FindsByXPath) context).findElementsByXPath(".//*[" + attributeEquals(attribute, word) + "]");
    }

    @Override
    public WebElement findElement(SearchContext context) {
        return ((FindsByXPath) context).findElementByXPath(".//*[" + attributeEquals(attribute, word) + "]");
    }

    @Override
    public int hashCode() {
        return toString().hashCode();
    }

    @Override
    public String toString() {
        return "ByAttribute(\"" + attribute + "\"): " + word;
    }

    private String attributeEquals(String attribute, String word) {
        return "@" + attribute + "='" + word + "'";
    }
}