Contents

require 'watir/elements/html_elements'

module PageObject
  module Elements
    class TableRow < Element
      include Enumerable

      #
      # iterator that yields with a PageObject::Elements::TableCell
      #
      # @return [PageObject::Elements::TableCell]
      #
      def each
        for index in 1..self.columns do
          yield self[index-1]
        end
      end

      #
      # Return the PageObject::Elements::TableCell for the index provided.  Index
      # is zero based.  If the index provided is a String then it
      # will be matched with the text from the columns in the first row.
      # The text can be a substring of the full column text.
      #
      def [](idx)
        idx = find_index_by_title(idx) if idx.kind_of?(String)
        return nil unless idx && columns >= idx + 1
        initialize_cell(element[idx])
      end

      #
      # Returns the number of columns in the table.
      #
      def columns
        element.wd.find_elements(:xpath, child_xpath).size
      end

      protected

      def child_xpath
        ".//child::td|th"
      end

      def initialize_cell(row_element)
        ::PageObject::Elements::TableCell.new(row_element)
      end

      def find_index_by_title(title)
        table = element.parent
        table = table.parent if table.tag_name == 'tbody'
        if table.instance_of? Watir::HTMLElement
          table = table.to_subtype
        end
        first_row = table[0]
        first_row.cells.find_index {|column| column.text.include? title }
      end

    end

    ::PageObject::Elements.tag_to_class[:tr] = ::PageObject::Elements::TableRow
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
page-object-2.2	lib/page-object/elements/table_row.rb