Python多线程Selenium跨浏览器自动化测试实战

#-*- coding:utf-8 -*-

__author__ = u'苦叶子'

from selenium import webdriver
import sys
from time import sleep
from threading import Thread

reload(sys)
sys.setdefaultencoding("utf-8")

def test_baidu_search(browser, url):
  driver = None
  # 你可以自定义这里，添加更多浏览器支持进来
  if browser == "ie":
    driver = webdriver.Ie() 
  elif browser == "firefox":
    driver = webdriver.Firefox() 
  elif browser == "chrome":
    driver = webdriver.Chrome() 

  if driver == None:
    exit()

  print u"开始[case_0001]百度搜索"
  driver.get(url)

  print u"清除搜索中数据，输入搜索关键词"
  driver.find_element_by_id("kw").clear()
  driver.find_element_by_id("kw").send_keys(u"开源优测")

  print u"单击 百度一下 按钮 开始搜索"
  driver.find_element_by_id("su").click()
  sleep(3)

  print u"关闭浏览器，退出webdriver"
  driver.quit()


if __name__ == "__main__":
  # 浏览器和首页url
  data = {
    "ie":"http://www.baidu.com", 
    "firefox":"http://www.baidu.com", 
    "chrome":"http://www.baidu.com"
    }

  # 构建线程 
  threads = [] 
  for b, url in data.items(): 
    t = Thread(target=test_baidu_search,args=(b,url))
    threads.append(t) 

  # 启动所有线程
  for thr in threads:
    thr.start()