33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
"""Screenshot Plugin for SeleniumBase tests that run with pynose / nosetests"""
|
|
import os
|
|
from nose.plugins import Plugin
|
|
from seleniumbase.config import settings
|
|
|
|
|
|
class ScreenShots(Plugin):
|
|
"""This plugin takes a screenshot when a test fails."""
|
|
name = "screen_shots"
|
|
logfile_name = settings.SCREENSHOT_NAME
|
|
|
|
def options(self, parser, env):
|
|
super().options(parser, env=env)
|
|
|
|
def configure(self, options, conf):
|
|
super().configure(options, conf)
|
|
if not self.enabled:
|
|
return
|
|
self.options = options
|
|
|
|
def add_screenshot(self, test, err, capt=None, tbinfo=None):
|
|
test_logpath = self.options.log_path + "/" + test.id()
|
|
if not os.path.exists(test_logpath):
|
|
os.makedirs(test_logpath)
|
|
screenshot_file = "%s/%s" % (test_logpath, self.logfile_name)
|
|
test.driver.get_screenshot_as_file(screenshot_file)
|
|
|
|
def addError(self, test, err, capt=None):
|
|
self.add_screenshot(test, err, capt=capt)
|
|
|
|
def addFailure(self, test, err, capt=None, tbinfo=None):
|
|
self.add_screenshot(test, err, capt=capt, tbinfo=tbinfo)
|