Selenium-webdriver系列教程(13)————如何处理table

Table对象是自动化测试中经常需要处理的对象。由于webdriver中没有专门的table类,所以我们需要简单的封装出一个易用易扩展的Table类来帮助简化代码。

module EasyWrap
    class EasyWrapError < StandardError;end 
    class NotValidElementError < EasyWrapError;end  
    class IncorrectdIndexError < EasyWrapError;end  
 
    class TableBase
        attr_reader :e
        def initialize e
            raise NotValidElementError unless e.is_a?(Selenium::WebDriver::Element)
            @e = e 
        end
 
        def method_missing(m, *params, &blk)
            @e.send(m, *params) if @e.respond_to?(m)
        end
    end #TableBase
    class Table < TableBase
        def initialize e
            super(e)
            __rows
        end
        def __rows
            @rows = @e.find_elements(:css => 'tr')
        end
        private :__rows
 
        def rows
            all_rows = []   
            @rows.each { |r| rows << TableRow.new(r)}
            all_rows
        end
 
        def row_count
            @rows.size
        end
 
        def [](index)
            valid_index? index
            TableRow.new @rows[index]
        end 
 
        def valid_index?(index)
            raise IncorrectdIndexError if index.to_i > row_count
        end
    end #Table
 
    class TableRow < TableBase
        def initialize e
            super(e)
            __cells
        end
 
        def __cells 
            @cells = @e.find_elements(:tag_name => 'td') 
            # 如果找不到td那么试着去找th
            @cells = @e.find_elements(:tag_name => 'th') if @cells.empty?
        end
        private :__cells
 
        def cells
            all_cells = []
            @cells.each {|c| all_cells << TableCell.new(c)} 
        end
 
        def cell_count
            @cells.size
        end
 
        def [](index)
            valid_index? index
            TableCell.new @cells[index]
        end
 
        def valid_index?(index)
            raise IncorrectdIndexError if index.to_i > cell_count
        end
    end #TableRow
 
    class TableCell < TableBase
    end #TableCell
end #EasyWrap

EasyWrap定义了3个类

  • Table类,代表1个table对象,可以通过该对象来访问该table的行和列;

  • TableRow类 代表table的某1行,可以通过其访问table的某一列;

  • TableCell类 代表table的某1个单元格

假如我们需要点击table的第3行第1列,通过EasyWrap,我们可以这样来实现

talbe = EasyWrap::Table(dr.find_element(:id => 'table_id'))
table[2][0].click

以下面的html代码为例:

<html>
    <head>
        <title>Table</title>
        <style>
            table {border: 1px solid #ccc}
        </style>
    </head>
    <body>
        <table id = "t">
            <th>C1</th><th>C2</th><th>C3</th>
            <tr>
                <td>v1</td>
                <td>v2</td>
                <td>v3</td>
            </tr>
            <tr>
                <td>k1</td>
                <td>k2</td>
                <td>k3</td>
            </tr>
        </table>
    </body>
</html>

接下来的脚本实现了高亮页面上table元素首行首列的功能

require 'rubygems'
require 'selenium-webdriver'
require './easy_wrap'
include EasyWrap
 
dr = Selenium::WebDriver.for :firefox
table_file = 'file:///'+File.expand_path(File.join(File.dirname(__FILE__), 'table.html'))
dr.get table_file
t = dr.find_element(:id => 't')
table = Table.new t
p table.row_count
highlight = <<JS
    arguments[0].style.border = "2px solid red"
JS
dr.execute_script highlight, table[0][0].e
# 打印第2行第2列的值
puts table[1][1].text
posted @ 2011-10-27 21:22  乙醇  阅读(3095)  评论(0编辑  收藏  举报

友情链接 虫师的blog  测试教程网  重定向科技  省略