actir 1.0.0

data/lib/actir/remote.rb

@@ -0,0 +1,80 @@
+require 'actir/config'
+
+module Actir
+
+  #
+  # 远程测试环境相关的方法
+  #
+  # @author: Hub
+  #
+  # @Date: 2015-3-6 
+  #
+  module Remote
+
+    #
+    # 获取远程selenium测试环境的docker镜像的IPAddress
+    #
+    # @example : get_remote_address 2
+    #
+    # @param   : num [Fixnum] selenium-grid 的node镜像的节点数,对应多进程测试时的进程数量  
+    #
+    # @return  : [Array] IPAddress字符串的数组,形如["127.0.0.1:5555", "127.0.0.2:5555"]
+    #
+    def self.get_remote_address(num = 0)
+      @docker_cfg = Actir::Config.get("config.test_mode.docker") if @docker_cfg == nil
+      docker_ip = @docker_cfg["ip"]
+      node_sub_name = ( (@docker_cfg["name"] == nil || @docker_cfg["name"] == "") ? "-node" : @docker_cfg["name"])
+      docker_node_name = $env + node_sub_name
+      docker_inspect_str = "docker inspect -f='{{.NetworkSettings.IPAddress}}' \\`docker ps | grep #{docker_node_name} | grep 5900 | awk '{print \\$11}'\\`"
+      #需要判断执行脚本的环境是本地还是Linux服务器,本地需要ssh 
+      puts docker_inspect_str if $debug 
+      ip_str = if is_local?
+        `ssh root@#{docker_ip} "#{docker_inspect_str}"`
+      else
+        `#{docker_inspect_str}`
+      end
+      ip_array = ip_str.split("\n")
+      address = []
+      #如果入参num小于address.size,则返回num个address
+      #如果入参num大于address.size,则返回所有address
+      #如果入参num为0即默认的不传入参,则返回所有address
+      ip_array.each_with_index do |ip, i| 
+        address[ip_array.size - 1 -i] = ip + ":" + @docker_cfg["port"]
+      end
+      if ip_array.size >= num && num != 0
+        return address.first(num)
+      else
+        return address
+      end
+    end
+
+    # 获取远端docker-selenium的node镜像的个数
+    #
+    # @example : get_remote_num
+    #
+    # @return  : [Fixnum] 远端docker-selenium的node镜像的个数
+    #
+    # def self.get_remote_num 
+    #   @docker_cfg = Actir::Config.get("config.test_mode.docker") if @docker_cfg == nil
+    #   docker_ip = @docker_cfg["ip"]
+    #   node_sub_name = ( (@docker_cfg["name"] == nil || @docker_cfg["name"] == "") ? "-node" : @docker_cfg["name"])
+    #   docker_node_name = $env + node_sub_name
+    #   #5900是node节点的端口号
+    #   docker_inspect_str = "docker ps | grep #{docker_node_name} | grep -c 5900"
+    #   num = if is_local?
+    #     `ssh root@#{docker_ip} "#{docker_inspect_str}"`
+    #   else
+    #     `#{docker_inspect_str}`
+    #   end
+    #   num.to_i
+    # end
+    
+    # 判断执行环境是否是本地环境(Mac)
+    def self.is_local?
+      hostname = `hostname`
+      hostname.include? "local"
+    end
+
+  end
+
+end

data/lib/actir/script/cookies_baidu.rb

@@ -0,0 +1,143 @@
+require 'watir-webdriver'
+
+module Actir
+  class CookiesBaidu
+
+    class << self
+
+      # 更新百度账号所有的配置文件
+      def update_all(address = [])
+        baidu_card = Actir::Config.get("cookies.baidu")
+        baidu_card.each do |card, value|
+          update_cookies(card, address)
+        end
+      end
+
+      #将所有百度支付卡的available状态恢复为true
+      def re_available
+        #每次登陆都判断一下cookies文件的上一次修改时间和当前时间
+        #如果日期不同，则刷新所有的cookies文件中baidu-card的状态
+        unless Actir::Config.is_same_day?("cookies")
+          Actir::Config.lock("cookies") do
+            str_array = IO.readlines(cookies_file)
+            str_array.each_with_index do |line, index|
+              if line =~ /available\:\s*false/  
+                str_array[index] = line.gsub(/false/, 'true')
+              end
+            end
+            cookiesfile = File.open(cookies_file, 'w')
+            str_array.each do |line|
+              cookiesfile.puts line
+            end
+            cookiesfile.close
+          end
+        end
+      end
+
+      # 获取可用的百度账号的hash
+      # 返回值{card1 => {"username" => "xxx", "password"=>"iloveyouzan", xxx}}
+      def get_useful_card
+        # old_config = Actir::Config.config_dir
+        # Actir::Config.config_dir = script_config_path
+        #通过配置文件判断取出可用的卡的参数
+        baidu_card = Actir::Config.get("cookies.baidu")
+        card = {}
+        baidu_card.each do |key, value|
+          if value["available"] == true
+            #有可用的卡，取出cookies等参数
+            card.store(key, value)
+            break
+          end
+        end
+        # Actir::Config.config_dir = old_config
+        card
+      end
+
+      # 设置不可用的卡
+      # 入参传入卡的key
+      def set_useless_card(card)
+        Actir::Config.set("cookies.baidu." + card + "." + "available", "false")
+      end
+
+      private
+
+      # 更新配置文件中的baidu_cookies
+      def update_cookies(card, address = [])
+        #打开百付宝
+        open_baifubao(address)
+        #获取对应卡的账号密码
+        args = Actir::Config.get("cookies.baidu." + card)
+        #登录百付宝
+        login_baifubao(args["username"], args["password"])
+        #获取cookies
+        cookies = get_baifubao_cookies
+        #清除之前的cookies
+        modify_cookies(card, cookies)
+        puts "Already update baifubao's cookies"
+      end
+
+      # 访问百付宝主页
+      def open_baifubao(address = [])
+        if address.size == 0
+          if $mode == :remote
+            address = Actir::Remote.get_remote_address(1)
+          end
+        end
+        @browser = Browser.new(:www, :url => address[0], :mode => $mode)
+        @browser.goto "baifubao.com"
+        @browser 
+      end
+
+      # 登录百付宝主页
+      def login_baifubao(username, password)
+        text_baifubao_username.set username
+        text_baifubao_password.set password
+        button_baifubao_login.click
+      end
+
+      # 获取cookies
+      def get_baifubao_cookies
+        id =  @browser.cookies[:BAIDUID][:value]
+        ss =  @browser.cookies[:BDUSS][:value]
+        @browser.close
+        #cookies = "  BAIDUID:\s\s\s\s\s\s\"" + id + "\"\n  BDUSS:\s\s\s\s\s\s\s\s\"" + ss + "\"\n"
+        #以hash形式返回
+        {:BAIDUID => id, :BDUSS => ss } 
+      end
+
+      def modify_cookies(card, cookies)
+        cookies.each do |key, value|
+          Actir::Config.set("cookies.baidu."+card+"."+key.to_s , "\""+value.to_s+"\"")
+        end
+      end
+
+      # 配置文件所在文件夹的路径
+      def script_config_path
+        File.join(File.dirname(__FILE__), "config")
+      end
+
+      # 配置文件相对路径
+      def cookies_file
+        File.join(Actir::Config.config_dir, "/cookies.yaml")
+      end
+
+      def text_baifubao_username
+        @browser.text_field(:id => 'TANGRAM__PSP_4__userName')
+      end
+
+      def text_baifubao_password
+        @browser.text_field(:id => 'TANGRAM__PSP_4__password')
+      end
+
+      def button_baifubao_login
+        #@browser.input(:xpath, "//input[@id='TANGRAM__PSP_4__submit']")
+        @browser.input(:id => 'TANGRAM__PSP_4__submit')
+      end
+
+    end
+  end
+
+end
+
+#Actir::CookiesBaidu.re_available
+

data/lib/actir/version.rb

@@ -0,0 +1,3 @@
+module Actir
+  VERSION = "1.0.0"
+end

data/lib/actir/webdriver/browser.rb

@@ -0,0 +1,201 @@
+require 'actir/webdriver/driver'
+
+#
+# 初始化测试用浏览器
+#
+# @author: Hub
+#
+# @Date: 2015-1-5 
+#
+class Browser
+
+  attr_reader :browser
+
+  PHANTOMJS_SIZE = {'width' => 1334, 'height' => 750}
+  REMOTE_SIZE    = {'width' => 1050, 'height' => 611}
+
+  #
+  # 初始化函数
+  #
+  # 可以配置测试的mode：
+  #  1.:env
+  #     :local  本地环境上测试
+  #     :remote 远程测试环境上测试
+  #  2.:browser
+  #     :chrome
+  #     :firefox 
+  #  3.:agent wap页面的useragent类型 
+  #     :iphone
+  #     :android_phone
+  #  4.:address  当env为:remote时，url指定远程执行脚本的地址
+  #
+  def initialize(type = :www, *args)
+    args = init_args(*args)
+    @browser_type = args[:browser]
+    @agent = args[:agent]
+    @env = args[:mode]
+    if @env == :remote
+      @url = if args[:url]
+        "http://#{args[:url]}/wd/hub"
+      else
+        #TO-DO ,远程模式没有传入IP,改成local模式
+        @env = :local
+        puts "selenium-node's IPAddress is null. switch to local mode"
+      end
+    end
+    if type == :www
+      @watir_browser = browser_www
+    elsif type == :wap
+      @watir_browser = browser_wap
+    else
+      raise "error browser type , please send args with :www or :wap!"
+    end
+    define_page_method
+  end
+
+  # 初始化入参
+  def init_args(args = {})
+    unless args.has_key?(:mode)
+      #若通过actir执行测试用例，则会配置ENV的模式
+      if ENV["mode"]
+        args[:mode] = ENV["mode"].to_sym
+      else
+        #若ENV为空，则读取配置文件，判断有无配置文件
+        if File.exist?(config_file) 
+          args[:mode] = $config["config"]["test_mode"]["env"]
+        else
+          args[:mode] = :local
+        end
+      end
+    end
+    unless args.has_key?(:browser)
+      if File.exist?(config_file) 
+        args[:browser] = $config["config"]["test_mode"]["browser"]
+      else
+        args[:browser] = :chrome
+      end
+    end
+    args[:agent] = :iphone  unless args.has_key?(:agent)
+    args[:url] = $address  unless args.has_key?(:url)
+    args
+  end
+
+
+  #
+  # 打开普通www浏览器
+  #
+  def browser_www
+    case @env
+    when :local
+      #本地chrome浏览器
+      browser = Watir::Browser.new @browser_type
+    when :remote
+      #远程服务器的chrome浏览器
+      browser = Watir::Browser.new(
+        :remote, 
+        :desired_capabilities => @browser_type, 
+        :url => @url
+      )
+    end
+    #重新设置窗口大小,不然phantomjs的ghost driver各种问题
+    if @browser_type == :phantomjs
+      browser.window.resize_to(PHANTOMJS_SIZE["width"], PHANTOMJS_SIZE["height"])
+    elsif @env == :remote
+      browser.window.resize_to(REMOTE_SIZE["width"], REMOTE_SIZE["height"])
+    end
+    browser
+  end
+
+  #
+  # 通过useragent打开WAP页面
+  #
+  # local模式下直接打开本地chrome浏览器,或者使用phantomjs无界面执行
+  #
+  # remote模式下通过设置chrome的switches参数远程打开测试环境上的chrome浏览器
+  #
+  # TO-DO: remote模式的phantomjs
+  #
+  def browser_wap
+    case @env
+    when :local
+      driver = Actir::Webdriver.driver(:browser => @browser_type, :agent => @agent)
+    when :remote
+      driver = Actir::Webdriver.driver(:browser => @browser_type, :agent => @agent, :url => @url)
+    end
+    browser = Watir::Browser.new driver
+    if @browser_type == :phantomjs
+      #重新设置窗口大小,不然phantomjs的ghost driver各种问题 = =
+      browser.window.resize_to(PHANTOMJS_SIZE["width"], PHANTOMJS_SIZE["height"])
+    end
+    browser
+  end
+  
+  # 自动定义驱动需要调用的所有元素相关的方法
+  # 需要所有的页面元素相关的类名以XXXPage方式命名
+  def define_page_method
+    p Module.constants.grep(/(Page|Wap)$/) if $debug
+    Module.constants.grep(/(Page|Wap)$/).each do |page_klass|
+      if basic_page?(page_klass)
+        method = ""
+        if page_klass.to_s =~ /Page$/
+          method_name = page_klass.to_s.gsub!(/Page$/, "_page")
+        elsif page_klass.to_s =~ /Wap$/
+          method_name = page_klass.to_s.gsub!(/Wap$/, "_wap")
+        end
+        self.class.send :define_method, "#{method_name.downcase}" do
+          page = Module.const_get(page_klass).new(@watir_browser)
+          page
+        end #define_method
+        puts "defined #{method_name.downcase}" if $debug
+      end #if
+    end #each
+  end
+
+  def valid_page_klass? klass
+    return false if klass.eql?(:Page)
+    return false if klass.eql?(:BasicPage)
+    Module.const_get(klass) < Actir::BasicPage
+  end
+  alias_method :basic_page?, :valid_page_klass?
+
+  def method_missing(m, *args, &blk)
+    if @watir_browser.respond_to? m
+      @watir_browser.send(m, *args, &blk)
+    else
+      super
+    end #if
+  end
+
+  #
+  # 在同一个浏览器的新tab页上打开指定url的页面,并自动将焦点放到新开窗口中
+  #
+  # @example : 
+  #    browser.new_window('youzan.com') do
+  #      action
+  #    end
+  #
+  # @param url : [String] url字符串
+  # 
+  def new_window url
+    @watir_browser.execute_script("window.open(\"#{url}\")")
+    #获取浏览器中的窗口数目
+    window_num = @watir_browser.windows.length
+    #默认要操作的新窗口都是最后一个窗口
+    @watir_browser.windows[window_num - 1].use do  
+      #重新设置窗口大小,不然phantomjs的ghost driver各种问题 = =
+      if @browser_type == :phantomjs
+        @watir_browser.window.resize_to(PHANTOMJS_SIZE["width"],PHANTOMJS_SIZE["height"])
+      end
+      #使用执行块，由调用者决定在新开的订单详情页面内做些啥
+      yield
+    end
+  end
+
+  private
+
+  def config_file
+    File.join($project_path, "config", "config.yaml")
+  end
+
+end
+

data/lib/actir/webdriver/browser_options.rb

@@ -0,0 +1,54 @@
+require 'facets/hash/except'
+
+module Actir
+  module Webdriver
+    class BrowserOptions
+
+      def initialize(opts, user_agent_string)
+        @options = opts
+        options[:browser] ||= :chrome
+        options[:agent] ||= :iphone
+        #options[:orientation] ||= :portrait
+        initialize_for_browser(user_agent_string)
+      end
+
+      def method_missing(*args, &block)
+        m = args.first
+        value = options[m]
+        super unless value
+        value.downcase
+      end
+
+      def browser_options
+        #options.except(:browser, :agent, :orientation)
+        options.except(:browser, :agent)
+      end
+
+      private
+
+      def options
+        @options ||= {}
+      end
+
+      def initialize_for_browser(user_agent_string)
+        case options[:browser]
+        when :firefox
+          options[:profile] ||= Selenium::WebDriver::Firefox::Profile.new
+          options[:profile]['general.useragent.override'] = user_agent_string
+        when :chrome
+          options[:switches] ||= []
+          options[:switches] << "--user-agent=#{user_agent_string}"
+        # add bu Hub
+        # support phantomjs
+        when :phantomjs
+          options[:desired_capabilities] ||= Selenium::WebDriver::Remote::Capabilities.phantomjs(
+            "phantomjs.page.settings.userAgent" => user_agent_string
+          )
+        else
+          raise "WebDriver currently only supports :chrome, :firefox and :phantomjs"
+        end
+        
+      end
+    end
+  end
+end