super-test 1.0.9

data/lib/common/data_model.rb

@@ -0,0 +1,87 @@
+#!/usr/bin/env ruby
+# coding: utf-8
+#
+# Author:: hanyu
+# Date:: 2009.5.7
+
+if __FILE__ == $0
+	$LIB_ROOT = "#{File.dirname(__FILE__)}/.."
+	$TEST_ROOT = "#{$LIB_ROOT}/.."
+	$: << "#{$LIB_ROOT}" 
+	require 'common/log_helper'
+	require 'common/config_helper'
+end
+
+
+# User 表示一个用户
+# 目前的实现只是一个普通的数据模型，没有行为
+class User
+
+	attr_accessor :name, :password, :url, :portrait, :sex
+	
+	# 创建User
+	# - name 	用户名，必填项
+	# - password	密码，选填项，默认值 passme123
+	# - url		url前缀名，选填项，默认与name相同
+	# - portrait	头像加密串，选填项，默认为nil
+	# - sex		性别，1男，2女，默认值1
+	#
+	# Example #1:
+	#
+	#	user_A = User.new "ruguo"
+	#	user_B = User.new "ruguo2","1234"
+	#	user_C = User.new "ruguo4","1234","ruguo","fl3k2jr23jr5o2u5o214"
+	def initialize(name, password=ConfigHelper::CONF['default_password'], url=name, portrait=nil, sex=1)
+		# 用户名
+		@name = name
+		# 用户密码
+		@password = password
+		# 空间url
+		@url = url
+		# 头像加密串 uid+uname
+		@portrait = portrait
+		# 性别
+		@sex = sex
+	end
+
+	# 重载之，方便日志打印
+	#
+	# Example #1:
+	#
+	#	user = User.new "ruguo","pass","ruguo"
+	#	puts user
+	#	# => #<User:0x00000000b19220 @name="ruguo", @password="pass", @space_url="ruguo", @portrait=nil>
+	def to_s
+		inspect
+	end
+
+	# 重载==，方便User对象之间的比较
+	#
+	# Example #1：
+	#
+	#	a = User.new "ruguo","pass","ruguo"
+	#	b = User.new "ruguo","pass33","ruguo33"
+	#	c = User.new "ruguo11","pass33","ruguo33"
+	#	d = Object.new
+	#
+	#	puts a == b	#=> true
+	#	puts a == c	#=> false
+	#	puts a == d	#=> false
+	def ==(obj)
+		obj.kind_of?(User) and obj.name == @name 
+	end
+	
+end
+
+if __FILE__ == $0
+	u = User.new "ruguo","pass","ruguo"
+	p u
+	puts u
+	b = User.new "ruguo","pass33","ruguo33"
+	c = User.new "ruguo1","pass","ruguo33"
+	d = Object.new
+	puts u == b
+	puts u == c
+	puts u == d
+end
+

data/lib/common/http/assert_helper.rb

@@ -0,0 +1,343 @@
+#!/usr/bin/env ruby
+# coding: utf-8
+#
+# Author:: hanyu
+# Date:: 2009.5.7
+
+if __FILE__ == $0
+    $LIB_ROOT = "#{File.dirname(__FILE__)}/.."
+    $TEST_ROOT = "#{$LIB_ROOT}/.."
+    $: << "#{$LIB_ROOT}" 
+    require 'common/log_helper'
+    require 'common/config_helper'
+    require 'common/http_helper'
+    require 'common/html_helper'
+end
+
+
+# case中尽量使用经过封装的assert
+#
+# - 扩展assert方法，支持html、json等解析
+# - 自定义上层可读message
+# - 打印可读日志
+module AssertHelper
+
+    # === 功能:
+    # 验证返回码是200
+    # 
+    # === 参数解释:
+    # - msg         失败之后的提示文字
+    #
+    # === Example:
+    # Example:
+    #   get "hi.baidu.com"
+    #   assert_ok
+    def assert_ok(msg=nil)
+        return if @is_press
+        raise "must send http request before assert!" if @response == nil
+        msg = "assert_ok fail!" if msg.nil?
+        assert_equal @response.code, "200", msg
+        $log.debug "assert_ok success!"
+    end
+
+    # === 功能:
+    # 验证HTTP请求的返回码
+    # 
+    # === 参数解释:
+    # - code        Fixnum，返回码
+    # - msg         失败之后的提示文字
+    #
+    # === Example:
+    # Example:
+    #   get "http://www.google.com/not_exist_page"
+    #   assert_code 404
+    def assert_code(code, msg=nil)
+        return if @is_press
+        raise "must send http request before assert!" if @response == nil
+        msg = "assert_code fail!" if msg.nil?
+        assert_equal @response.code, code.to_s, msg
+        $log.debug "assert_code code[#{code}] success!"
+    end
+    
+    # === 功能:
+    # 验证重定向
+    # 
+    # === 参数解释:
+    # - url         要跳转的url
+    # - msg         失败之后的提示文字
+    #
+    # === Example:
+    # Example:
+    #   get "http://hi.baidu.com/sys/not_exist_page"
+    #   assert_redirected_to "error.html"
+    def assert_redirected_to(url, msg=nil)
+        return if @is_press
+        raise "must send http request before assert!" if @response == nil
+        msg = "assert_redirected_to fail!" if msg.nil?
+        assert_equal @response.code, "302", msg
+        assert @response.header.key?("Location"), msg
+        assert_equal @response.header["Location"], url, msg
+        $log.debug "assert_redirected_to url[#{url}] success!"
+    end
+
+    # === 功能:
+    # grep方式匹配
+    # 
+    # === 参数解释:
+    # - regexp      正则表达式
+    # - msg         失败之后的提示文字
+    #
+    # === Example:
+    #   get "http://hi.baidu.com/"
+    #   assert_grep /百度空间/
+    #
+    def assert_grep(regexp, msg=nil)
+        return if @is_press
+        raise "must send http request before assert!" if @response == nil
+        msg = "assert_grep fail!" if msg.nil?
+        assert_match regexp, @response.body, msg
+        $log.debug "assert_grep regexp[#{regexp}] success!"
+    end
+
+    # === 功能:
+    # 见 assert_grep
+    def assert_no_grep(regexp,msg=nil)
+        return if @is_press
+        raise "must send http request before assert!" if @response == nil
+        msg = "assert_no_grep fail!" if msg.nil?
+        exp = /#{Regexp.escape(regexp)}/ if String === regexp
+        assert_no_match exp, @response.body, msg
+        $log.debug "assert_no_grep regexp[#{regexp}] success!"
+    end
+
+    # === 功能:
+    # 断言包含某字符串
+    # 
+    # === 参数解释:
+    # - str        字符串，不能是正则表达式
+    # - msg        失败之后的提示文字
+    #
+    # === Example:
+    #    get "http://hi.baidu.com/"
+    #    assert_has "百度空间", "something wrong!"
+    def assert_has(str,msg=nil)
+        return if @is_press
+        assert find_has?(str), msg
+        $log.debug "assert_has str[#{str}] success!"
+    end
+    # === 功能:
+    # 与assert_has相反
+    def assert_has_not(str,msg=nil)
+        return if @is_press
+        assert !find_has?(str), msg
+        $log.debug "assert_has_not str[#{str}] success!"
+    end
+
+    # === 功能:
+    # 断言元素存在
+    # 
+    # === 参数解释:
+    # - xpath    定位，可以使xpath or selector
+    # - msg        失败之后的提示文字
+    #
+    def assert_exist(xpath, msg=nil, &blk)
+        return if @is_press
+        msg = "assert_exist xpath[#{xpath}] fail!" if msg.nil?
+        assert find_exist?(xpath,&blk), msg
+        $log.debug "assert_exist xpath[#{xpath}] success!"
+    end
+
+    # === 功能:
+    # 断言元素不存在
+    # 
+    # === 参数解释:
+    # - xpath       定位，可以使xpath or selector
+    # - msg         失败之后的提示文字
+    #
+    def assert_no_exist(xpath, msg=nil, &blk)
+        return if @is_press
+        msg = "assert_no_exist xpath[#{xpath}]" if msg.nil?
+        assert_false find_exist?(xpath,&blk), msg
+        $log.debug "assert_no_exist xpath[#{xpath}] success!"
+    end
+
+    # === 功能:
+    # 验证某元素内text的内容
+    # 
+    # === 参数解释:
+    # - xpath       定位，可以使xpath or selector
+    # - regexp      期待的结果，可以是正则表达式
+    # - msg         失败之后的提示文字
+    #
+    # === Example:
+    # Example #1:
+    #   get "http://hi.baidu.com/"
+    #   assert_text "#m_space", "更多精彩空间"
+    #   
+    def assert_text(xpath, regexp, msg=nil, &blk)
+        return if @is_press
+        raise "must send http request before assert!" if @response == nil
+        expect = find_text xpath, &blk
+        msg = "assert_text xpath[#{xpath}] regexp[#{regexp}] fail!" if msg.nil?
+        assert_match regexp, expect, msg
+        $log.debug "assert_text xpath[#{xpath}] regexp[#{regexp}] success!"
+    end
+
+    # === 功能:
+    # 见 assert_text
+    def assert_no_text(xpath, regexp, msg=nil, &blk)
+        return if @is_press
+        raise "must send http request before assert!" if @response == nil
+        expect = find_text xpath, &blk
+        exp = /#{Regexp.escape(regexp)}/ if String === regexp
+        msg = "assert_no_text xpath[#{xpath}] regexp[#{regexp}] fail!" if msg.nil?
+        assert_no_match exp, expect, msg
+        $log.debug "assert_no_text xpath[#{xpath}] regexp[#{regexp}] success!"
+    end
+
+    # === 功能:
+    # 验证某元素的html片段的内容
+    # 
+    # === 参数解释:
+    # - xpath       定位，可以使xpath or selector
+    # - regexp      期待的结果，可以是正则表达式
+    # - msg         失败之后的提示文字
+    #
+    # === Example:
+    # Example #1:
+    #   get "http://hi.baidu.com/"
+    #   assert_html "#m_space", "更多精彩空间"
+    #
+    # Example #2:
+    #   get "http://hi.baidu.com/"
+    #   assert_html "//div[@id='m_space']", "更多精彩空间"
+    def assert_html(xpath, regexp, msg=nil, &blk)
+        return if @is_press
+        raise "must send http request before assert!" if @response == nil
+        expect = find_html xpath, &blk
+        msg = "assert_html xpath[#{xpath}] regexp[#{regexp}] fail!" if msg.nil?
+        assert_match regexp, expect, "assert_html xpath[#{xpath}] regexp[#{regexp}] fail!"
+        $log.debug "assert_html xpath[#{xpath}] regexp[#{regexp}] success!"
+    end
+
+    # === 功能:
+    # 见 assert_html
+    def assert_no_html(xpath, regexp, msg=nil, &blk)
+        return if @is_press
+        raise "must send http request before assert!" if @response == nil
+        expect = find_html xpath, &blk
+        exp = /#{Regexp.escape(regexp)}/ if String === regexp
+        msg = "assert_no_html xpath[#{xpath}] regexp[#{regexp}] fail!" if msg.nil?
+        assert_no_match exp, expect, msg
+        $log.debug "assert_no_html xpath[#{xpath}] regexp[#{regexp}] success!"
+    end
+
+    # === 功能:
+    # 验证某Element的属性值
+    # 
+    # === 参数解释:
+    # - xpath       定位，可以使xpath or selector
+    # - attr        属性名
+    # - regexp      期待的结果，可以是正则表达式
+    # - msg         失败之后的提示文字
+    #
+    # === Example:
+    # Example:
+    #   get "http://hi.baidu.com/"
+    #   assert_attr "div#m_space div.t1 a.grnlk", "href" "http://hi.baidu.com/hi/more/wonderful/"
+    #   # html代码片段
+    #   # <div id="m_space"> 
+    #   # <div class="t1">
+    #   #     <a href="http://hi.baidu.com/hi/more/wonderful/" target="_blank" class="grnlk"><img src="http://img.baidu.com/hi/img/index/i1.gif" border="0" align="top"> 空间中的海外华人  </a>
+    #   # ...
+    def assert_attr(xpath, attr, regexp, msg=nil, &blk)
+        return if @is_press
+        raise "must send http request before assert!" if @response == nil
+        expect = find_attr xpath, attr, &blk
+        msg = "assert_attr xpath[#{xpath}] attr[#{attr}] regexp[#{regexp}] fail!" if msg.nil?
+        assert_match regexp, expect, msg
+        $log.debug "assert_attr xpath[#{xpath}] attr[#{attr}] regexp[#{regexp}] success!"
+    end
+
+    # === 功能:
+    # 见 assert_attr
+    def assert_no_attr(xpath, attr, regexp, msg=nil, &blk)
+        return if @is_press
+        raise "must send http request before assert!" if @response == nil
+        expect = find_attr xpath, attr, &blk
+        exp = /#{Regexp.escape(regexp)}/ if String === regexp
+        msg = "assert_no_attr xpath[#{xpath}] attr[#{attr}] regexp[#{regexp}] fail!" if msg.nil?
+        assert_no_match exp, expect, msg
+        $log.debug "assert_no_attr xpath[#{xpath}] attr[#{attr}] regexp[#{regexp}] success!"
+    end
+
+    # === 功能:
+    # 验证匹配的个数，如当前页某栏目的内容数
+    # 
+    # === 参数解释:
+    # - xpath       定位，可以使xpath or selector
+    # - size        期待的结果，Fixnum
+    # - msg         失败之后的提示文字
+    #
+    # === Example:
+    # Example:
+    #   get "http://hi.baidu.com/"
+    #   assert_size "div#m_space ul li", 12
+    #   # 百度空间首页，精彩空间栏会出12个推荐空间
+    def assert_size(xpath, size, msg=nil, &blk)
+        return if @is_press
+        raise "must send http request before assert!" if @response == nil
+        expect = find_size xpath, &blk
+        msg = "assert_size xpath[#{xpath}] size[#{size}] expect[#{expect}] fail!" if msg.nil?
+        assert expect == size, msg
+        $log.debug "assert_size xpath[#{xpath}] size[#{size}] expect[#{expect}] success!"
+    end
+    
+    # === 功能:
+    # 验证某元素不含有非预计的儿子节点
+    # 
+    # === 参数解释:
+    # - xpath       父亲节点的定位，可以使xpath or selector
+    # - children    预计的儿子节点名字的集合，如%w{title author body}
+    # - msg         失败之后的提示文字
+    #
+    # === Example:
+    # Example:
+    #   get "http://hi.baidu.com/"
+    #   assert_children "head", %w{title script link meta}      
+    #   #=> assert_children xpath[head] has children[["title", "script", "link", "meta"]], AND has other children[["style"]]!.
+    #
+    def assert_children(xpath, children, msg=nil, &blk)
+        return if @is_press
+        raise "must send http request before assert!" if @response == nil
+        expect = find_children xpath,children, &blk
+        
+        msg = "assert_children xpath[#{xpath}] has children[#{children}], AND has other children[#{expect}]!" if msg.nil?
+        assert expect.empty?, msg
+        $log.debug "assert_children xpath[#{xpath}] children[#{children}] success!"
+    end
+
+end
+
+
+if __FILE__ == $0
+    class MyTest
+    include AssertHelper
+    include HttpHelper
+
+        def initialize
+            @cookies = {}
+        end
+
+        def test
+            get "http://www.baidu.com/"
+            doc = Hpricot(@response.body)
+            s = doc.search("div#u")
+            p s
+            puts s.html
+            puts s.attr("id")
+        end
+    end
+    Test.new.test
+end
+

data/lib/common/http/fix_hpricot.rb

@@ -0,0 +1,118 @@
+#!/usr/bin/env ruby
+# coding: utf-8
+#
+# Author:: hanyu
+# Date:: 2009.9.6
+
+
+# hack Hpricot
+# 因为 Hpricot 将实体字符进行了反转义，这个过程存在很多的编码问题，解决起来不太容易，
+# 而且这种反转义可能并不是我们需要的，所以去掉了
+module Hpricot # :nodoc:
+    def self.uxs(str)
+        str.to_s.force_encoding("gbk")
+        #gsub(/\&(\w+);/){ [NamedCharacters[$1] || ??].pack("U*") }
+        #gsub(/\&\#(\d+);/){ [$1.to_i].pack("U*").encode("gbk") }.force_encoding("gbk")
+    end
+end
+
+
+# 基于hpricot (0.8.2)修改，如果版本发生变化，可能会有问题
+
+
+# 尝试fix Hpricot在解析xpath的时候 对带冒号的字符串匹配有问题
+# 可能会存在引入新bug的风险，对Hpricot的解析方式没有完全搞懂
+module Hpricot
+    module Traverse
+        def search(expr, &blk)
+            if Range === expr
+                return Elements.expand(at(expr.begin), at(expr.end), expr.exclude_end?)
+            end
+            last = nil
+            nodes = [self]
+            done = []
+            expr = expr.to_s
+            hist = []
+            until expr.empty?
+                expr = clean_path(expr)
+                expr.gsub!(%r!^//!, '')
+
+                case expr
+                when %r!^/?\.\.!
+                    last = expr = $'
+                    nodes.map! { |node| node.parent }
+                when %r!^[>/]\s*!
+                    last = expr = $'
+                    nodes = Elements[*nodes.map { |node| node.children if node.respond_to? :children }.flatten.compact]
+                when %r!^\+!
+                    last = expr = $'
+                    nodes.map! do |node|
+                        siblings = node.parent.children
+                        siblings[siblings.index(node)+1]
+                    end
+                    nodes.compact!
+                when %r!^~!
+                    last = expr = $'
+                    nodes.map! do |node|
+                        siblings = node.parent.children
+                        siblings[(siblings.index(node)+1)..-1]
+                    end
+                    nodes.flatten!
+                when %r!^[|,]!
+                    last = expr = " #$'"
+                    nodes.shift if nodes.first == self
+                    done += nodes
+                    nodes = [self]
+                else
+                    #puts "++++++++++++++++"
+                    #p   expr
+                    m = expr.match(%r!^([#.]?)([a-z0-9\\*_-]*)!i).to_a
+                    #p   m
+                    after = $'
+                    #p   after
+                    #mt = after[%r!:[a-z0-9\\*_-]+!i, 0]
+                    mt = after[%r!^:[a-z0-9\\*_-]+!i, 0]
+                    #p   mt
+                    oop = false
+                    if mt and not (mt == ":not" or Traverse.method_defined? "filter[#{mt}]")
+                        after = $' 
+                        m[2] += mt
+                        expr = after
+                    end
+                    if m[1] == '#'
+                        oid = get_element_by_id(m[2])
+                        nodes = oid ? [oid] : []
+                        expr = after
+                    else
+                        m[2] = "*" if after =~ /^\(\)/ || m[2] == "" || m[1] == "."
+                        ret = []
+                        nodes.each do |node|
+                            case m[2]
+                            when '*'
+                                node.traverse_element { |n| ret << n }
+                            else
+                                if node.respond_to? :get_elements_by_tag_name
+                                  ret += [*node.get_elements_by_tag_name(m[2])] - [*(node unless last)]
+                                end
+                            end
+                        end
+                        nodes = ret
+                    end
+                    last = nil
+                end
+
+                hist << expr
+                break if hist[-1] == hist[-2]
+                nodes, expr = Elements.filter(nodes, expr)
+            end
+            nodes = done + nodes.flatten.uniq
+            if blk
+                nodes.each(&blk)
+                self
+            else
+                Elements[*nodes]
+            end
+        end
+    end
+end
+