Ensure the screenshots directory exists before trying to save regression test screenshots

pull/3/head
Atira Odhner 2017-03-15 22:06:04 +00:00 committed by Dave Page
parent cc4686f1bb
commit 2abd3724c2
1 changed files with 16 additions and 7 deletions

View File

@ -61,8 +61,23 @@ class BaseFeatureTest(BaseTestGenerator):
return False
def _screenshot(self):
path = '{0}/../screenshots/{1}'.format(self.CURRENT_PATH, self.server["name"].replace(" ", "_"))
screenshots_directory = '{0}/../screenshots'.format(self.CURRENT_PATH)
screenshots_server_directory = '{0}/{1}'.format(screenshots_directory,
self.server["name"].replace(" ", "_"))
self.ensure_directory_exists(screenshots_directory)
self.ensure_directory_exists(screenshots_server_directory)
date = datetime.now().strftime("%Y.%m.%d_%H.%M.%S")
python_version = sys.version.split(" ")[0]
self.page.driver.save_screenshot(
'{0}/{1}-{2}-Python-{3}.png'.format(screenshots_server_directory,
self.__class__.__name__,
date,
python_version))
def ensure_directory_exists(self, path):
try:
os.mkdir(path)
except OSError as e:
@ -70,9 +85,3 @@ class BaseFeatureTest(BaseTestGenerator):
pass
else:
raise
date = datetime.now().strftime("%Y.%m.%d_%H.%M.%S")
python_version = sys.version.split(" ")[0]
self.page.driver.save_screenshot(
'{0}/{1}-{2}-Python-{3}.png'.format(path, self.__class__.__name__, date, python_version))