firewatir 1.6.2 → 1.6.5

data/lib/firewatir/elements/select_list.rb

@@ -0,0 +1,188 @@
+module FireWatir
+  #
+  # Description:
+  #   Class for SelectList element.
+  #
+  class SelectList < InputElement
+    INPUT_TYPES = ["select-one", "select-multiple"]
+
+    attr_accessor :o
+
+    #
+    # Description:
+    #   Clears the selected items in the select box.
+    #
+    def clear
+      assert_exists
+      #highlight( :set)
+      wait = false
+      each do |selectBoxItem|
+        if selectBoxItem.selected
+          selectBoxItem.selected = false
+          wait = true
+        end
+      end
+      self.wait if wait
+      #highlight( :clear)
+    end
+    alias clearSelection clear
+
+    def each
+      assert_exists
+      arr_options = js_options
+      #puts arr_options[0]#.length
+      for i in 0..arr_options.length - 1 do
+        yield Option.new(self, :jssh_name, arr_options[i])
+      end
+    end
+
+    #
+    # Description:
+    #   Get option element at specified index in select list.
+    #
+    # Input:
+    #   key - option index
+    #
+    # Output:
+    #   Option element at specified index
+    #
+    def [] (key)
+      assert_exists
+      arr_options = js_options
+      return Option.new(self, :jssh_name, arr_options[key - 1])
+    end
+
+    #
+    # Description:
+    #   Selects an item by text. If you need to select multiple items you need to call this function for each item.
+    #
+    # Input:
+    #   - item - Text of item to be selected.
+    #
+    def select( item )
+      select_items_in_select_list(:text, item)
+    end
+    alias :set :select
+
+    #
+    # Description:
+    #   Selects an item by value. If you need to select multiple items you need to call this function for each item.
+    #
+    # Input:
+    # - item - Value of the item to be selected.
+    #
+    def select_value( item )
+      select_items_in_select_list(:value, item)
+    end
+
+    #
+    # Description:
+    #   Gets all the items in the select list as an array.
+    #   An empty array is returned if the select box has no contents.
+    #
+    # Output:
+    #   Array containing the items of the select list.
+    #
+    def options
+      assert_exists
+      #element.log "There are #{@o.length} items"
+      returnArray = []
+      each { |thisItem| returnArray << thisItem.text }
+      return returnArray
+    end
+
+    alias getAllContents options
+
+    #
+    # Description:
+    #   Gets all the selected items in the select list as an array.
+    #   An empty array is returned if the select box has no selected item.
+    #
+    # Output:
+    #   Array containing the selected items of the select list.
+    #
+    def selected_options
+      assert_exists
+      returnArray = []
+      #element.log "There are #{@o.length} items"
+      each do |thisItem|
+        #puts "#{thisItem.selected}"
+        if thisItem.selected
+          #element.log "Item ( #{thisItem.text} ) is selected"
+          returnArray << thisItem.text
+        end
+      end
+      return returnArray
+    end
+
+    alias getSelectedItems selected_options
+
+    #
+    # Description:
+    #   Get the option using attribute and its value.
+    #
+    # Input:
+    #   - attribute - Attribute used to find the option.
+    #   - value - value of that attribute.
+    #
+    def option (attribute, value)
+      assert_exists
+      Option.new(self, attribute, value)
+    end
+    
+    private
+    
+    # Description:
+    #   Selects items from the select box.
+    #
+    # Input:
+    #   - name  - :value or :text - how we find an item in the select box
+    #   - item  - value of either item text or item value.
+    #
+    def select_items_in_select_list(attribute, value)
+      assert_exists
+      
+      attribute = attribute.to_s
+      found     = false
+      
+      value = value.to_s unless [Regexp, String].any? { |e| value.kind_of? e }
+
+      highlight( :set )
+      each do |option|
+        next unless value.matches(option.invoke(attribute))
+        found = true  
+        next if option.selected
+        
+        option.selected = true
+        fireEvent("onChange")
+        wait
+      end
+      highlight( :clear )
+
+      unless found
+        raise NoValueFoundException, "No option with #{attribute} of #{value.inspect} in this select element"
+      end
+      
+      value
+    end
+
+    #
+    # Description:
+    #   Gets all the options of the select list element.
+    #
+    # Output:
+    #   Array of option elements.
+    #
+    def js_options
+      jssh_socket.send("#{element_object}.options.length;\n", 0)
+      length = read_socket().to_i
+      # puts "options length is : #{length}"
+      arr_options = Array.new(length)
+      for i in 0..length - 1
+        arr_options[i] = "#{element_object}.options[#{i}]"
+      end
+      return arr_options
+    end
+  
+  end # Selects
+end # FireWatir

data/lib/firewatir/elements/table.rb

@@ -0,0 +1,218 @@
+module FireWatir
+  class Table < Element
+    attr_accessor :element_name
+    TAG = 'TABLE'
+
+    #   - how - Attribute to identify the table element.
+    #   - what - Value of that attribute.
+    def initialize(container, how, what)
+      @how = how
+      @what = what
+      @container = container
+      @o = nil
+      #super nil
+    end
+
+    #
+    # Description:
+    #   Locate the table element.
+    #
+    def locate
+      case @how
+      when :jssh_name
+        @element_name = @what
+      when :xpath
+        @element_name = element_by_xpath(@container, @what)
+      else
+        @element_name = locate_tagged_element('TABLE', @how, @what)
+      end
+      @o = self
+    end
+
+    #
+    # Description:
+    #   Override the highlight method, as if the tables rows are set to have a background color,
+    #   this will override the table background color,  and the normal flash method wont work
+    #
+    def highlight(set_or_clear )
+
+      if set_or_clear == :set
+        begin
+          @original_border = @o.border.to_i
+          if @o.border.to_i==1
+            @o.border = 2
+          else
+            @o.border=1
+          end
+        rescue
+          @original_border = nil
+        end
+      else
+        begin
+          @o.border= @original_border unless @original_border == nil
+          @original_border = nil
+        rescue
+          # we could be here for a number of reasons...
+        ensure
+          @original_border = nil
+        end
+      end
+      super
+    end
+
+    #
+    # Description:
+    #   Used to populate the properties in the to_s method.
+    #
+    #def table_string_creator
+    #    n = []
+    #    n <<   "rows:".ljust(TO_S_SIZE) + self.row_count.to_s
+    #    n <<   "cols:".ljust(TO_S_SIZE) + self.column_count.to_s
+    #    return n
+    #end
+    #private :table_string_creator
+
+    # returns the properties of the object in a string
+    # raises an ObjectNotFound exception if the object cannot be found
+    # TODO: Implement to_s method for this class.
+
+    def to_s
+      assert_exists
+      r = super({"rows" => "rows.length","columns" => "columnLength", "cellspacing" => "cellspacing", "cellpadding" => "cellpadding", "border" => "border"})
+      # r += self.column_count.to_s
+    end
+
+    #
+    # Description:
+    #   Gets the number of rows in the table.
+    #
+    # Output:
+    #   Number of rows.
+    #
+    def row_count
+      assert_exists
+      return rows.length
+    end
+
+    #
+    # Description:
+    #   Gets the table as a 2 dimensional array. Dont expect too much if there are nested tables, colspan etc.
+    #
+    # Output:
+    #   2D array with rows and column text of the table.
+    #
+    def to_a
+      assert_exists
+      y = []
+      table_rows = rows
+      for row in table_rows
+        x = []
+        row.each do |td|
+          x << td.to_s.strip
+        end
+        y << x
+      end
+      return y
+    end
+
+    #
+    # Description:
+    #   Gets the array of rows in the table.
+    #
+    # Output:
+    #   Array of rows.
+    #
+    def rows
+      assert_exists
+      arr_rows = get_rows
+      table_rows = Array.new(arr_rows.length)
+      for i in 0..arr_rows.length - 1 do
+        table_rows[i] = TableRow.new(@container, :jssh_name, arr_rows[i])
+      end
+      return table_rows
+    end
+
+    #
+    # Description:
+    #   Get row at particular index in table.
+    #
+    # Input:
+    #   key - row index
+    #
+    # Output:
+    #   Table Row element
+    #
+    def [](key)
+      assert_exists
+      arr_rows = rows
+      return arr_rows[key - 1]
+    end
+
+    #
+    # Desription:
+    #   Iterate over each table row element.
+    #
+    def each
+      assert_exists
+      arr_rows = rows
+      for i in 0..arr_rows.length - 1 do
+        yield arr_rows[i]
+      end
+    end
+
+    #
+    # Description:
+    #   Get column count of first row in the table.
+    #
+    # Output:
+    #   Number of columns in first row.
+    #
+    def column_count
+      assert_exists
+      arr_rows = rows
+      return arr_rows[0].column_count
+    end
+
+    #
+    # Description:
+    #   Get values of specified column in each row.
+    #
+    # Input:
+    #   Column number
+    #
+    # Output:
+    #   Values of column (specified as input) in each row
+    #
+    def column_values(column)
+      assert_exists
+      arr_rows = rows
+      values = Array.new(arr_rows.length)
+      for i in 0..arr_rows.length - 1 do
+        values[i] = arr_rows[i][column].to_s
+      end
+      return values
+    end
+
+    #
+    # Description:
+    #   Get values of all the column in specified row.
+    #
+    # Input:
+    #   Row number.
+    #
+    # Output:
+    #   Value of all columns present in the row.
+    #
+    def row_values(row)
+      assert_exists
+      arr_rows = rows
+      cells = arr_rows[row - 1].cells
+      values = Array.new(cells.length)
+      for i in 0..cells.length - 1 do
+        values[i] = cells[i].to_s
+      end
+      return values
+    end
+
+  end # Table
+end # FireWatir

data/lib/firewatir/elements/table_cell.rb

@@ -0,0 +1,54 @@
+module FireWatir
+  #
+  # Description:
+  # Class for Table Cell.
+  #
+  class TableCell < Element
+    attr_accessor :element_name
+
+    # Description:
+    #   Locate the table cell element on the page.
+    #
+    def locate
+      case @how
+      when :jssh_name
+        @element_name = @what
+      when :xpath
+        @element_name = element_by_xpath(@container, @what)
+      else
+        @element_name = locate_tagged_element("TD", @how, @what)
+      end
+      @o = self
+    end
+
+    #
+    # Description:
+    #   Initializes the instance of table cell object.
+    #
+    # Input:
+    #   - how - Attribute to identify the table cell element.
+    #   - what - Value of that attribute.
+    #
+    def initialize(container, how, what)
+      @how = how
+      @what = what
+      @container = container
+      #super nil
+    end
+
+    alias to_s text
+
+    #
+    # Description:
+    #   Gets the col span of table cell.
+    #
+    # Output:
+    #   Colspan of table cell.
+    #
+    def colspan
+      assert_exists
+      @o.colSpan
+    end
+
+  end # TableCell
+end # FireWatir