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_wd_selector(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_wd_selector => xpath if $DEBUG

      [:xpath, 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

10 entries across 10 versions & 1 rubygems

Version	Path
watir-webdriver-0.6.4	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.6.3	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.6.2	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.6.1	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.6.0	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.5.8	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.5.7	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.5.6	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.5.5	lib/watir-webdriver/locators/button_locator.rb
watir-webdriver-0.5.4	lib/watir-webdriver/locators/button_locator.rb