python - Python 中 Selenium Webdriver 测试失败时的自动截图

Question

如果我的 webdriver 测试失败(任何异常或断言错误)，我想自动捕获屏幕截图。我正在使用 Python unittest 和 Selenium Webdriver。有没有人可以解决这个问题？

Answer 1

在 Firefox 中做一些 webdriver 的工作...将任何异常的屏幕截图保存到过时的图像文件中:

from datetime import datetime
from selenium import webdriver

browser = webdriver.Firefox()

try:
    # do some webdriver stuff here
except Exception as e:
    print e
    now = datetime.now().strftime('%Y-%m-%d_%H-%M-%S')
    browser.get_screenshot_as_file('screenshot-%s.png' % now)