From 97ca29b220848a228e22cf634d6aeaf647cf5e15 Mon Sep 17 00:00:00 2001 From: Hrutvik Jagtap Date: Mon, 7 Oct 2024 14:25:16 +0530 Subject: [PATCH] Return base64 image in case of EMBED This can be useful in case if someone wants to reuse the image i.e set test message or set suite documentation --- src/SeleniumLibrary/keywords/screenshot.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/SeleniumLibrary/keywords/screenshot.py b/src/SeleniumLibrary/keywords/screenshot.py index 8cd8dc299..aeb614655 100644 --- a/src/SeleniumLibrary/keywords/screenshot.py +++ b/src/SeleniumLibrary/keywords/screenshot.py @@ -89,7 +89,7 @@ def capture_page_screenshot(self, filename: str = DEFAULT_FILENAME_PAGE) -> str: format string syntax]. An absolute path to the created screenshot file is returned or if - ``filename`` equals to EMBED, word `EMBED` is returned. + ``filename`` equals to EMBED, base64 image is returned. Support for EMBED is new in SeleniumLibrary 4.2 @@ -126,7 +126,7 @@ def _capture_page_screenshot_to_file(self, filename): def _capture_page_screen_to_log(self): screenshot_as_base64 = self.driver.get_screenshot_as_base64() self._embed_to_log_as_base64(screenshot_as_base64, 800) - return EMBED + return screenshot_as_base64 @keyword def capture_element_screenshot( @@ -173,7 +173,7 @@ def _capture_element_screenshot_to_file(self, element, filename): def _capture_element_screen_to_log(self, element): self._embed_to_log_as_base64(element.screenshot_as_base64, 400) - return EMBED + return element.screenshot_as_base64 @property def _screenshot_root_directory(self):