Skip to content
This repository was archived by the owner on Aug 10, 2022. It is now read-only.

Commit b0e3329

Browse files
committed
Add cut_provider to viewport screenshot
1 parent 0a07417 commit b0e3329

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

eyes_selenium/applitools/selenium/selenium_eyes.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
PositionProvider,
2828
RegionProvider,
2929
TextTrigger,
30+
NullCutProvider,
3031
)
3132
from applitools.selenium.capture.eyes_webdriver_screenshot import (
3233
EyesWebDriverScreenshotFactory,
@@ -756,10 +757,18 @@ def _viewport_screenshot(self, scale_provider):
756757
sleep(self.configuration.wait_before_screenshots / 1000.0)
757758
image = self._image_provider.get_image()
758759
self._debug_screenshot_provider.save(image, "original")
760+
759761
scale_provider.update_scale_ratio(image.width)
760762
pixel_ratio = 1 / scale_provider.scale_ratio
761763
if pixel_ratio != 1.0:
764+
logger.info("Scalling")
762765
image = image_utils.scale_image(image, 1.0 / pixel_ratio)
766+
self._debug_screenshot_provider.save(image, "scaled")
767+
768+
if not isinstance(self.cut_provider, NullCutProvider):
769+
logger.info("Cutting")
770+
image = self.cut_provider.cut(image)
771+
self._debug_screenshot_provider.save(image, "cutted")
763772

764773
return EyesWebDriverScreenshot.create_viewport(self._driver, image)
765774

eyes_selenium/applitools/selenium/visual_grid/render_task.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ class RenderTask(VGTask):
3636
rendering_info = attr.ib()
3737
region_selectors = attr.ib(hash=False, factory=list)
3838
size_mode = attr.ib(default=None)
39-
region_to_check = attr.ib(hash=False, default=None)
39+
region_to_check = attr.ib(hash=False, default=None) # type: Region
4040
agent_id = attr.ib(default=None)
4141
func_to_run = attr.ib(default=None, hash=False, repr=False) # type: Callable
4242

0 commit comments

Comments
 (0)