Implemented Viewport Screenshot Strategy.

Author: iamcharankumar

src/main/java/io/swaglabs/portal/qa/screenshotsmanager/ViewportScreenshotStrategy.java

@@ -0,0 +1,17 @@
+package io.swaglabs.portal.qa.screenshotsmanager;
+
+import com.microsoft.playwright.Page;
+import lombok.RequiredArgsConstructor;
+
+import java.nio.file.Paths;
+
+@RequiredArgsConstructor
+public class ViewportScreenshotStrategy implements ScreenshotStrategy {
+
+    private final Page PAGE;
+
+    @Override
+    public void capture(String filePath) {
+        PAGE.screenshot(new Page.ScreenshotOptions().setPath(Paths.get(filePath)).setFullPage(false));
+    }
+}

src/main/java/io/swaglabs/portal/qa/utils/ScreenshotsUtils.java

@@ -6,6 +6,7 @@
 import io.swaglabs.portal.qa.screenshotsmanager.ElementScreenshotStrategy;
 import io.swaglabs.portal.qa.screenshotsmanager.FullPageScreenshotStrategy;
 import io.swaglabs.portal.qa.screenshotsmanager.ScreenshotContext;
+import io.swaglabs.portal.qa.screenshotsmanager.ViewportScreenshotStrategy;
 import lombok.AccessLevel;
 import lombok.NoArgsConstructor;
 
@@ -29,10 +30,16 @@ public static ScreenshotsUtils getInstance() {
         return instance;
     }
 
-    public ScreenshotContext getFullPageScreenshotContext(Page page) {
+    public ScreenshotContext takeFullPageScreenshotContext(Page page) {
+        Objects.requireNonNull(page, "Page cannot be null.");
         return new ScreenshotContext(new FullPageScreenshotStrategy(page));
     }
 
+    public ScreenshotContext takeViewportScreenshotContext(Page page) {
+        Objects.requireNonNull(page, "Page cannot be null.");
+        return new ScreenshotContext(new ViewportScreenshotStrategy(page));
+    }
+
     public void takeElementScreenshot(Locator locator, String fileName) {
         Objects.requireNonNull(locator, "Locator cannot be null.");
         Objects.requireNonNull(fileName, "File name cannot be null.");

src/test/java/io/swaglabs/portal/qa/listeners/WebTestListeners.java

@@ -71,7 +71,7 @@ private void takeScreenshot(ITestResult testResult) {
         String filePath = Paths.get(dirPath, fileName).toString();
         try {
             Files.createDirectories(Paths.get(dirPath));
-            WebPortalConstants.SCREENSHOTS_UTILS.getFullPageScreenshotContext(currentPage).captureScreenshot(filePath);
+            WebPortalConstants.SCREENSHOTS_UTILS.takeFullPageScreenshotContext(currentPage).captureScreenshot(filePath);
         } catch (IOException e) {
             log.error("Screenshot failed for {}: {}", testName, e.getMessage());
         }