def test_can_make_screenshot_with_custom_name():
visit(start_page)
actual = take_screenshot(filename="custom")
expected = os.path.join(get_default_screenshot_folder(), 'custom.png')
assert expected == actual
assert os.path.exists(actual)
def test_can_make_screenshot_with_default_name():
visit(start_page)
actual = take_screenshot()
expected = os.path.join(get_default_screenshot_folder(),
'screen_{id}.png'.format(id=get_screen_id()))
assert expected == actual
assert os.path.exists(actual)
def test_can_save_screenshot_to_custom_folder_specified_as_parameter():
screenshot_folder = os.path.dirname(os.path.abspath(__file__)) + '/../../build/screenshots'
open_url(start_page)
actual = take_screenshot(path=screenshot_folder)
expected = os.path.join(screenshot_folder,
'screen_{id}.png'.format(id=get_screen_id()))
assert expected == actual
assert os.path.isfile(actual)
assert os.path.exists(actual)
def test_can_save_screenshot_to_custom_folder_specified_through_config():
config.reports_folder = os.path.dirname(os.path.abspath(__file__)) + '/../../build/screenshots'
open_url(start_page)
actual = take_screenshot()
expected = os.path.join(get_default_screenshot_folder(),
'screen_{id}.png'.format(id=get_screen_id()))
assert expected == actual
assert os.path.isfile(actual)
assert os.path.exists(actual)
def test_can_save_screenshot_to_custom_folder_with_custom_name():
screenshot_folder = os.path.dirname(os.path.abspath(__file__)) + '/../../build/screenshots'
open_url(start_page)
actual = take_screenshot(path=screenshot_folder,
filename="custom_file_in_custom_folder")
expected = os.path.join(screenshot_folder, 'custom_file_in_custom_folder.png')
assert expected == actual
assert os.path.isfile(actual)
assert os.path.exists(actual)
def compare_canvas(self, f, name, name_expected, test):
time.sleep(1)
#TODO add object as a paremeter
image_folder = f
element = s("#canvas123")
location = element.location
size = element.size
path = browser.take_screenshot(path=image_folder, filename='{}'.format(name))
x = location['x']
y = location['y']
width = location['x'] + size['width']
height = location['y'] + size['height']
im = Image.open(path)
im = im.crop((int(x), int(y), int(width), int(height)))
im.save(path)
return self.compare_screens(name_expected, path, image_folder, test)
def compare(self, f, name, name_expected, test):
image_folder = f
time.sleep(2)
#Will take a screenshot as png and will place it in newly created folder
path = browser.take_screenshot(path=image_folder, filename='{}'.format(name))
return self.compare_screens(name_expected, path, image_folder, test)
def screenshot(self, f, name):
image_folder = f
time.sleep(2)
browser.take_screenshot(path=image_folder, filename='{}'.format(name))