webdriver高级应用- 测试过程中发生异常或断言失败时进行屏幕截图

封装了三个类来实现这个功能：

1.DataUtil.py 用于获取当前的日期以及时间，用于生成保存截图文件的目录名，代码如下：

#encoding=utf-8
import time
from datetime import datetime

'''
本文件主要用于获取当前的日期以及时间，
用于生成保存截图文件目录名
'''

def currentDate():
    date = time.localtime()
    # 构造今天的日期字符串
    today = str(date.tm_year) + "-" + str(date.tm_mon) + "-" + str(date.tm_mday)
    return today

def currentTime():
     timeStr = datetime.now()
     # 构造当前时间字符串
     now = timeStr.strftime('%H-%M-%S')
     return now

if __name__ == "__main__":
    print currentDate()
    print currentTime()

FileUtil.py 主要用于创建目录，用于存放异常截图，代码如下：

#encoding=utf-8
from DateUtil import currentDate, currentTime
import os

'''
本文件主要用于创建目录，用于存放异常截图，
创建目录的方法仅供大家参考，将来用于根据测试
需要创建测试人员需要的目录或文件等
'''

def createDir():
    # 获得当前文件所在目录绝对路径
    currentPath = os.path.dirname(os.path.abspath(__file__))
    # 获取今天的日期字符串
    today = currentDate()
    # 构造以今天日期命名的目录的绝对路径
    dateDir = os.path.join(currentPath, today)
    print dateDir
    if not os.path.exists(dateDir):
        # 如果已今天日期命名的目录不存在则创建
        os.mkdir(dateDir)
    # 获得当前的时间字符串
    now = currentTime()
    # 构造以当前时间命名的目录的绝对路径
    timeDir = os.path.join(dateDir, now)
    print timeDir
    if not os.path.exists(timeDir):
        # 如果已当前时间命名的目录不存在则创建
        os.mkdir(timeDir)
    return timeDir

if  __name__=='__main__':
    createDir()

3. Failure_Capture.py 截屏方法封装，代码如下：

#encoding=utf-8
from selenium import webdriver
import unittest, time, os
from FileUtil import createDir
import traceback

# 创建存放异常截图的目录，并得到本次实例中存放图片目录的绝对路径，
# 并且作为全局变量，以供本次所有测试用例调用
picDir = createDir()

def takeScreenshot(driver, savePath, picName):
    # 封装截屏方法
    # 构造屏幕截图路径及图片名
    # 因为Windows默认编码是GBK，而传入的图片名为utf8编码，
    # 所以这里需要进行转码，以便让图片名中的中文字符能正常显示
    picPath = os.path.join(savePath, str(picName).decode("utf-8").encode("gbk")
                           + ".png")
    try:
        # 调用WebDriver提供的get_screenshot_as_file()方法，
        # 将截取的屏幕图片保存为本地文件
        driver.get_screenshot_as_file(picPath)
    except Exception, e:
        # 打印异常堆栈信息
        print traceback.print_exc()

class TestFailCaptureScreen(unittest.TestCase):

    def setUp(self):
        # 启动Firefox浏览器
        self.driver = webdriver.Ie(executable_path = "e:\IEDriverServer")

    def testSoGouSearch(self):
        url = "http://www.sogou.com"
        # 访问搜狗首页
        self.driver.get(url)
        try:
            self.driver.find_element_by_id("query").
                send_keys(u"光荣之路自动化测试")
            self.driver.find_element_by_id("stb").click()
            time.sleep(3)
            # 断言页面的代码中是否存在“事在人为”这4个关键字，
            # 因为页面中没有这4个字，所以会触发except语句的执行，并触发截图操作
            self.assertTrue(u"事在人为" in self.driver.page_source, 
                            "“事在人为”关键字串在页面源代码中未找到")
        except AssertionError, e:
            # 调用封装好的截图方法，进行截图并保存在本地磁盘
            takeScreenshot(self.driver, picDir, e)
        except Exception, e:
            print traceback.print_exc()
            takeScreenshot(self.driver, picDir, e)

    def tearDown(self):
        # 退出IE浏览器
        self.driver.quit()

if __name__ == '__main__':
    unittest.main()