Contents

module Watir
  class ButtonLocator < ElementLocator

    def locate_all
      find_all_by_multiple
    end

    private

    def wd_find_first_by(how, what)
      if how == :tag_name
        how  = :xpath
        what = ".//button | .//input[#{attribute_expression :type => Button::VALID_TYPES}]"
      end

      super
    end

    def build_xpath(selectors)
      return if selectors.values.any? { |e| e.kind_of? Regexp }

      selectors.delete(:tag_name) || raise("internal error: no tag_name?!")

      @building = :button
      button_attr_exp = attribute_expression(selectors)

      @building = :input
      selectors[:type] = Button::VALID_TYPES
      input_attr_exp = attribute_expression(selectors)

      xpath = ".//button"
      xpath << "[#{button_attr_exp}]" unless button_attr_exp.empty?
      xpath << " | .//input"
      xpath << "[#{input_attr_exp}]"

      p :build_xpath => xpath if $DEBUG

      xpath
    end

    def lhs_for(key)
      if @building == :input && key == :text
        "@value"
      elsif @building == :button && key == :value
        "text()"
      else
        super
      end
    end

    def matches_selector?(element, rx_selector)
      rx_selector = rx_selector.dup
      tag_name    = element.tag_name.downcase

      [:value, :caption].each do |key|
        if rx_selector.has_key?(key)
          correct_key = tag_name == 'button' ? :text : :value
          rx_selector[correct_key] = rx_selector.delete(key)
        end
      end

      super
    end

    def tag_name_matches?(tag_name, _)
      !!(/^(input|button)$/ === tag_name)
    end

    def validate_element(element)
      return if element.tag_name.downcase == "input" && !Button::VALID_TYPES.include?(element.attribute(:type))
      super
    end

  end # ButtonLocator
end # Watir

Version data entries

25 entries across 25 versions & 1 rubygems

Version	Path
watir-webdriver-0.5.3	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.5.2	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.5.1	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.5.0	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.4.1	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.4.0	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.3.9	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.3.8	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.3.7	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.3.6	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.3.5	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.3.4	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.3.3	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.3.2	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.3.1	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.3.0	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.2.9	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.2.8	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.2.7	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.2.6	lib/watir-webdriver/locators/button_locator.rb