I want to automatic capturing screenshots if my webdriver tests failed (any exception or assertion error). I am using a Python unit test and Selenium Webdriver. Does anyone have any solution to this problem?
Use the following code using Firefox WebDriver which saves a screenshot on any exception to a dated image file:
from datetime import datetimefrom selenium import webdriverbrowser = webdriver.Firefox()try: # do some webdriver stuff hereexcept Exception as e: print e now = datetime.now().strftime('%Y-%m-%d_%H-%M-%S') browser.get_screenshot_as_file('screenshot-%s.png' % now)
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)