diff --git a/src/main/java/testUI/elements/Element.java b/src/main/java/testUI/elements/Element.java index 7af6aaf..33ca149 100644 --- a/src/main/java/testUI/elements/Element.java +++ b/src/main/java/testUI/elements/Element.java @@ -817,8 +817,12 @@ public Asserts should() { return shouldHave(); } - public UIElement saveScreenshot(String path) { + if (!path.toLowerCase().endsWith(".png")) { + Logger.putLogWarn("Invalid file extension: Only .png files are supported. Screenshot saved in: "+ Configuration.screenshotPath + path); + return getElementObject(); + } + if (!Configuration.automationType.equals(Configuration.DESKTOP_PLATFORM)) { if (getDrivers().size() != 0) { Configuration.driver = Math.min(Configuration.driver, getDrivers().size());