selenium提供的截图功能

get_screenshot_as_file()提供一个截屏功能。在自动化执行过程中，执行失败后只能看到代码的运行错误，而不能直接看到ui上的错误，利用截屏保存下来很容易的进行问题的判断
 
先来执行一个通过的用例：

 from selenium import webdriver
 import time
 import os
 
 def test_search_baidu():
     driver = webdriver.Chrome()
     driver.implicitly_wait(5) # 隐性等待5秒
     driver.get('http://www.baidu.com')
 
     word = '韩国 v 发觉你狗日价格每公斤'
     driver.find_element_by_id('kw').send_keys(word) # 搜索框中输入word
     driver.find_element_by_id('su').click() # 点击搜索
     time.sleep(1)
 
     # 判断，如果word在请求的页面中则通过，不在则截屏
     img_path = os.getcwd() # 保存截图的路径
     if word in driver.page_source:
         print('testcase is pass!')
     else:
         driver.get_screenshot_as_file(img_path+"fail_img.png")
 
     # 关闭浏览器
     driver.quit()
 
 # 运行程序
 if __name__ == '__main__':
     test_search_baidu()
 # 结果：testcase is pass!

改下代码，让用例执行成功后截图，失败后打印测试失败：

 from selenium import webdriver
 import time
 import os
 
 def test_search_baidu():
     driver = webdriver.Chrome()
     driver.implicitly_wait(5) # 隐性等待5秒
     driver.get('http://www.baidu.com')
 
     word = '韩国 v 发觉你狗日价格每公斤'
     driver.find_element_by_id('kw').send_keys(word) # 搜索框中输入word
     driver.find_element_by_id('su').click() # 点击搜索
     time.sleep(1)
 
     # 判断，如果word在请求的页面中通过则截图
     img_path = os.getcwd() # 保存截图的路径
     print(img_path)
     if word in driver.page_source:
         driver.get_screenshot_as_file(img_path + "sucess_img.png")
     else:
         print('testcase is fail!')
 
     # 关闭浏览器
     driver.quit()
 
 # 运行程序
 if __name__ == '__main__':
     test_search_baidu()

截图如下：