watir-webdriver 0.0.1.dev

data/lib/watir-webdriver/elements/headings.rb

@@ -0,0 +1,48 @@
+# encoding: utf-8
+module Watir
+
+  #
+  # H1 is handled by the generated code
+  #
+
+
+  class H2 < Heading
+    default_selector.clear
+
+    identifier         :tag_name => 'h2'
+    container_method   :h2
+    collection_method  :h2s
+  end
+
+  class H3 < Heading
+    default_selector.clear
+
+    identifier         :tag_name => 'h3'
+    container_method   :h3
+    collection_method  :h3s
+  end
+
+  class H4 < Heading
+    default_selector.clear
+
+    identifier         :tag_name => 'h4'
+    container_method   :h4
+    collection_method  :h4s
+  end
+
+  class H5 < Heading
+    default_selector.clear
+
+    identifier         :tag_name => 'h5'
+    container_method   :h5
+    collection_method  :h5s
+  end
+
+  class H6 < Heading
+    default_selector.clear
+
+    identifier         :tag_name => 'h6'
+    container_method   :h6
+    collection_method  :h6s
+  end
+end

data/lib/watir-webdriver/elements/hidden.rb

@@ -0,0 +1,18 @@
+# encoding: utf-8
+module Watir
+  class Hidden < Input
+    identifier :type => 'hidden'
+
+    container_method  :hidden
+    collection_method :hiddens
+    
+    # deprecate?
+    def value=(val)
+      raise NotImplementedError
+    end
+    
+    def visible?
+      false
+    end
+  end
+end

data/lib/watir-webdriver/elements/image.rb

@@ -0,0 +1,33 @@
+# encoding: utf-8
+module Watir
+  class Image < HTMLElement
+
+    alias_method :loaded?, :complete?
+
+    def width
+      assert_exists
+      driver.execute_script "return arguments[0].width", @element
+    end
+
+    def height
+      assert_exists
+      driver.execute_script "return arguments[0].height", @element
+    end
+
+    def file_created_date
+      assert_exists
+      raise NotImplementedError
+    end
+
+    def file_size
+      assert_exists
+      raise NotImplementedError
+    end
+
+    def save(path)
+      assert_exists
+      raise NotImplementedError
+    end
+
+  end # Image
+end # Watir

data/lib/watir-webdriver/elements/input.rb

@@ -0,0 +1,39 @@
+# encoding: utf-8
+module Watir
+  class Input
+
+    alias_method :readonly?, :read_only?
+
+    def enabled?
+      !disabled?
+    end
+
+    #
+    # not sure about this
+    #
+    # this is mostly useful if you're using Browser#element_by_xpath, and want to
+    # 'cast' the returned Input instance to one of the subclasses
+    #
+
+    def to_checkbox
+      assert_exists
+      Watir::CheckBox.new(@parent, :element, @element)
+    end
+
+    def to_radio
+      assert_exists
+      Watir::Radio.new(@parent, :element, @element)
+    end
+
+    def to_button
+      assert_exists
+      Watir::Button.new(@parent, :element, @element)
+    end
+
+    def to_select_list
+      assert_exists
+      Watir::SelectList.new(@parent, :element, @element)
+    end
+
+  end # Input
+end # Watir

data/lib/watir-webdriver/elements/link.rb

@@ -0,0 +1,13 @@
+# encoding: utf-8
+module Watir
+  class Link < Anchor
+    default_selector.clear
+
+    identifier        :tag_name => 'a'
+    container_method  :link
+    collection_method :links
+
+    alias_method :url, :href # deprecate?
+
+  end # Link
+end # Watir

data/lib/watir-webdriver/elements/option.rb

@@ -0,0 +1,36 @@
+# encoding: utf-8
+module Watir
+  class Option < HTMLElement
+
+    def select
+      assert_exists
+      @element.select
+    end
+    
+    def toggle
+      assert_exists
+      @element.toggle
+    end
+    
+    def selected?
+      assert_exists
+      @element.selected?
+    end
+
+    def text
+      assert_exists
+
+      # A little unintuitive - we'll return the 'label' or 'text' attribute if
+      # they exist, otherwise the inner text of the element
+
+      attribute = [:label, :text].find { |a| attribute? a }
+
+      if attribute
+        @element.attribute(attribute)
+      else
+        @element.text
+      end
+    end
+
+  end # Option
+end # Watir

data/lib/watir-webdriver/elements/radio.rb

@@ -0,0 +1,23 @@
+# encoding: utf-8
+module Watir
+  class Radio < Input
+    include SharedRadioCheckbox
+
+    identifier :type => 'radio' # a text field is the default for input elements, so this needs to be changed
+
+    container_method  :radio
+    collection_method :radios
+
+    def set(bool = true)
+      assert_exists
+      assert_enabled
+
+      if set?
+        @element.click unless bool
+      else
+        @element.click if bool
+      end
+    end
+
+  end
+end

data/lib/watir-webdriver/elements/select_list.rb

@@ -0,0 +1,73 @@
+# encoding: utf-8
+module Watir
+  class SelectList < Select
+    include Watir::Exception
+
+    container_method  :select_list
+    collection_method :select_lists
+
+    def enabled?
+      !disabled?
+    end
+
+    def clear
+      assert_exists
+      raise "you can only clear multi-selects" unless multiple?
+      options.each do |o|
+        o.toggle if o.selected?
+      end
+    end
+
+    def includes?(str_or_rx)
+      assert_exists
+      options.any? { |e| str_or_rx === e.text }
+    end
+
+    def select(str_or_rx)
+      select_by :text, str_or_rx
+    end
+
+    # TODO: deprecate?
+    def select_value(str_or_rx)
+      select_by :value, str_or_rx
+    end
+
+    def selected?(str_or_rx)
+      assert_exists
+      matches = options.select { |e| str_or_rx === e.text }
+
+      if matches.empty?
+        raise UnknownObjectException, "Unable to locate option matching #{str_or_rx.inspect}"
+      end
+
+      matches.any? { |e| e.selected? }
+    end
+
+    def value
+      o = options.find { |e| e.selected? }
+      return if o.nil?
+
+      o.value
+    end
+
+    def selected_options
+      assert_exists
+      options.map { |e| e.text if e.selected? }.compact
+    end
+
+    private
+
+    def select_by(how, str_or_rx)
+      assert_exists
+      os = options.select { |e| str_or_rx === e.send(how) }
+
+      if os.empty?
+        raise NoValueFoundException, "#{str_or_rx.inspect} not found in select list"
+      end
+
+      os.each { |e| e.select unless e.selected? }
+      os.first.text
+    end
+
+  end
+end

data/lib/watir-webdriver/elements/shared_radio_checkbox.rb

@@ -0,0 +1,13 @@
+# encoding: utf-8
+module Watir
+  module SharedRadioCheckbox
+    def set?
+      assert_exists
+      @element.selected?
+    end
+
+    def clear
+      set false
+    end
+  end # SharedRadioCheckbox
+end # Watir

data/lib/watir-webdriver/elements/table.rb

@@ -0,0 +1,18 @@
+# encoding: utf-8
+module Watir
+  class Table < HTMLElement
+    
+    def to_a
+      assert_exists
+      
+      trs.inject [] do |res, row|
+        res << row.tds.map { |cell| cell.text }
+      end
+    end
+    
+    def [](idx)
+      row(:index, idx)
+    end
+
+  end # Table
+end # Watir

data/lib/watir-webdriver/elements/table_row.rb

@@ -0,0 +1,17 @@
+module Watir
+  class TableRow < HTMLElement
+
+    def [](idx)
+      td(:index, idx)
+    end
+
+    def locate
+      if @parent.kind_of?(Watir::Table)
+        TableRowLocator.new(@parent.wd, @selector, self.class.attribute_list).locate
+      else
+        super
+      end
+    end
+
+  end # TableRow
+end # Watir

data/lib/watir-webdriver/elements/text_field.rb

@@ -0,0 +1,60 @@
+# encoding: utf-8
+module Watir
+  class TextField < Input
+
+    # perhaps we'll need a custom locate(), since this should also cover textareas
+    attributes Watir::TextArea.typed_attributes
+
+    container_method  :text_field
+    collection_method :text_fields
+
+    def locate
+      TextFieldLocator.new(@parent.wd, @selector, self.class.attribute_list).locate
+    end
+
+    def inspect
+      '#<%s:0x%x located=%s selector=%s>' % [self.class, hash*2, !!@element, selector_without_type.inspect]
+    end
+
+    def selector_string
+      selector_without_type.inspect
+    end
+
+    def set(*args)
+      assert_exists
+      assert_writable
+
+      @element.clear
+
+      append(*args)
+    end
+    alias_method :value=, :set
+
+    def append(*args)
+      assert_exists
+      assert_writable
+
+      @element.send_keys(*args)
+    end
+
+    def clear
+      assert_exists
+      @element.clear
+    end
+
+
+    def value
+      # since 'value' is an attribute on input fields, we override this here
+      assert_exists
+      @element.value
+    end
+
+    private
+
+    def selector_without_type
+      s = @selector.dup
+      s[:type] = '(any text type)'
+      s
+    end
+  end
+end

data/lib/watir-webdriver/exception.rb

@@ -0,0 +1,20 @@
+# encoding: utf-8
+
+module Watir
+  module Exception
+    class Error < StandardError; end
+
+    # TODO: rename Object -> Element?
+    class UnknownObjectException < Error; end
+    class ObjectDisabledException < Error; end
+    class ObjectReadOnlyException < Error; end
+    class NoValueFoundException < Error; end
+    class MissingWayOfFindingObjectException < Error; end
+    class UnknownCellException < Error; end
+    class NoMatchingWindowFoundException < Error; end
+    class NoStatusBarException < Error; end
+    class NavigationException < Error; end
+    class UnknownFrameException < Error; end
+
+  end # Exception
+end # Watir

data/lib/watir-webdriver/locators/button_locator.rb

@@ -0,0 +1,57 @@
+module Watir
+  class ButtonLocator < ElementLocator
+
+    def locate_all
+      find_all_by_multiple
+    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] = %w[button reset submit image]
+      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?(rx_selector, element)
+      rx_selector = rx_selector.dup
+
+      [:value, :caption].each do |key|
+        if rx_selector.has_key?(key)
+          correct_key = element.tag_name == 'button' ? :text : :value
+          rx_selector[correct_key] = rx_selector.delete(key)
+        end
+      end
+
+      super
+    end
+
+    def tag_name_matches?(element, _)
+      !!(/^(input|button)$/ === element.tag_name)
+    end
+  end # ButtonLocator
+end # Watir