super-test 1.0.9

data/lib/patterns/inc.rb

@@ -0,0 +1,10 @@
+$LOAD_PATH << File.join(File.dirname(__FILE__), '../../conf/')
+$LOAD_PATH << File.join(File.dirname(__FILE__), '.')
+#!/usr/bin/env ruby
+#coding: utf-8
+
+require 'some_pattern'
+require 'mc_base_pattern'
+
+include SomePatternHelper
+include MCBasePatternHelper

data/lib/patterns/mc_base_pattern.rb

@@ -0,0 +1,474 @@
+#!/usr/bin/env ruby
+# coding: utf-8
+#============================================================
+# mc base pattern
+# Author:: liqiang01
+# Data:: 2009-8-28 
+#============================================================
+module MCBasePatternHelper
+	include Env
+	include Network
+	include Check
+	include Stub
+	
+	#===============================================
+	# send 
+	#===============================================
+  
+	# ===功能
+	# send_mcpack
+	# ===参数
+	# - ip target ip
+	# - port target port
+	# - nshead_hash nshead hash, nil to use default value
+	# - module_name the testing module name
+	# - packno sending mcpack file number
+	# - postfix=nil the postfix of the data
+	# - delay=0 sending interval(s)
+	# ===example
+	# mc_send "127.0.0.1", nil, 1
+	def mc_send ip, port, nshead_hash, packno, postfix=nil, delay=0
+	        $log.debug "sending mcpack #{packno}..."
+	        puts "\n##########sending mcpack #{packno}...##########"
+	      
+	        #case_path = "#{CASE_ROOT}#{$globals[:MODULE_NAME]}/#{$globals[:CASE_NAME]}/"
+		    case_path = "#{$CASE_ROOT}/"
+		    recv_path = ""
+		    if postfix == nil
+			    recv_path = "#{case_path}tmp/#{packno}.recv"
+		    else
+			    recv_path = "#{case_path}tmp/#{packno}_#{postfix}.recv"
+		    end
+	        nshead_send(ip, port, nshead_hash, "#{case_path}data/#{packno}.send", recv_path, delay)
+	        
+	        #set :SEND,:RECV to $globals
+	        Context.set :SEND, "#{case_path}data/#{packno}.send"
+	        Context.set :RECV, "#{case_path}tmp/#{packno}.recv"
+	
+	        $log.debug "send end..."
+
+		    #copy res_nshead.data to module case's tmp dir
+		    if File.exist? "#{MCSEND_PATH}res_nshead.data"
+			    `cp #{MCSEND_PATH}res_nshead.data #{case_path}tmp/`
+		    end
+	
+		    #set :RES_HEAD to $globalsdd
+		    Context.set :RES_HEAD, "#{case_path}tmp/res_nshead.data"
+	end
+
+	# ===功能
+	# send_mcpack 为通用transfer定制
+	# ===参数
+	# - ip target ip
+	# - port target port
+	# - nshead_hash nshead hash, nil to use default value
+	# - module_name the testing module name
+	# - packno sending mcpack file number
+	# - postfix=nil the postfix of the data
+	# - delay=0 sending interval(s)
+	# ===example
+	# mc_trans_send "127.0.0.1", nil, 1
+	def mc_trans_send ip, port, nshead_hash, packno, postfix=nil, delay=0
+	        $log.debug "sending mcpack #{packno}..."
+	        puts "\n##########sending mcpack #{packno}...##########"
+	      
+	        #case_path = "#{CASE_ROOT}#{$globals[:MODULE_NAME]}/#{$globals[:CASE_NAME]}/"
+		    case_path = "#{$CASE_ROOT}/"
+		    recv_path = ""
+		    if postfix == nil
+			    recv_path = "#{case_path}tmp/#{packno}.recv"
+		    else
+			    recv_path = "#{case_path}tmp/#{packno}_#{postfix}.recv"
+		    end
+	        nshead_trans_send(ip, port, nshead_hash, "#{case_path}data/#{packno}.send", recv_path, delay)
+	        
+	        #set :SEND,:RECV to $globals
+	        Context.set :SEND, "#{case_path}data/#{packno}.send"
+	        Context.set :RECV, "#{case_path}tmp/#{packno}.recv"
+	
+	        $log.debug "send end..."
+
+		    #copy res_nshead.data to module case's tmp dir
+		    if File.exist? "#{MCSEND_PATH}res_nshead.data"
+			    `cp #{MCSEND_PATH}res_nshead.data #{case_path}tmp/`
+		    end
+	
+		    #set :RES_HEAD to $globalsdd
+		    Context.set :RES_HEAD, "#{case_path}tmp/res_nshead.data"
+	end
+
+	# ===功能
+	# 支持下游桩的mc_send
+	# ===参数
+	# - ip
+	# - port
+ 	# - nshead_hash
+	# - prefix:　.../$casedir/data/$prefix.send
+	# - postfix: tells which stub return what data, .../$casedir/stub/$prefix_$stubname_$postfix.stub; nil by default means the response data is $prefix_$stubname.stub
+	# - delay: sending interval
+	# ===example
+	# mc_stub_send ip, 3308, nil, 1
+	def mc_stub_send ip, port, nshead_hash, prefix, postfix=nil, delay=0
+		$log.debug "sending mcpack #{prefix}, with stub data #{postfix}..."
+		puts "sending mcpack #{prefix}, with stub data #{postfix}..."
+
+		#case_path = "#{CASE_ROOT}#{$globals[:MODULE_NAME]}/#{$globals[:CASE_NAME]}/"
+		case_path = "#{$CASE_ROOT}/"
+
+		# modify stub's server.conf
+		stubs = Context.get :STUBS
+		heads = {}
+		files = {}
+		stubs.each{|stub|
+			if postfix != nil
+				heads[stub] = "#{case_path}stub/#{prefix}_#{stub}_#{postfix}.head"
+				files[stub] = "#{case_path}stub/#{prefix}_#{stub}_#{postfix}.stub"
+			else
+				heads[stub] = "#{case_path}stub/#{prefix}_#{stub}.head"
+				files[stub] = "#{case_path}stub/#{prefix}_#{stub}.stub"
+			end
+		}
+		set_stub_head heads
+		set_stub_file files
+		sleep 1
+
+
+		#send
+		mc_send ip, port, nshead_hash, prefix, postfix
+	end
+
+
+	#===============================================
+	# asserts
+	#===============================================
+
+	# ===功能
+	# 校验两组prefix相同的.expc, .recv文件是否内容一致
+	# ===参数
+	# - prefix
+	# - msg=""
+	# ===example
+	# mc_assert_pack 1
+	def mc_assert_pack prefix, msg=""
+		module_name = $globals[:MODULE_NAME]
+		case_name = $globals[:CASE_NAME]
+		#case_path = "#{CASE_ROOT}#{module_name}/#{case_name}/"
+		case_path = "#{$CASE_ROOT}/"
+		
+		postfixes = get_postfixes(module_name, case_name, prefix)
+		if postfixes == nil
+			flunk "get postfix from data dir fail, check the data dir and the .expc files"
+		end
+		postfixes.each{|postfix|
+		  assert_file_same_by_str(case_path+"data/" + prefix + ".expc" + postfix, case_path+"tmp/" + prefix + ".recv" + postfix, msg)
+		}
+	end
+	
+	# ===功能
+	# 校验两组prefix相同的.expc, .recv文件是否内容包含
+	# ===参数
+	# - prefix
+	# - msg=""
+	# ===example
+	# mc_assert_pack_include 1
+	def mc_assert_pack_include prefix, msg=""
+		module_name = $globals[:MODULE_NAME]
+		case_name = $globals[:CASE_NAME]
+		#case_path = "#{CASE_ROOT}#{module_name}/#{case_name}/"
+		case_path = "#{$CASE_ROOT}/"
+		
+		postfixes = get_postfixes(module_name, case_name, prefix)
+		if postfixes == nil
+			flunk "get postfix from data dir fail, check the data dir and the .expc files"
+		end
+		postfixes.each{|postfix|
+		  assert_file_include_by_str(case_path+"data/" + prefix + ".expc" + postfix, case_path+"tmp/" + prefix + ".recv" + postfix, msg)
+		}
+	end
+
+	# ===功能
+	# 校验响应头中指定key的期望值是否正确
+	# ===参数
+	# - key
+	# - expect_value
+	# - msg=""
+	# ===exmaple
+	# mc_assert_head_value "name", "new"
+	def mc_assert_head_value key, expect_value, msg=""
+		assert_head_value $globals[:RES_HEAD], key, expect_value, msg
+	end
+
+
+	#============helper methods=============
+		
+	# ===功能
+	# return all the pack numbers in the specified module and case
+	# ===example
+	# ar = getpacks
+	def getpacks(suffix=".send")
+		#data_path = "#{CASE_ROOT}#{$globals[:MODULE_NAME]}/#{$globals[:CASE_NAME]}/data/"
+		# 保证返回的datai是数字类型的且已经排序
+		data_path = "#{$CASE_ROOT}/data/"
+		datai = []
+		datas = get_all_pack data_path, suffix
+		datas.each{ |item|
+			datai.push(item.to_i)
+		}
+		$log.debug "Get packnos from " + data_path
+		return datai.sort
+	end
+	
+	# ===功能
+	# return all the pack in the specified module and case
+	# ===example
+	# ar = getpacks_all
+	def getpacks_all(suffix=".send")
+		#data_path = "#{CASE_ROOT}#{$globals[:MODULE_NAME]}/#{$globals[:CASE_NAME]}/data/"
+		# 保证返回data目录下的所有待发送的数据文件 *.send
+		data_path = "#{$CASE_ROOT}/data/"
+		datai = []
+		datas = get_all_pack data_path, suffix
+		datas.each{ |item|
+			datai.push(item)
+		}
+		$log.debug "Get packnos from " + data_path
+		# 按照字母表进行排序
+		return datai.sort
+	end
+
+	# ===功能
+	# get the packs query by pattern
+	# ===example
+	# packs = get_packs
+	def get_packs pattern
+		#data_path = "#{CASE_ROOT}#{$globals[:MODULE_NAME]}/#{$globals[:CASE_NAME]}/data/"
+		data_path = "#{$CASE_ROOT}/data/"
+		
+		packs = []
+		files = Dir.glob(File.join(data_path, pattern))
+		files.each{|file|
+			arr = file.split /\//
+			entry = arr[arr.size-1]
+			pack = entry[0, entry.size-5]
+			packs << pack
+		}
+		
+		return packs
+	end
+	
+
+	# ===功能
+	# get all the postfix of a specified packno
+	# === Example
+	#
+	#    data/1.expc, 1.expc.1, 1.expc.2
+	#    return array ["", ".1", ".2"]
+	def get_postfixes(module_name, case_name, packno)
+		#data_path = "#{CASE_ROOT}#{module_name}/#{case_name}/data/"
+		data_path = "#{$CASE_ROOT}/data/"
+		$log.debug "get postfixes from #{data_path} where packno = #{packno}"
+		
+		#param check
+		if !File.directory?(data_path) || !File.exist?(data_path) then
+			$log.debug "bad data path=" + data_path
+			return nil
+		end
+		
+		postfixes = Array.new
+		Dir.foreach(data_path) do |entry| 
+			if entry.match("#{packno}.expc.*") then
+				matches = entry.scan(/.*expc(.*)/)
+				postfixes << matches[0][0]
+			end
+		end
+		if postfixes.size == 0 then 
+			return nil
+			$log.warn "packno=#{packno} should have at least one expct data"
+		end
+		
+		$log.debug "postfixes = #{postfixes} and size = #{postfixes.size}"
+		return postfixes
+	end
+	
+	# ===功能
+	# mc_base模式
+	#
+	# ===参数
+	# 无
+	# 
+	# ===Example
+	# mc_base_pattern.each{ |packno|
+	# 	monitor_log IP, LOG_PATH, LOG
+	# 	mc_send IP, PORT, nil, packno
+	# 	对于给通用tranfer模块发送，请使用:
+	#	mc_trans_send IP, PORT, nil, packno
+	# 	mc_assert_pack "#{packno}", "@#{packno}: expect pack is not equal to recv pack!"
+	# 	assert_log_match_times IP, LOG_PATH, LOG, ["NOTICE"], "users_getInfo", 1
+	# }
+	#
+	# 备注：
+	# 能保证发送的数据按照前缀的顺序发送
+	def mc_base_pattern()
+		puts "+++++++++about to execute case #{Context.get 'CASE_NAME'}+++++++++"
+        $log.info "+++++++++about to execute case #{Context.get 'CASE_NAME'}+++++++++"
+		datas = getpacks
+
+        if datas != nil && datas.length > 0 then
+			return datas
+        else
+			return nil
+		end
+        puts "+++++++++end executing case #{Context.get 'CASE_NAME'}+++++++++"
+        $log.info "+++++++++end executing case #{Context.get 'CASE_NAME'}+++++++++"
+	end
+
+
+	# ===功能
+	# mc_base模式的变种,将发送data目录下的所有数据文件
+	#
+	# ===参数
+	# 无
+	# 
+	# ===Example
+	# mc_base_all_pattern.each{ |packno|
+	# 	monitor_log IP, LOG_PATH, LOG
+	# 	mc_send IP, PORT, nil, packno
+	# 	对于给通用tranfer模块发送，请使用:
+	#	mc_trans_send IP, PORT, nil, packno
+	# 	mc_assert_pack "#{packno}", "@#{packno}: expect pack is not equal to recv pack!"
+	# 	assert_log_match_times IP, LOG_PATH, LOG, ["NOTICE"], "users_getInfo", 1
+	# }
+	#
+	# 备注：
+	# 能保证发送的数据按照前缀的顺序发送
+	def mc_base_all_pattern()
+		puts "+++++++++about to execute case #{Context.get 'CASE_NAME'}+++++++++"
+        $log.info "+++++++++about to execute case #{Context.get 'CASE_NAME'}+++++++++"
+		datas = getpacks_all
+
+        if datas != nil && datas.length > 0 then
+			return datas
+        else
+			return nil
+		end
+        puts "+++++++++end executing case #{Context.get 'CASE_NAME'}+++++++++"
+        $log.info "+++++++++end executing case #{Context.get 'CASE_NAME'}+++++++++"
+	end
+
+
+        # ===功能
+        # cm transfer测试模式
+        # 发送cm，检查cm返回数据，检查transfer发送数据
+	    # 通过前缀来进行数据的匹配检查，不要求每个cm的发送的数据对有对应的transfer返回数据
+	    # 例如： 1.send 对应1.expc， 不要求同时有 1.trans，如果有则进行transfer返回的数据的检验
+	    #
+        # 
+        # ===参数
+        #
+        # ===Example
+	    # def test_xxx
+        # 	cm_transfer_pattern
+	    # end
+	    # def teardown
+	    #	cm_transfer_pattern_stop
+	    # end
+        # 
+	    # 备注：
+        # 增加teardown函数处理case运行失败的情况,见: cm_transfer_pattern_stop
+	    # 能保证发送的数据按照前缀的顺序发送
+        def cm_transfer_pattern(ip=HOST, port=PORT)
+                # 清理cm transfer原来的数据
+                ssh_cmd "rm -fr /home/space/phpcm/data/*; killall -9 phpcm;"
+                ssh_cmd "rm -fr /home/space/phptrans/data/transid/*; killall -9 transfer;"
+                sleep 2
+                cmd_str = "cd #{MCSERVER_PATH}; rm log/foo.log; nohup ./mcserver_trans &"
+                system "#{cmd_str}"
+                datas = getpacks
+                datas.each{ |packno|
+                        mc_send ip, port, nil, packno
+                        mc_assert_pack_include "#{packno}", "@#{packno}: expect pack is not equal to recv pack!"
+                }
+                sleep 1
+		        datas = getpacks ".trans"
+                i=0
+                open("#{MCSERVER_PATH}/log/foo.log", "r").each{ |line|
+                        if line=~/--------/ then next end
+                        puts "trans_rev=================>"
+                        puts line
+                        puts "==========================>"
+                        case_path = "#{$CASE_ROOT}/"
+                        file = open(case_path+"/data/#{datas[i]}.trans", "r")
+                        i=i+1
+                        line2 = file.gets
+                        file.close
+                        assert_include line2, line, "Transfer data not match"
+                }
+                system "killall -9 mcserver_trans"
+        end
+
+
+        # ===功能
+        # cm transfer测试模式的一种修改，发送的数据就已经是通用transfer的格式,可用于类cm transfer的子系统
+        # 发送cm，检查cm返回数据，检查transfer发送数据
+	    # 通过前缀来进行数据的匹配检查，不要求每个cm的发送的数据对有对应的transfer返回数据
+	    # 例如： 1.send 对应1.expc， 不要求同时有 1.trans，如果有则进行transfer返回的数据的检验
+	    #
+        # 
+        # ===参数
+        #
+        # ===Example
+	    # def test_xxx
+        # 	cm_transfer_pattern2
+	    # end
+	    # def teardown
+	    #	cm_transfer_pattern_stop
+	    # end
+        # 
+	    # 备注：
+        # 增加teardown函数处理case运行失败的情况,见: cm_transfer_pattern_stop
+    	# 能保证发送的数据按照前缀的顺序发送
+        def cm_transfer_pattern2(ip=HOST, port=PORT)
+                # 清理cm transfer原来的数据
+                cmd_str = "cd #{MCSERVER_PATH}; rm log/foo.log; nohup ./mcserver_trans &"
+                system "#{cmd_str}"
+                datas = getpacks
+                datas.each{ |packno|
+                        mc_trans_send ip, port, nil, packno
+                        mc_assert_pack_include "#{packno}", "@#{packno}: expect pack is not equal to recv pack!"
+                }
+                sleep 1
+	        	datas = getpacks ".trans"
+                i=0
+                open("#{MCSERVER_PATH}/log/foo.log", "r").each{ |line|
+                        if line=~/--------/ then next end
+                        puts "trans_rev=================>"
+                        puts line
+                        puts "==========================>"
+                        case_path = "#{$CASE_ROOT}/"
+                        file = open(case_path+"/data/#{datas[i]}.trans", "r")
+                        i=i+1
+                        line2 = file.gets
+                        file.close
+                        assert_include line2, line, "Transfer data not match"
+                }
+                system "killall -9 mcserver_trans"
+        end
+
+
+        # ===功能
+        # cm transfer模式的停止
+        # 主要是停止mcserver的工作
+        # 
+        # ===参数 
+        # 无
+        # 
+        # ===Example
+        # def teardown
+        #       cm_transfer_pattern_stop
+        # end
+        #
+        def cm_transfer_pattern_stop
+                system "killall -9 mcserver_trans"
+        end
+
+end